Merge pull request #40 from terrateamio/add-format-atlantis-comment

ADD Format atlantis-comment support

Author: orbitz

src/oiq/oiq_pricer.ml

@@ -529,3 +529,132 @@ let to_markdown_string t =
         render_section "🔴 Removed resources" removes;
         render_section "⚪ Existing resources" existing;
       ])
+
+let to_atlantis_comment_string t =
+  let fmt v = Printf.sprintf "$%.2f" (CCFloat.abs v) in
+  let fmt_with_sign v = Printf.sprintf "%s$%.2f" (if v >= 0.0 then "" else "-") (CCFloat.abs v) in
+
+  (* Initial box header with emojis *)
+  let lines =
+    [
+      "\n";
+      (* Add an empty line for spacing *)
+      "╔══════════════════════════════════════════════════════════╗\n";
+      "║        💸 OpenInfraQuote Monthly Cost Estimate           ║\n";
+      "╚══════════════════════════════════════════════════════════╝\n";
+    ]
+  in
+
+  (* Price difference section *)
+  let delta_min = t.price_diff.Oiq_range.min in
+  let delta_max = t.price_diff.Oiq_range.max in
+  let lines =
+    if delta_min = 0.0 && delta_max = 0.0 then lines @ [ "\n"; "No change in monthly cost\n" ]
+    else if delta_max < 0.0 then
+      lines
+      @ [
+          "\n";
+          Printf.sprintf
+            "Decrease: %s → %s/month\n"
+            (fmt (CCFloat.abs delta_min))
+            (fmt (CCFloat.abs delta_max));
+        ]
+    else
+      lines @ [ "\n"; Printf.sprintf "Increase: %s → %s/month\n" (fmt delta_min) (fmt delta_max) ]
+  in
+
+  (* Before/After summary *)
+  let lines =
+    lines
+    @ [
+        "\n";
+        Printf.sprintf
+          "Before: %s - %s\n"
+          (fmt_with_sign t.prev_price.Oiq_range.min)
+          (fmt_with_sign t.prev_price.Oiq_range.max);
+        Printf.sprintf
+          "After:  %s - %s\n"
+          (fmt_with_sign t.price.Oiq_range.min)
+          (fmt_with_sign t.price.Oiq_range.max);
+      ]
+  in
+
+  (* Group resources by change type *)
+  let adds, removes, existing =
+    CCList.fold_left
+      (fun (a, r, e) resource ->
+        match resource.Resource.change with
+        | `Add -> (resource :: a, r, e)
+        | `Remove -> (a, resource :: r, e)
+        | `Noop -> (a, r, resource :: e))
+      ([], [], [])
+      t.resources
+  in
+
+  (* Add resource counts *)
+  let lines =
+    lines
+    @ [
+        "\n";
+        Printf.sprintf "%-2s %-10s %d\n" "🟢" "Added:" (CCList.length adds);
+        Printf.sprintf "%-2s %-10s %d\n" "🔴" "Removed:" (CCList.length removes);
+        Printf.sprintf "%-2s %-10s %d\n" "⚪" "Existing:" (CCList.length existing);
+      ]
+  in
+
+  (* Helper to render a section of resources *)
+  let render_section title resources =
+    if CCList.is_empty resources then ""
+    else
+      let header = Printf.sprintf "\n%s:\n%s\n" title (String.make (String.length title) '-') in
+      let table_header =
+        Printf.sprintf "%-40s %-20s %12s %12s\n" "Resource" "Type" "Before" "After"
+      in
+      let table_divider =
+        Printf.sprintf
+          "%-40s %-20s %12s %12s\n"
+          (String.make 40 '-')
+          (String.make 20 '-')
+          (String.make 12 '-')
+          (String.make 12 '-')
+      in
+      let rows =
+        CCString.concat
+          ""
+          (CCList.map
+             (fun r ->
+               let name =
+                 if
+                   not
+                     (CCString.equal r.Resource.address (r.Resource.type_ ^ "." ^ r.Resource.name))
+                 then
+                   CCString.concat "." @@ CCList.tl @@ CCString.split_on_char '.' r.Resource.address
+                 else r.Resource.name
+               in
+               let typ = r.Resource.type_ in
+               let est_min = r.Resource.price.Oiq_range.min in
+               let est_max = r.Resource.price.Oiq_range.max in
+               (* Use spaces around hyphens and preserve signs for removed resources *)
+               let before_range =
+                 Printf.sprintf
+                   "%s - %s"
+                   (fmt_with_sign t.prev_price.Oiq_range.min)
+                   (fmt_with_sign t.prev_price.Oiq_range.max)
+               in
+               let after_range =
+                 Printf.sprintf "%s - %s" (fmt_with_sign est_min) (fmt_with_sign est_max)
+               in
+               Printf.sprintf "%-40s %-20s %12s %12s\n" name typ before_range after_range)
+             resources)
+      in
+      header ^ table_header ^ table_divider ^ rows
+  in
+
+  CCString.concat
+    ""
+    (lines
+    @ [
+        render_section "Added resources" adds;
+        render_section "Removed resources" removes;
+        render_section "Existing resources" existing;
+      ])

src/oiq/oiq_pricer.mli

@@ -14,3 +14,4 @@ val price :
 
 val pretty_to_string : t -> string
 val to_markdown_string : t -> string
+val to_atlantis_comment_string : t -> string

src/oiq_cli/oiq_cli.ml

@@ -27,14 +27,14 @@ module Cli = struct
 
   let output_format =
     let docv = "FORMAT" in
-    let doc = "Format of output (text, json, or markdown)." in
+    let doc = "Format of output (text, json, markdown, atlantis-comment)." in
     let env =
       let doc = "Specify output format" in
       C.Cmd.Env.info ~doc "OIQ_OUTPUT_FORMAT"
     in
     C.Arg.(
       value
-      & opt (enum [ ("text", `Text); ("json", `Json); ("markdown", `Markdown) ]) `Text
+      & opt (enum [ ("text", `Text); ("json", `Json); ("markdown", `Markdown); ("atlantis-comment", `Atlantis_comment); ]) `Text
       & info [ "f"; "format" ] ~docv ~doc ~env)
 
   let match_query =
@@ -162,7 +162,8 @@ let price usage input output_format match_query regions =
       match output_format with
       | `Text -> print_endline @@ Oiq_pricer.pretty_to_string priced
       | `Json -> print_endline @@ Yojson.Safe.pretty_to_string @@ Oiq_pricer.to_yojson priced
-      | `Markdown -> print_endline @@ Oiq_pricer.to_markdown_string priced)
+      | `Markdown -> print_endline @@ Oiq_pricer.to_markdown_string priced
+      | `Atlantis_comment -> print_endline @@ Oiq_pricer.to_atlantis_comment_string priced)
   | Error (#Oiq_match_query.err as err) ->
       Logs.err (fun m -> m "%a" Oiq_match_query.pp_err err);
       exit 1