command.go

package main

import (
	"errors"
	"fmt"
	"os"
	"path/filepath"

	"github.com/jdkato/twine/nlp/tag"
	"github.com/pterm/pterm"

	"github.com/errata-ai/vale/v3/internal/check"
	"github.com/errata-ai/vale/v3/internal/core"
	"github.com/errata-ai/vale/v3/internal/lint"
	"github.com/errata-ai/vale/v3/internal/nlp"
)

// TaggedWord is a word with an NLP context.
type TaggedWord struct {
	Token tag.Token
	Line  int
	Span  []int
}

// CompiledRule is a rule's compiled regex.
type CompiledRule struct {
	Pattern string
}

var commandInfo = map[string]string{
	"ls-config":      "Print the current configuration to stdout.",
	"ls-metrics":     "Print the given file's internal metrics to stdout.",
	"ls-dirs":        "Print the default configuration directories to stdout.",
	"ls-vars":        "Print the supported environment variables to stdout.",
	"sync":           "Download and install external configuration sources.",
	"host-install":   "Install the Vale native messaging host for the given browser.",
	"host-uninstall": "Uninstall the Vale native messaging host for the given browser.",
	"fix":            "Attempt to automatically fix the given alert.",
}

// Actions are the available CLI commands.
var Actions = map[string]func(args []string, flags *core.CLIFlags) error{
	"ls-config":  printConfig,
	"ls-metrics": printMetrics,
	"ls-dirs":    printDirs,
	"ls-vars":    printVars,
	"sync":       sync,

	// private
	"host-install":   installNativeHost,
	"host-uninstall": uninstallNativeHost,
	"compile":        compileRule,
	"run":            runRule,
	"transform":      transform,
	"ls-path":        pathInfo,
	"fix":            fix,
	"tag":            runTag,
	"dc":             printConfig,
}

func fix(args []string, flags *core.CLIFlags) error {
	if len(args) != 1 {
		return core.NewE100("fix", errors.New("one argument expected"))
	}

	alert := args[0]
	if core.FileExists(args[0]) {
		b, err := os.ReadFile(args[0])
		if err != nil {
			return err
		}
		alert = string(b)
	}

	cfg, err := core.ReadPipeline(flags, false)
	if err != nil {
		return err
	}

	resp, err := lint.ParseAlert(alert, cfg)
	if err != nil {
		return err
	}

	return printJSON(resp)
}

func sync(_ []string, flags *core.CLIFlags) error {
	cfg, err := core.ReadPipeline(flags, true)
	if err != nil {
		return err
	} else if err = initPath(cfg); err != nil {
		return err
	}

	// NOTE: sync should *only* run for a single config file.
	rootINI, noRoot := cfg.Root()
	if noRoot != nil {
		return core.NewE100("sync", noRoot)
	}

	pkgs, err := core.GetPackages(rootINI)
	if err != nil {
		return err
	}
	stylesPath := cfg.StylesPath()

	p, err := pterm.DefaultProgressbar.WithTotal(len(pkgs)).Start()
	if err != nil {
		return err
	}

	for idx, pkg := range pkgs {
		name := fileNameWithoutExt(pkg)

		p.UpdateTitle("Syncing " + name)
		p.Increment()

		if err = readPkg(pkg, stylesPath, idx); err != nil {
			return err
		}
	}

	msg := fmt.Sprintf("Synced %d package(s) to '%s'.", len(pkgs), stylesPath)
	pterm.Success.Println(msg)

	return nil
}

func printConfig(_ []string, flags *core.CLIFlags) error {
	cfg, err := core.ReadPipeline(flags, false)
	if err != nil {
		return err
	}
	fmt.Println(cfg.String())
	return nil
}

func printMetrics(args []string, _ *core.CLIFlags) error {
	if len(args) != 1 {
		return core.NewE100("ls-metrics", errors.New("one argument expected"))
	} else if !core.FileExists(args[0]) {
		return errors.New("file not found")
	}

	cfg, err := core.NewConfig(&core.CLIFlags{})
	if err != nil {
		return err
	}

	cfg.MinAlertLevel = 0
	cfg.GBaseStyles = []string{"Vale"}
	cfg.Flags.InExt = ".txt" // default value

	linter, err := lint.NewLinter(cfg)
	if err != nil {
		return err
	}

	linted, err := linter.Lint([]string{args[0]}, "*")
	if err != nil {
		return err
	}

	computed, _ := linted[0].ComputeMetrics()
	return printJSON(computed)
}

func runTag(args []string, _ *core.CLIFlags) error {
	if len(args) != 3 {
		return core.NewE100("tag", errors.New("three arguments expected"))
	}

	text, err := os.ReadFile(args[0])
	if err != nil {
		return err
	}

	out := core.TextToContext(
		string(text), &nlp.Info{Lang: args[1], Endpoint: args[2]})

	return printJSON(out)
}

func compileRule(args []string, _ *core.CLIFlags) error {
	if len(args) != 1 {
		return core.NewE100("compile", errors.New("one argument expected"))
	}

	cfg, err := core.NewConfig(&core.CLIFlags{})
	if err != nil {
		return err
	}

	path := args[0]
	name := filepath.Base(path)

	mgr, err := check.NewManager(cfg)
	if err != nil {
		return err
	}

	err = mgr.AddRuleFromFile(name, path)
	if err != nil {
		return err
	}

	rule := CompiledRule{Pattern: mgr.Rules()[name].Pattern()}
	return printJSON(rule)
}

func runRule(args []string, _ *core.CLIFlags) error {
	if len(args) != 2 {
		return core.NewE100("run", errors.New("two arguments expected"))
	}

	cfg, err := core.NewConfig(&core.CLIFlags{})
	if err != nil {
		return err
	}

	cfg.MinAlertLevel = 0
	cfg.GBaseStyles = []string{"Test"}
	cfg.Flags.InExt = ".txt" // default value

	linter, err := lint.NewLinter(cfg)
	if err != nil {
		return err
	}

	err = linter.Manager.AddRuleFromFile("Test.Rule", args[0])
	if err != nil {
		return err
	}

	linted, err := linter.Lint([]string{args[1]}, "*")
	if err != nil {
		return err
	}

	PrintJSONAlerts(linted)
	return nil
}

func printVars(_ []string, _ *core.CLIFlags) error {
	tableData := pterm.TableData{
		{"Variable", "Description", "Value"},
	}

	for name, info := range core.ConfigVars {
		found := pterm.FgRed.Sprint("✗")
		if value, ok := os.LookupEnv(name); ok {
			found = value
		}
		tableData = append(tableData, []string{toCodeStyle(name), info, found})
	}

	return pterm.DefaultTable.WithHasHeader().WithData(tableData).Render()
}

func printDirs(_ []string, _ *core.CLIFlags) error {
	styles, _ := core.DefaultStylesPath()

	stylesFound := pterm.FgGreen.Sprint("✓")
	if !core.IsDir(styles) {
		stylesFound = pterm.FgRed.Sprint("✗")
	}

	cfg, _ := core.DefaultConfig()

	configFound := pterm.FgGreen.Sprint("✓")
	if !core.FileExists(cfg) {
		configFound = pterm.FgRed.Sprint("✗")
	}

	native, _ := getNativeConfig()
	nativeDir := filepath.Dir(native)
	nativeExe := filepath.Join(nativeDir, getExecName("vale-native"))

	nativeFound := pterm.FgGreen.Sprint("✓")
	if !core.FileExists(native) {
		nativeFound = pterm.FgRed.Sprint("✗")
	}

	tableData := pterm.TableData{
		{"Asset", "Default Location", "Found"},
		{toCodeStyle("StylesPath"), styles, stylesFound},
		{toCodeStyle(".vale.ini"), cfg, configFound},
		{toCodeStyle("vale-native"), nativeExe, nativeFound},
	}

	return pterm.DefaultTable.WithHasHeader().WithData(tableData).Render()
}

func transform(args []string, flags *core.CLIFlags) error {
	if len(args) != 1 {
		return core.NewE100("transform", errors.New("one argument expected"))
	} else if !core.FileExists(args[0]) {
		return fmt.Errorf("file not found: %s", args[0])
	}

	cfg, err := core.ReadPipeline(flags, false)
	if err != nil {
		return err
	}

	linter, err := lint.NewLinter(cfg)
	if err != nil {
		return err
	}

	f, err := core.NewFile(args[0], cfg)
	if err != nil {
		return err
	}

	out, err := linter.Transform(f)
	if err != nil {
		return err
	}

	fmt.Println(out)
	return nil
}

func pathInfo(_ []string, flags *core.CLIFlags) error {
	system := map[string][]string{}

	cfg, err := core.ReadPipeline(flags, false)
	if err != nil {
		return err
	}

	system["configs"] = append(system["configs"], cfg.ConfigFiles...)
	return printJSON(system)
}