Source file cli.ml

(* 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_findings =
  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 findingIdentifiers =
         flag "finding-identifiers" (required json_arg)
           ~doc:"JSON FindingIdentifiers" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.batch_get_findings
           (Values.BatchGetFindingsRequest.make
              ~findingIdentifiers:(Values.FindingIdentifiers.of_json
                                     findingIdentifiers) ())
           (Some Values.BatchGetFindingsResponse.to_json)
           (Some Values.BatchGetFindingsResponse.error_to_json)])
let create_scan =
  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 clientToken =
         flag "client-token" (optional string) ~doc:"STRING ClientToken"
       and scanType =
         flag "scan-type" (optional json_arg) ~doc:"JSON ScanType"
       and analysisType =
         flag "analysis-type" (optional json_arg) ~doc:"JSON AnalysisType"
       and tags = flag "tags" (optional json_arg) ~doc:"JSON TagMap"
       and resourceId =
         flag "resource-id" (required json_arg) ~doc:"JSON ResourceId"
       and scanName =
         flag "scan-name" (required string) ~doc:"STRING ScanName" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.create_scan
           (Values.CreateScanRequest.make ?clientToken
              ?scanType:(Option.map ~f:Values.ScanType.of_json scanType)
              ?analysisType:(Option.map ~f:Values.AnalysisType.of_json
                               analysisType)
              ?tags:(Option.map ~f:Values.TagMap.of_json tags)
              ~resourceId:(Values.ResourceId.of_json resourceId) ~scanName ())
           (Some Values.CreateScanResponse.to_json)
           (Some Values.CreateScanResponse.error_to_json)])
let create_upload_url =
  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 scanName =
         flag "scan-name" (required string) ~doc:"STRING ScanName" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.create_upload_url
           (Values.CreateUploadUrlRequest.make ~scanName ())
           (Some Values.CreateUploadUrlResponse.to_json)
           (Some Values.CreateUploadUrlResponse.error_to_json)])
let get_account_configuration =
  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 () = return () in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_account_configuration
           (Values.GetAccountConfigurationRequest.make ())
           (Some Values.GetAccountConfigurationResponse.to_json)
           (Some Values.GetAccountConfigurationResponse.error_to_json)])
let get_findings =
  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 nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and maxResults =
         flag "max-results" (optional int)
           ~doc:"INT GetFindingsRequestMaxResultsInteger"
       and status = flag "status" (optional json_arg) ~doc:"JSON Status"
       and scanName =
         flag "scan-name" (required string) ~doc:"STRING ScanName" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_findings
           (Values.GetFindingsRequest.make ?nextToken ?maxResults
              ?status:(Option.map ~f:Values.Status.of_json status) ~scanName
              ()) (Some Values.GetFindingsResponse.to_json)
           (Some Values.GetFindingsResponse.error_to_json)])
let get_metrics_summary =
  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 date = flag "date" (required json_arg) ~doc:"JSON Timestamp" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_metrics_summary
           (Values.GetMetricsSummaryRequest.make
              ~date:(Values.Timestamp.of_json date) ())
           (Some Values.GetMetricsSummaryResponse.to_json)
           (Some Values.GetMetricsSummaryResponse.error_to_json)])
let get_scan =
  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 runId = flag "run-id" (optional string) ~doc:"STRING Uuid"
       and scanName =
         flag "scan-name" (required string) ~doc:"STRING ScanName" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.get_scan (Values.GetScanRequest.make ?runId ~scanName ())
           (Some Values.GetScanResponse.to_json)
           (Some Values.GetScanResponse.error_to_json)])
let list_findings_metrics =
  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 nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and maxResults =
         flag "max-results" (optional int)
           ~doc:"INT ListFindingsMetricsRequestMaxResultsInteger"
       and startDate =
         flag "start-date" (required json_arg) ~doc:"JSON Timestamp"
       and endDate =
         flag "end-date" (required json_arg) ~doc:"JSON Timestamp" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_findings_metrics
           (Values.ListFindingsMetricsRequest.make ?nextToken ?maxResults
              ~startDate:(Values.Timestamp.of_json startDate)
              ~endDate:(Values.Timestamp.of_json endDate) ())
           (Some Values.ListFindingsMetricsResponse.to_json)
           (Some Values.ListFindingsMetricsResponse.error_to_json)])
let list_scans =
  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 nextToken =
         flag "next-token" (optional string) ~doc:"STRING NextToken"
       and maxResults =
         flag "max-results" (optional int)
           ~doc:"INT ListScansRequestMaxResultsInteger" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_scans
           (Values.ListScansRequest.make ?nextToken ?maxResults ())
           (Some Values.ListScansResponse.to_json)
           (Some Values.ListScansResponse.error_to_json)])
let list_tags_for_resource =
  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 resourceArn =
         flag "resource-arn" (required string) ~doc:"STRING ScanNameArn" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.list_tags_for_resource
           (Values.ListTagsForResourceRequest.make ~resourceArn ())
           (Some Values.ListTagsForResourceResponse.to_json)
           (Some Values.ListTagsForResourceResponse.error_to_json)])
let tag_resource =
  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 resourceArn =
         flag "resource-arn" (required string) ~doc:"STRING ScanNameArn"
       and tags = flag "tags" (required json_arg) ~doc:"JSON TagMap" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.tag_resource
           (Values.TagResourceRequest.make ~resourceArn
              ~tags:(Values.TagMap.of_json tags) ())
           (Some Values.TagResourceResponse.to_json)
           (Some Values.TagResourceResponse.error_to_json)])
let untag_resource =
  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 resourceArn =
         flag "resource-arn" (required string) ~doc:"STRING ScanNameArn"
       and tagKeys =
         flag "tag-keys" (required json_arg) ~doc:"JSON TagKeyList" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.untag_resource
           (Values.UntagResourceRequest.make ~resourceArn
              ~tagKeys:(Values.TagKeyList.of_json tagKeys) ())
           (Some Values.UntagResourceResponse.to_json)
           (Some Values.UntagResourceResponse.error_to_json)])
let update_account_configuration =
  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 encryptionConfig =
         flag "encryption-config" (required json_arg)
           ~doc:"JSON EncryptionConfig" in
       fun () ->
         call ?endpoint_url ?profile:cli_profile ?region:cli_region
           Io.update_account_configuration
           (Values.UpdateAccountConfigurationRequest.make
              ~encryptionConfig:(Values.EncryptionConfig.of_json
                                   encryptionConfig) ())
           (Some Values.UpdateAccountConfigurationResponse.to_json)
           (Some Values.UpdateAccountConfigurationResponse.error_to_json)])
let main =
  Command.group
    ~summary:((Awso.Service.to_string Values.service) ^ " commands")
    [("batch-get-findings", batch_get_findings);
    ("create-scan", create_scan);
    ("create-upload-url", create_upload_url);
    ("get-account-configuration", get_account_configuration);
    ("get-findings", get_findings);
    ("get-metrics-summary", get_metrics_summary);
    ("get-scan", get_scan);
    ("list-findings-metrics", list_findings_metrics);
    ("list-scans", list_scans);
    ("list-tags-for-resource", list_tags_for_resource);
    ("tag-resource", tag_resource);
    ("untag-resource", untag_resource);
    ("update-account-configuration", update_account_configuration)]