main.go

package main

import (
	"encoding/json"
	"fmt"
	"github.com/PaluMacil/decisive-oak/analysis"
	"github.com/PaluMacil/decisive-oak/parse"
	"io/ioutil"
	"os"
	"path"
	"path/filepath"
	"strings"
)

func main() {
	files, err := filepath.Glob("data/*.data.txt")
	if err != nil {
		fmt.Printf("finding data files: %v", err)
		os.Exit(1)
	}
	for _, filename := range files {
		fmt.Println("opening", filename)
		sample, err := parse.FromFile(filename)
		if err != nil {
			fmt.Printf("parsing %s: %v", filename, err)
			os.Exit(1)
		}
		jsonData, err := json.MarshalIndent(sample, "", "  ")
		if err != nil {
			fmt.Printf("marshalling sample to JSON: %v", err)
			os.Exit(1)
		}
		outFilename := strings.TrimSuffix(filename, path.Ext(filename)) + ".json"
		outFilename = path.Join("out", filepath.Base(outFilename))
		err = ioutil.WriteFile(outFilename, jsonData, 0644)
		if err != nil {
			fmt.Printf("writing output file: %v", err)
			os.Exit(1)
		}

		rootNode, err := analysis.BuildTree(sample)
		if err != nil {
			fmt.Printf("building tree failed: %s", err.Error())
			os.Exit(1)
		}
		jsonData, err = json.MarshalIndent(rootNode, "", "  ")
		if err != nil {
			fmt.Printf("marshalling tree to JSON: %v", err)
			os.Exit(1)
		}
		treeFilename := strings.TrimSuffix(filename, path.Ext(filename)) + ".tree.json"
		treeFilename = path.Join("out", filepath.Base(treeFilename))
		err = ioutil.WriteFile(treeFilename, jsonData, 0644)
		if err != nil {
			fmt.Printf("writing tree file: %v", err)
			os.Exit(1)
		}
		fmt.Printf("Wrote %s\n\n", treeFilename)
	}
}