config.go

package main

import (
	"fmt"
	"os"
	"runtime"

	"github.com/cosiner/flag"
)

const (
	defaultRawFilePattern = ".(3fr|ari|arw|bay|crw|cr2|cr3|cap|data|dcs|dcr|drf|eip|erf|fff|gpr|iiq|k25|kdc|mdc|mef|mos|mrw|nef|nrw|obm|orf|pef|ptx|pxn|r3d|raf|raw|rwl|rw2|rwz|sr2|srf|srw|x3f)$"
)

type config struct {
	Concurrency          int    `names:"--concurrency, -c" usage:"Amount of routines to spawn at the same time for checksum verification"`
	Complete             bool   `names:"--complete" usage:"Completion for shell"`
	DataDir              string `names:"--datadir, -d" usage:"Data directory path to run new files scan"`
	DbPath               string `names:"--database, -D" usage:"Database file path (required)"`
	DeleteMissed         bool   `names:"--delete-missed" usage:"Delete missed files from database"`
	GenerateChecksumOnly bool   `names:"--generate-checksums-only" usage:"Skip verification step and add new files only"`
	Pattern              string `names:"--pattern, -p" usage:"Pattern to match filenames which checking for new files"`
	SkipFailed           bool   `names:"--skip-failed, --sf" usage:"Skip FAIL verification results from output"`
	SkipMissed           bool   `names:"--skip-missed, --sm" usage:"Skip MISS verification results from output"`
	SkipOk               bool   `names:"--skip-ok, --so" usage:"Skip OK verification results from output"`
	Progressbar          bool   `names:"--progressbar" usage:"Show progress bar instead of printing handled files"`
	Version              bool   `names:"--version, -V" usage:"Print application and Golang versions"`
}

func newConfig() *config {
	var c config

	set := flag.NewFlagSet(flag.Flag{})
	set.ParseStruct(&c, os.Args...)

	if c.Version == true {
		printVersion()
		os.Exit(0)
	}

	if c.DbPath == "" {
		set.Help()
		os.Exit(1)
	}

	if c.Concurrency == 0 {
		c.Concurrency = runtime.NumCPU()
	}

	if c.Pattern == "" {
		c.Pattern = defaultRawFilePattern
	}

	return &c
}

func (c *config) Metadata() map[string]flag.Flag {
	var (
		usage = `Utility to verify files consistency with length, SHA1 and SHA256`

		version = fmt.Sprintf(`
			version: %s
			built with: %s
			built at: %s
		`, appVersion, runtime.Version(), buildTimestamp)

		desc = `
		checksum creates database (actually just a JSON file) to store file length, SHA1, SHA256
		to verify file consistency and report if something goes wrong.
		`
	)
	return map[string]flag.Flag{
		"": {
			Usage:   usage,
			Version: version,
			Desc:    desc,
		},
		"--concurrency": {
			Desc: fmt.Sprintf("Default value is %d for your system", runtime.NumCPU()),
		},
		"--pattern": {
			Desc: fmt.Sprintf("Default is `%s`", defaultRawFilePattern),
		},
	}
}