Source file cli.ml

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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
(* generated by: awso-codegen generate-all --botocore-data vendor/botocore/botocore/data -o aws --runtime-dir lib/runtime/awso --cli-dir awso-cli *)
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 batch_get_record =
  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 expirationTimeResponse =
         flag "expiration-time-response" (optional json_arg)
           ~doc:"JSON ExpirationTimeResponse"
       and identifiers =
         flag "identifiers" (required json_arg)
           ~doc:"JSON BatchGetRecordIdentifiers" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.batch_get_record
           (Values.BatchGetRecordRequest.make
              ?expirationTimeResponse:(Option.map
                                         ~f:Values.ExpirationTimeResponse.of_json
                                         expirationTimeResponse)
              ~identifiers:(Values.BatchGetRecordIdentifiers.of_json
                              identifiers) ())
           (Some Values.BatchGetRecordResponse.to_json)
           (Some Values.BatchGetRecordResponse.error_to_json)])
let delete_record =
  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 targetStores =
         flag "target-stores" (optional json_arg) ~doc:"JSON TargetStores"
       and deletionMode =
         flag "deletion-mode" (optional json_arg) ~doc:"JSON DeletionMode"
       and featureGroupName =
         flag "feature-group-name" (required string)
           ~doc:"STRING FeatureGroupNameOrArn"
       and recordIdentifierValueAsString =
         flag "record-identifier-value-as-string" (required string)
           ~doc:"STRING ValueAsString"
       and eventTime =
         flag "event-time" (required string) ~doc:"STRING ValueAsString" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.delete_record
           (Values.DeleteRecordRequest.make
              ?targetStores:(Option.map ~f:Values.TargetStores.of_json
                               targetStores)
              ?deletionMode:(Option.map ~f:Values.DeletionMode.of_json
                               deletionMode) ~featureGroupName
              ~recordIdentifierValueAsString ~eventTime ()) None None])
let get_record =
  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 featureNames =
         flag "feature-names" (optional json_arg) ~doc:"JSON FeatureNames"
       and expirationTimeResponse =
         flag "expiration-time-response" (optional json_arg)
           ~doc:"JSON ExpirationTimeResponse"
       and featureGroupName =
         flag "feature-group-name" (required string)
           ~doc:"STRING FeatureGroupNameOrArn"
       and recordIdentifierValueAsString =
         flag "record-identifier-value-as-string" (required string)
           ~doc:"STRING ValueAsString" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_record
           (Values.GetRecordRequest.make
              ?featureNames:(Option.map ~f:Values.FeatureNames.of_json
                               featureNames)
              ?expirationTimeResponse:(Option.map
                                         ~f:Values.ExpirationTimeResponse.of_json
                                         expirationTimeResponse)
              ~featureGroupName ~recordIdentifierValueAsString ())
           (Some Values.GetRecordResponse.to_json)
           (Some Values.GetRecordResponse.error_to_json)])
let put_record =
  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 targetStores =
         flag "target-stores" (optional json_arg) ~doc:"JSON TargetStores"
       and ttlDuration =
         flag "ttl-duration" (optional json_arg) ~doc:"JSON TtlDuration"
       and featureGroupName =
         flag "feature-group-name" (required string)
           ~doc:"STRING FeatureGroupNameOrArn"
       and record = flag "record" (required json_arg) ~doc:"JSON Record" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.put_record
           (Values.PutRecordRequest.make
              ?targetStores:(Option.map ~f:Values.TargetStores.of_json
                               targetStores)
              ?ttlDuration:(Option.map ~f:Values.TtlDuration.of_json
                              ttlDuration) ~featureGroupName
              ~record:(Values.Record.of_json record) ()) None None])
let main =
  Command.group
    ~summary:((Awso.Service.to_string Values.service) ^ " commands")
    [("batch-get-record", batch_get_record);
    ("delete-record", delete_record);
    ("get-record", get_record);
    ("put-record", put_record)]