summaryrefslogtreecommitdiff
path: root/cmd/perc/main.go
blob: 848428e4a5d03dca75e5adb816ba153ee9691e07 (plain)
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
package main

import (
	"fmt"
	"os"
	"strings"

	"codeberg.org/snonux/perc/internal"
	"codeberg.org/snonux/perc/internal/calculator"
	"codeberg.org/snonux/perc/internal/rpn"
	"codeberg.org/snonux/perc/internal/repl"
	"github.com/mattn/go-isatty"
)

func main() {
	output, err := runCommand(os.Args)
	if err != nil {
		fmt.Println("Error:", err)
		os.Exit(1)
	}
	fmt.Println(output)
}

func runCommand(args []string) (string, error) {
	if len(args) < 2 {
		// No args provided - check if stdin is a TTY for REPL mode
		if isatty.IsTerminal(os.Stdin.Fd()) {
			repl.RunREPL()
			return "", nil
		}
		printUsage()
		return "", fmt.Errorf("no input provided")
	}

	if args[1] == "version" {
		return internal.Version, nil
	}

	// Check for --repl flag
	if args[1] == "--repl" || args[1] == "repl" {
		repl.RunREPL()
		return "", nil
	}

	// Check for calc subcommand
	if args[1] == "calc" || args[1] == "rpn" {
		if len(args) < 3 {
			return "", fmt.Errorf("missing expression after '%s'", args[1])
		}
		input := strings.Join(args[2:], " ")
		result, err := runRPN(input)
		if err != nil {
			return "", err
		}
		return result, nil
	}

	input := strings.Join(args[1:], " ")
	
	// Try RPN parsing first (for bare RPN expressions like "3 4 +")
	if rpnResult, rpnErr := runRPN(input); rpnErr == nil {
		return rpnResult, nil
	}
	
	// Fall back to percentage calculation
	result, err := calculator.Parse(input)
	if err != nil {
		return "", err
	}

	return result, nil
}

// runRPN parses and evaluates an RPN expression
func runRPN(input string) (string, error) {
	vars := rpn.NewVariables()
	rpnCalc := rpn.NewRPN(vars)
	return rpnCalc.ParseAndEvaluate(input)
}

func printUsage() {
	fmt.Println("Usage: perc <calculation>")
	fmt.Println("       perc calc <rpn-expression>")
	fmt.Println("       perc rpn <rpn-expression>")
	fmt.Println("       perc version")
	fmt.Println("       perc [--repl|repl]")
	fmt.Println("\nPercentage calculator examples:")
	fmt.Println("  perc 20% of 150")
	fmt.Println("  perc what is 20% of 150")
	fmt.Println("  perc 30 is what % of 150")
	fmt.Println("  perc 30 is 20% of what")
	fmt.Println("\nRPN (postfix notation) examples:")
	fmt.Println("  perc calc 3 4 +")
	fmt.Println("  perc calc 3 4 + 4 4 - *")
	fmt.Println("  perc calc x = 5 x x +")
	fmt.Println("  perc calc 2 3 ^")
	fmt.Println("  perc calc dup swap pop show")
	fmt.Println("\nStart REPL mode interactively by running without arguments:")
	fmt.Println("  perc")
}