feat: better structured internal

internal/cli/arguments.go

@@ -5,18 +5,24 @@ import (
 	"fmt"
 )
 
+// Arguments given to program.
 type CLIOptions struct {
-	Input       string
-	Verbose     bool
+	// The source code given to the program.
+	Input string
+	// Whether or not to print debug logs.
+	Verbose bool
+	// Whether or not to print an explanation of the reduction.
 	Explanation bool
 }
 
-func ParseOptions(args []string) (*CLIOptions, error) {
-	// Parse flags and arguments.
+// Extract the program configuration from the command-line arguments.
+func ParseOptions() (*CLIOptions, error) {
+	// Parse flags.
 	verbose := flag.Bool("v", false, "Verbosity. If set, the program will print logs.")
 	explanation := flag.Bool("x", false, "Explanation. Whether or not to show all reduction steps.")
 	flag.Parse()
 
+	// Parse non-flag arguments.
 	if flag.NArg() == 0 {
 		return nil, fmt.Errorf("No input given.")
 	} else if flag.NArg() > 1 {

internal/cli/exit.go

@@ -5,6 +5,8 @@ import (
 	"os"
 )
 
+// A helper function to handle errors in the program. If it is given an error,
+// the program will exist, and print the error.
 func HandleError(err error) {
 	if err == nil {
 		return
@@ -12,4 +14,4 @@ func HandleError(err error) {
 
 	fmt.Fprintln(os.Stderr, err)
 	os.Exit(1)
-}
+}

internal/cli/read.go

@@ -1,15 +0,0 @@
-package cli
-
-import (
-	"io"
-	"os"
-)
-
-func ReadInput() (string, error) {
-	data, err := io.ReadAll(os.Stdin)
-	if err != nil {
-		return "", err
-	}
-
-	return string(data), nil
-}

internal/config/config.go

@@ -0,0 +1,11 @@
+package config
+
+// Arguments given to program.
+type Config struct {
+	// The source code given to the program.
+	Source Source
+	// Whether or not to print debug logs.
+	Verbose bool
+	// Whether or not to print an explanation of the reduction.
+	Explanation bool
+}

internal/config/get_logger.go

@@ -1,13 +1,14 @@
-package cli
+package config
 
 import (
 	"log/slog"
 	"os"
 )
 
-func GetLogger(o CLIOptions) *slog.Logger {
+// Define the correct logger for the program to use.
+func (this Config) GetLogger() *slog.Logger {
 	var level slog.Level
-	if o.Verbose {
+	if this.Verbose {
 		level = slog.LevelInfo
 	} else {
 		level = slog.LevelError

internal/config/parse_from_args.go

@@ -0,0 +1,35 @@
+package config
+
+import (
+	"flag"
+	"fmt"
+)
+
+// Extract the program configuration from the command-line arguments.
+func FromArgs() (*Config, error) {
+	// Parse flags.
+	verbose := flag.Bool("v", false, "Verbosity. If set, the program will print logs.")
+	explanation := flag.Bool("x", false, "Explanation. Whether or not to show all reduction steps.")
+	flag.Parse()
+
+	// Parse non-flag arguments.
+	if flag.NArg() == 0 {
+		return nil, fmt.Errorf("No input given.")
+	} else if flag.NArg() > 1 {
+		return nil, fmt.Errorf("More than 1 command-line argument.")
+	}
+
+	// Parse source type.
+	var source Source
+	if flag.Arg(0) == "-" {
+		source = StdinSource{}
+	} else {
+		source = StringSource{data: flag.Arg(0)}
+	}
+
+	return &Config{
+		Source:      source,
+		Verbose:     *verbose,
+		Explanation: *explanation,
+	}, nil
+}

internal/config/source.go

@@ -0,0 +1,29 @@
+package config
+
+import (
+	"io"
+	"os"
+)
+
+// Defines the consumption of different types of input sources.
+type Source interface {
+	// Get the data.
+	Pull() (string, error)
+}
+
+// A source coming from a string.
+type StringSource struct{ data string }
+
+func (this StringSource) Pull() (string, error) { return this.data, nil }
+
+// A source coming from standard input.
+type StdinSource struct{}
+
+func (this StdinSource) Pull() (string, error) {
+	data, err := io.ReadAll(os.Stdin)
+	if err != nil {
+		return "", err
+	}
+
+	return string(data), nil
+}