main.go

package main

import (
	"strings"

	"git.sr.ht/~flobar/apoco/cmd/align"
	"git.sr.ht/~flobar/apoco/cmd/correct"
	"git.sr.ht/~flobar/apoco/cmd/csv"
	"git.sr.ht/~flobar/apoco/cmd/eval"
	"git.sr.ht/~flobar/apoco/cmd/print"
	"git.sr.ht/~flobar/apoco/cmd/profile"
	"git.sr.ht/~flobar/apoco/cmd/train"
	"git.sr.ht/~flobar/apoco/cmd/version"
	"git.sr.ht/~flobar/apoco/pkg/apoco"
	"github.com/spf13/cobra"
)

var root = &cobra.Command{
	Use:   "apoco",
	Short: "[A]utomatic [po]st [co]rrection of (historical) OCR",
	PersistentPreRun: func(_ *cobra.Command, _ []string) {
		apoco.SetLog(strings.ToLower(logLevel) == "debug")
	},
}

var logLevel string

func init() {
	root.PersistentFlags().StringVarP(&logLevel, "log-level", "l", "INFO", "set log level")
	root.AddCommand(
		align.Cmd,
		correct.Cmd,
		csv.Cmd,
		eval.Cmd,
		print.Cmd,
		profile.Cmd,
		train.Cmd,
		version.Cmd,
	)
}

func main() {
	root.Execute()
}