{
(***********************************************************************)
(*                             OCamldoc                                *)
(*                                                                     *)
(*            Maxence Guesdon, projet Cristal, INRIA Rocquencourt      *)
(*                                                                     *)
(*  Copyright 2001 Institut National de Recherche en Informatique et   *)
(*  en Automatique.  All rights reserved.  This file is distributed    *)
(*  under the terms of the Q Public License version 1.0.               *)
(*                                                                     *)
(***********************************************************************)

(** The lexer for special comments. *)

open Lexing
open Odoc_parser

let line_number = ref 0


let string_buffer = Buffer.create 32

(** Fonction de remise à zéro de la chaine de caractères tampon *)
let reset_string_buffer () = Buffer.reset string_buffer

(** Fonction d'ajout d'un caractère dans la chaine de caractères tampon *)
let ajout_char_string = Buffer.add_char string_buffer

(** Add a string to the buffer. *)
let ajout_string = Buffer.add_string string_buffer

let lecture_string () = Buffer.contents string_buffer

(** The variable which will contain the description string. 
   Is initialized when we encounter the start of a special comment. *)
let description = ref ""

let blank = "[ \013\009\012]"

(** The nested comments level. *)
let comments_level = ref 0

let print_DEBUG2 s = print_string s; print_newline ()

(** This function returns the given string without the leading and trailing blanks.*)
let remove_blanks s =
  print_DEBUG2 ("remove_blanks "^s);
  let l = Str.split_delim (Str.regexp "\n") s in
  let l2 =
    let rec iter liste =
      match liste with
	h :: q ->
	  let h2 = Str.global_replace (Str.regexp ("^"^blank^"+")) "" h in 
	  if h2 = "" then
	    (
	     print_DEBUG2 (h^" n'a que des blancs");
	     (* we remove this line and must remove leading blanks of the next one *)
	     iter q
	    )
	  else
	    (* we don't remove leading blanks in the remaining lines *)
	    h2 :: q
      | _ ->
	  []
    in iter l
  in
  let l3 = 
    let rec iter liste = 
      match liste with
	h :: q ->
	  let h2 = Str.global_replace (Str.regexp (blank^"+$")) "" h in 
	  if h2 = "" then
	    (
	     print_DEBUG2 (h^" n'a que des blancs");
	     (* we remove this line and must remove trailing blanks of the next one *)
	     iter q
	    )
	  else
	    (* we don't remove trailing blanks in the remaining lines *)
	    h2 :: q
      | _ ->
	  []
    in
    List.rev (iter (List.rev l2))
  in
  String.concat "\n" l3

(** Remove first blank characters of each line of a string, until the first '*' *)
let remove_stars s =
  let s2 = Str.global_replace (Str.regexp ("^"^blank^"*\\*")) "" s in
  s2
} 

let lowercase = ['a'-'z' '\223'-'\246' '\248'-'\255' '_']
let uppercase = ['A'-'Z' '\192'-'\214' '\216'-'\222']
let identchar = 
  ['A'-'Z' 'a'-'z' '_' '\192'-'\214' '\216'-'\246' '\248'-'\255' '\'' '0'-'9']

rule main = parse
    [' ' '\013' '\009' '\012'] +
      { 
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	main lexbuf
      }

  | [ '\010' ]
      {
	incr line_number;
	incr Odoc_comments_global.nb_chars;
        main lexbuf 
      }
  | "(**)"
      {
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	Description ("", None)
      }	

  | "(**"("*"+)")"
      {
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	main lexbuf
      }	

  | "(***"
      {
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	incr comments_level;
	main lexbuf
      }	

  | "(**"
      { 
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	incr comments_level;
	if !comments_level = 1 then
	  (
	   reset_string_buffer ();
	   description := "";
	   special_comment lexbuf 
	  )
	else
	  main lexbuf
      }

  | eof
      { EOF }

  |  "*)"
      {
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	decr comments_level ;
	main lexbuf
      }	

  |  "(*"
      {
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	incr comments_level ;
	main lexbuf
      }	

  | _
      { 
        incr Odoc_comments_global.nb_chars;
	main lexbuf
      }

and special_comment = parse
  | "*)"
      { 
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	if !comments_level = 1 then
	  (
           (* there is just a description *)
	   let s2 = lecture_string () in
	   let s3 = remove_blanks s2 in
	   let s4 = 
	     if !Odoc_args.remove_stars then
	       remove_stars s3
	     else
	       s3
	   in
	   Description (s4, None)
	  )
	else
	  (
	   ajout_string s;
	   decr comments_level;
	   special_comment lexbuf
	  )
      }

  |  "(*"
      {
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	incr comments_level ;
	ajout_string s;
	special_comment lexbuf
      }	

  | "\\@"
      { 
	let s = Lexing.lexeme lexbuf in
	let c = (Lexing.lexeme_char lexbuf 1) in
        ajout_char_string c;
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
        special_comment lexbuf 
      }	

  | "@"lowercase+
      {
	(* we keep the description before we go further *)
	let s = lecture_string () in
	description := remove_blanks s;
	reset_string_buffer ();
	let len = String.length (Lexing.lexeme lexbuf) in
	lexbuf.Lexing.lex_abs_pos <- lexbuf.Lexing.lex_abs_pos - len;
	lexbuf.Lexing.lex_curr_pos <- lexbuf.Lexing.lex_curr_pos - len;
	lexbuf.Lexing.lex_last_pos <- lexbuf.Lexing.lex_last_pos - len;
	(* we don't increment the Odoc_comments_global.nb_chars *)
	special_comment_part2 lexbuf
      }	

  | _
      { 
	let c = (Lexing.lexeme_char lexbuf 0) in
        ajout_char_string c;
        if c = '\010' then incr line_number;
	incr Odoc_comments_global.nb_chars;
        special_comment lexbuf 
      }	

and special_comment_part2 = parse
  | "*)"
      { 
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	if !comments_level = 1 then
  	  (* finally we return the description we kept *)
	  let desc = 
	    if !Odoc_args.remove_stars then
	      remove_stars !description
	     else
	      !description
	  in
	  let remain = lecture_string () in
	  let remain2 = 
	    if !Odoc_args.remove_stars then
	      remove_stars remain
	     else
	       remain
	  in
	  Description (desc, Some remain2)
	else
	  (
	   ajout_string s ;
	   decr comments_level ;
	   special_comment_part2 lexbuf
	  )
      }

  |  "(*"
      {
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	ajout_string s;
	incr comments_level ;
	special_comment_part2 lexbuf
      }	

  | _
      { 
	let c = (Lexing.lexeme_char lexbuf 0) in
        ajout_char_string c;
        if c = '\010' then incr line_number;
	incr Odoc_comments_global.nb_chars;
        special_comment_part2 lexbuf 
      }	

and elements = parse
  | [' ' '\013' '\009' '\012'] +
      { 
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	elements lexbuf
      }

  | [ '\010' ]
      { incr line_number;
	incr Odoc_comments_global.nb_chars;
	print_DEBUG2 "newline";
        elements lexbuf }

  | "@"lowercase+
      { 
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	let s2 = String.sub s 1 ((String.length s) - 1) in
	print_DEBUG2 s2;
	match s2 with
	  "param" ->
	    T_PARAM 
	 | "author" ->
	    T_AUTHOR
	 | "version" ->
	     T_VERSION
	 | "see" ->
	     T_SEE
	 | "since" ->
	     T_SINCE
	 | "deprecated" ->
	     T_DEPRECATED
	 | "raise" ->
	     T_RAISES
	 | "return" ->
	     T_RETURN
	 | s ->
	     if !Odoc_args.no_custom_tags then
	       raise (Failure (Odoc_messages.not_a_valid_tag s))
	     else
	       T_CUSTOM s
      }	

  | ("\\@" | [^'@'])+
      {
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	let s = Lexing.lexeme lexbuf in
	let s2 = remove_blanks s in
	print_DEBUG2 ("Desc "^s2);
	Desc s2
      }	
  | eof
      {
	EOF
      }	
    

and simple = parse
    [' ' '\013' '\009' '\012'] +
      { 
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	simple lexbuf
      }

  | [ '\010' ]
      { incr line_number;
	incr Odoc_comments_global.nb_chars;
        simple lexbuf 
      }

  | "(**"("*"+) 
      {
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length (Lexing.lexeme lexbuf));
	incr comments_level;
	simple lexbuf
      }	

  | "(*"("*"+)")"
      {
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	simple lexbuf
      }	
  | "(**"
      {
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	incr comments_level;
	simple lexbuf
      }	

  | "(*"
      { 
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	incr comments_level;
	if !comments_level = 1 then
	  (
	   reset_string_buffer ();
	   description := "";
	   special_comment lexbuf 
	  )
	else
	  (
	   ajout_string s;
	   simple lexbuf
	  )
      }

  | eof
      { EOF }

  |  "*)"
      {
	let s = Lexing.lexeme lexbuf in
	Odoc_comments_global.nb_chars := !Odoc_comments_global.nb_chars + (String.length s);
	decr comments_level ;
	simple lexbuf
      }	

  | _
      {
	incr Odoc_comments_global.nb_chars;
	simple lexbuf
      }