main.go

package main

import (
	"fmt"
	"github.com/opencontrol/fedramp-templater/opencontrols"
	"github.com/opencontrol/fedramp-templater/ssp"
	"github.com/opencontrol/fedramp-templater/templater"
	"log"
	"os"
	"path/filepath"
)

type subCommand uint8

const (
	_ subCommand = iota // Default value. This value is used as a placeholder when creating instances of subCommands
	diff
	fill
)

func (cmd subCommand) isType(otherCmd subCommand) bool {
	return cmd == otherCmd
}

type options struct {
	openControlsDir string
	inputPath       string
	outputPath      string
	cmd             subCommand
}

func printUsage() {
	log.Fatal(`Usage:
	fedramp-templater fill <optional_opencontrols_dir> <inputDoc> <outputDoc>

	or

	fedramp-templater diff <optional_opencontrols_dir> <inputDoc>`)
}

func parseArgs() (opts options) {
	if len(os.Args) < 4 || len(os.Args) > 5 {
		printUsage()
	}

	switch os.Args[1] {
	case "diff":
		opts.cmd = diff
	case "fill":
		opts.cmd = fill
	default:
		log.Printf("Unknown command: %s\n", os.Args[1])
		printUsage()
	}
	if opts.cmd.isType(diff) && len(os.Args) == 4 || len(os.Args) == 3 {
		// diff command only has three or four args
		if len(os.Args) == 4 {
			opts.openControlsDir = os.Args[2]
			opts.inputPath = os.Args[3]
		} else {
			dirname := "opencontrols"
			if _, err := os.Stat(dirname); os.IsNotExist(err) {
				fmt.Println("OpenControls directory not found in current directory. Please specify opencontrols directory")
			} else {
				opts.openControlsDir = dirname
				opts.inputPath = os.Args[2]
			}

		}

	} else if opts.cmd.isType(fill) && len(os.Args) == 5 || len(os.Args) == 4 {
		// fill command only has four or five args
		if len(os.Args) == 5 {
			opts.openControlsDir = os.Args[2]
			opts.inputPath = os.Args[3]
			opts.outputPath = os.Args[4]
		} else {
			dirname := "opencontrols"
			if _, err := os.Stat(dirname); os.IsNotExist(err) {
				fmt.Println("OpenControls directory not found in current directory. Please specify opencontrols directory")
			} else {
				opts.openControlsDir = dirname
				opts.inputPath = os.Args[2]
				opts.outputPath = os.Args[3]
			}

		}

	} else {
		printUsage()
	}
	return
}

func loadOpenControls(path string) opencontrols.Data {
	path, err := filepath.Abs(path)
	if err != nil {
		log.Fatalln(err)
	}

	openControlData, errors := opencontrols.LoadFrom(path)
	if len(errors) > 0 {
		log.Fatal(errors)
	}
	return openControlData
}

func diffCmd(openControlData opencontrols.Data, doc *ssp.Document) {
	reporters, err := templater.DiffSSP(doc, openControlData)
	if err != nil {
		log.Fatalln(err)
	}
	if len(reporters) == 0 {
		log.Println("No diff detected")
		return
	}
	for _, reporter := range reporters {
		reporter.WriteTextTo(os.Stdout)
	}
	log.Fatalf("%d diffs detected\n", len(reporters))
}

func fillCmd(openControlData opencontrols.Data, doc *ssp.Document, opts options) {
	err := templater.TemplatizeSSP(doc, openControlData)
	if err != nil {
		log.Fatalln(err)
	}

	outputDir := filepath.Dir(opts.outputPath)
	err = os.MkdirAll(outputDir, 0755)
	if err != nil {
		log.Fatalln(err)
	}

	err = doc.CopyTo(opts.outputPath)
	if err != nil {
		log.Fatalln(err)
	}
}

func main() {
	opts := parseArgs()

	openControlData := loadOpenControls(opts.openControlsDir)
	doc, err := ssp.Load(opts.inputPath)
	if err != nil {
		log.Fatalln(err)
	}
	defer doc.Close()

	// right now we don't want to do a fill and diff together.
	if opts.cmd.isType(diff) {
		diffCmd(openControlData, doc)

	} else if opts.cmd.isType(fill) {
		fillCmd(openControlData, doc, opts)
	}
}