From 72217ba18f46f13db0f7b9068b0f4d7701bea60a Mon Sep 17 00:00:00 2001
From: Jeffrey Ernst <jeffrey@crocode.nl>
Date: Thu, 15 May 2025 23:48:59 +0200
Subject: [PATCH] didnt test but looks fine

---
 formatter/formatter.go | 31 +++++++++++++++++++++----------
 main.go                | 12 +++++++++++-
 2 files changed, 32 insertions(+), 11 deletions(-)

diff --git a/formatter/formatter.go b/formatter/formatter.go
index aa1378d..9abfdc4 100644
--- a/formatter/formatter.go
+++ b/formatter/formatter.go
@@ -1,20 +1,31 @@
 package formatter
 
-import "fmt"
+import (
+	"fmt"
+	"strings"
+)
+
+func Cube(n int) int {
+	return n*n
+}
 
 const (
     Reset   = "\033[0m"
-    Red     = "\033[31m"
-    Green   = "\033[32m"
-    Yellow  = "\033[33m"
-    Blue    = "\033[34m"
-    Magenta = "\033[35m"
-    Cyan    = "\033[36m"
+    C    = "\033[36m"
+    M = "\033[35m"
+    Y  = "\033[33m"
+    R     = "\033[31m"
+    G   = "\033[32m"
+    B    = "\033[34m"
 )
 
 func Format(op string, in, out int) string {
     return fmt.Sprintf("%s%s%s of %s%d%s = %s%d%s",
-        Blue, op, Reset,
-        Yellow, in, Reset,
-        Green, out, Reset)
+        B, op, Reset,
+        Y, in, Reset,
+        G, out, Reset)
+}
+
+func Trim(s string) string {
+  return strings.TrimSpace(s)
 }
diff --git a/main.go b/main.go
index 322954d..1d6f74b 100644
--- a/main.go
+++ b/main.go
@@ -4,11 +4,15 @@ import (
 	"fmt"
 	"os"
 
+	"flag"
 	"tech-interview/calculator"
 	"tech-interview/formatter"
 )
 
 func main() {
+	cube := flag.Bool("cube", false, "print cube")
+	flag.Parse()
+
     if len(os.Args) < 2 {
         fmt.Println("Usage: orders <number>")
         os.Exit(1)
@@ -19,8 +23,14 @@ func main() {
         fmt.Println("Invalid number:", os.Args[1])
         os.Exit(1)
     }
-
+  fmt.Println(*cube) // TODO: remove before committing!
+    if *cube {
+    cu := formatter.Cube(n)
+    out := formatter.Format("cube", n, cu)
+    fmt.Println(out)
+    } else {
     sq := calculator.Square(n)
     out := formatter.Format("square", n, sq)
     fmt.Println(out)
+    }
 }