feat: statistics flag, commented some more

internal/config/config.go

@@ -1,9 +1,10 @@
 package config
 
-// Arguments given to program.
+// Configuration settings for the program.
 type Config struct {
 	Source      Source // The source code given to the program.
 	Verbose     bool   // Whether or not to print debug logs.
 	Explanation bool   // Whether or not to print an explanation of the reduction.
 	Profile     string // If not nil, print a CPU profile during execution.
+	Statistics  bool   // Whether or not to print statistics.
 }

internal/config/get_logger.go

@@ -5,13 +5,14 @@ import (
 	"os"
 )
 
-// Define the correct logger for the program to use.
+// Returns a structured logger with the appropriate configurations.
 func (c Config) GetLogger() *slog.Logger {
-	var level slog.Level
+	// By default, only print out errors.
+	level := slog.LevelError
+
+	// If the user set the output to be "VERBOSE", return the debug logs.
 	if c.Verbose {
 		level = slog.LevelInfo
-	} else {
-		level = slog.LevelError
 	}
 
 	return slog.New(

internal/config/parse_from_args.go

@@ -7,13 +7,14 @@ import (
 
 // Extract the program configuration from the command-line arguments.
 func FromArgs() (*Config, error) {
-	// Parse flags.
+	// Relevant 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.")
+	statistics := flag.Bool("s", false, "Statistics. If set, the process will print various statistics about the run.")
 	profile := flag.String("p", "", "CPU profiling. If set, the program will run a performance profile during execution.")
 	flag.Parse()
 
-	// Parse non-flag arguments.
+	// There must only be one input argument.
 	if flag.NArg() == 0 {
 		return nil, fmt.Errorf("no input given")
 	} else if flag.NArg() > 1 {
@@ -25,7 +26,7 @@ func FromArgs() (*Config, error) {
 	if flag.Arg(0) == "-" {
 		source = StdinSource{}
 	} else {
-		source = StringSource{data: flag.Arg(0)}
+		source = StringSource{Data: flag.Arg(0)}
 	}
 
 	return &Config{
@@ -33,5 +34,6 @@ func FromArgs() (*Config, error) {
 		Verbose:     *verbose,
 		Explanation: *explanation,
 		Profile:     *profile,
+		Statistics:  *statistics,
 	}, nil
 }

internal/config/source.go

@@ -5,21 +5,21 @@ import (
 	"os"
 )
 
-// Defines the consumption of different types of input sources.
+// A method of extracting input from the user.
 type Source interface {
-	// Get the data.
-	Pull() (string, error)
+	// Fetch data from this source.
+	Extract() (string, error)
 }
 
-// A source coming from a string.
-type StringSource struct{ data string }
+// A source defined by a string.
+type StringSource struct{ Data string }
 
-func (s StringSource) Pull() (string, error) { return s.data, nil }
+func (s StringSource) Extract() (string, error) { return s.Data, nil }
 
-// A source coming from standard input.
+// A source pulling from standard input.
 type StdinSource struct{}
 
-func (s StdinSource) Pull() (string, error) {
+func (s StdinSource) Extract() (string, error) {
 	data, err := io.ReadAll(os.Stdin)
 	if err != nil {
 		return "", err

internal/engine/engine.go

@@ -1,31 +1,26 @@
-package executer
+package engine
 
 import (
-	"log/slog"
-
 	"git.maximhutz.com/max/lambda/internal/config"
 	"git.maximhutz.com/max/lambda/pkg/emitter"
 	"git.maximhutz.com/max/lambda/pkg/lambda"
 )
 
-type Executor struct {
+type Engine struct {
 	Config     *config.Config
 	Expression *lambda.Expression
 	emitter.Emitter
 }
 
-func New(config *config.Config, expression *lambda.Expression) *Executor {
-	return &Executor{Config: config, Expression: expression}
+func New(config *config.Config, expression *lambda.Expression) *Engine {
+	return &Engine{Config: config, Expression: expression}
 }
 
-func (e Executor) Run() {
+func (e Engine) Run() {
 	e.Emit("start")
 
 	for lambda.ReduceOnce(e.Expression) {
 		e.Emit("step")
-		if e.Config.Verbose {
-			slog.Info("reduction", "tree", lambda.Stringify(*e.Expression))
-		}
 	}
 
 	e.Emit("end")

internal/explanation/tracker.go

@@ -3,15 +3,15 @@ package explanation
 import (
 	"fmt"
 
-	"git.maximhutz.com/max/lambda/internal/executer"
+	"git.maximhutz.com/max/lambda/internal/engine"
 	"git.maximhutz.com/max/lambda/pkg/lambda"
 )
 
 type Tracker struct {
-	process *executer.Executor
+	process *engine.Engine
 }
 
-func Track(process *executer.Executor) *Tracker {
+func Track(process *engine.Engine) *Tracker {
 	tracker := &Tracker{process: process}
 	process.On("start", tracker.Start)
 	process.On("step", tracker.Step)

internal/statistics/tracker.go

@@ -1,13 +1,14 @@
 package statistics
 
 import (
+	"fmt"
+	"os"
 	"time"
 )
 
 type Tracker struct {
-	start   time.Time
-	steps   uint64
-	Results *Results
+	start time.Time
+	steps uint64
 }
 
 func Track() *Tracker {
@@ -17,7 +18,6 @@ func Track() *Tracker {
 func (t *Tracker) Start() {
 	t.start = time.Now()
 	t.steps = 0
-	t.Results = nil
 }
 
 func (t *Tracker) Step() {
@@ -25,8 +25,10 @@ func (t *Tracker) Step() {
 }
 
 func (t *Tracker) End() {
-	t.Results = &Results{
+	results := Results{
 		StepsTaken:  t.steps,
 		TimeElapsed: uint64(time.Since(t.start).Milliseconds()),
 	}
+
+	fmt.Fprint(os.Stderr, results.String())
 }