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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
open Core
open Async
let json_arg = Command.Arg_type.create Yojson.Safe.from_string
let call ?endpoint_url ?profile ?region f m result_to_json error_to_json =
let region =
match region with
| Some region -> Some (Awso.Region.of_string region)
| None -> None in
(Awso_async.Cfg.get_exn ?profile ?region ()) >>=
(fun cfg ->
(f ?endpoint_url ?cfg:(Some cfg) m) >>=
(fun result ->
match result with
| Error err ->
(match error_to_json with
| None ->
failwithf
"endpoint error, but no error values defined in boto"
()
| Some to_json ->
let s = (err |> to_json) |> Yojson.Safe.to_string in
failwithf "AWS error: %s" s ())
| Ok result ->
((match result_to_json with
| None -> print_endline "ok response from endpoint"
| Some to_json ->
((result |> to_json) |> Yojson.Safe.to_string) |>
print_endline);
return ())))
let search =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and cursor = flag "cursor" (optional string) ~doc:"STRING Cursor"
and expr = flag "expr" (optional string) ~doc:"STRING Expr"
and facet = flag "facet" (optional string) ~doc:"STRING Facet"
and filterQuery =
flag "filter-query" (optional string) ~doc:"STRING FilterQuery"
and highlight =
flag "highlight" (optional string) ~doc:"STRING Highlight"
and partial = flag "partial" (optional bool) ~doc:"BOOL Partial"
and queryOptions =
flag "query-options" (optional string) ~doc:"STRING QueryOptions"
and queryParser =
flag "query-parser" (optional json_arg) ~doc:"JSON QueryParser"
and return = flag "return" (optional string) ~doc:"STRING Return"
and size = flag "size" (optional json_arg) ~doc:"JSON Size"
and sort = flag "sort" (optional string) ~doc:"STRING Sort"
and start = flag "start" (optional json_arg) ~doc:"JSON Start"
and stats = flag "stats" (optional string) ~doc:"STRING Stat"
and query = flag "query" (required string) ~doc:"STRING Query" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region Io.search
(Values.SearchRequest.make ?cursor ?expr ?facet ?filterQuery
?highlight ?partial ?queryOptions
?queryParser:(Option.map ~f:Values.QueryParser.of_json
queryParser) ?return
?size:(Option.map ~f:Values.Size.of_json size) ?sort
?start:(Option.map ~f:Values.Start.of_json start) ?stats ~query
()) (Some Values.SearchResponse.to_json)
(Some Values.SearchResponse.error_to_json)])
let suggest =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and size = flag "size" (optional json_arg) ~doc:"JSON SuggestionsSize"
and query = flag "query" (required string) ~doc:"STRING Query"
and suggester =
flag "suggester" (required string) ~doc:"STRING Suggester" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.suggest
(Values.SuggestRequest.make
?size:(Option.map ~f:Values.SuggestionsSize.of_json size)
~query ~suggester ()) (Some Values.SuggestResponse.to_json)
(Some Values.SuggestResponse.error_to_json)])
let upload_documents =
Command.async ~summary:""
([%map_open.Command
let cli_profile =
flag "-cli-profile" (optional string) ~doc:"NAME aws profile to use"
and cli_region =
flag "-cli-region" (optional string) ~doc:"REGION override region"
and endpoint_url =
flag "-endpoint-url" (optional string)
~doc:"URL override endpoint url"
and documents = flag "documents" (required json_arg) ~doc:"JSON Blob"
and contentType =
flag "content-type" (required json_arg) ~doc:"JSON ContentType" in
fun () ->
call ?endpoint_url ?profile:cli_profile ?region:cli_region
Io.upload_documents
(Values.UploadDocumentsRequest.make
~documents:(Values.Blob.of_json documents)
~contentType:(Values.ContentType.of_json contentType) ())
(Some Values.UploadDocumentsResponse.to_json)
(Some Values.UploadDocumentsResponse.error_to_json)])
let main =
Command.group
~summary:((Awso.Service.to_string Values.service) ^ " commands")
[("search", search);
("suggest", suggest);
("upload-documents", upload_documents)]