1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
module Opts = struct
open Arg
type lasttype = NAME | SYMB | NUMB
(** Positional arguments search based on the last type searched. *)
let ltype = ref NAME
(** Json file that generates the Sqlite3 database. *)
let jsonf = ref ""
(** Limit the amount of matches. *)
let maxmatch = ref 3
(** Lists to search. *)
let match_name : string list ref = ref []
let match_symb : string list ref = ref []
let match_numb : int list ref = ref []
let dbfile = ref "elements.sqlite3"
let add_name s =
match_name := s::!match_name;
ltype := NAME
let add_symb s =
match_symb := s::!match_symb;
ltype := SYMB
let add_numb i =
match_numb := i::!match_numb;
ltype := NUMB
let add_def s = match !ltype with
| NAME -> add_name s
| SYMB -> add_symb s
| NUMB -> add_numb (match int_of_string_opt s with
| None -> raise (Bad (Printf.sprintf "expected integer, got '%s'" s))
| Some x -> x
)
let msg = "Periodic table search\n\
elements -gendb file [-dbfile outfile]\n\
elements [-maxmatch n] [-dbfile search] [-name ...] [-symb ...]\n\
"
let setfile name v f =
if f = "" then raise
(Bad (Printf.sprintf "%s requires a filename" name))
else v := f
let spec = [
"-gendb", String (setfile "-gendb" jsonf), "Generate DB file";
"-maxmatch", Set_int maxmatch, "Limit the amount of matches (def. 3)";
"-dbfile", String (setfile "-dbfile" dbfile), "Set the DB filename";
"-name", String add_name, "Search by name (default)";
"-symb", String add_symb, "Search by symbol";
"-num", Int add_numb, "Search by atomic number"
]
let exec() = parse spec add_def msg
end
let () = Opts.exec()
let db = Sqlite3.db_open !Opts.dbfile
let _ = if !Opts.jsonf <> "" then
Makedb.makedb db (if !Opts.jsonf = "-" then
Yojson.Basic.from_channel stdin
else
Yojson.Basic.from_file !Opts.jsonf
)
|
