package main

import (
	"encoding/json"
	"flag"
	"fmt"
	"github.com/cjslep/activity/tools/exp/convert"
	"github.com/cjslep/activity/tools/exp/rdf"
	"github.com/cjslep/activity/tools/exp/rdf/owl"
	"github.com/cjslep/activity/tools/exp/rdf/rdfs"
	"github.com/cjslep/activity/tools/exp/rdf/schema"
	"github.com/cjslep/activity/tools/exp/rdf/xsd"
	"io/ioutil"
	"os"
	"strings"
)

var registry *rdf.RDFRegistry

func mustAddOntology(o rdf.Ontology) {
	if registry == nil {
		registry = rdf.NewRDFRegistry()
	}
	if err := registry.AddOntology(o); err != nil {
		panic(err)
	}
}

func init() {
	mustAddOntology(&xsd.XMLOntology{Package: "xml"})
	mustAddOntology(&owl.OWLOntology{})
	mustAddOntology(&rdf.RDFOntology{Package: "rdf"})
	mustAddOntology(&rdfs.RDFSchemaOntology{})
	mustAddOntology(&schema.SchemaOntology{})
}

var (
	input = flag.String("input", "spec.json", "Input JSON-LD specification used to generate Go code.")
	// TODO: Use this flag
	root   = flag.String("root", "github.com/go-fed/activity/", "Go import path prefix for generated packages")
	xmlpkg = flag.String("xmlpkg", "github.com/go-fed/activity/tools/exp/ref/xml", "Go package location for known XML references")
	rdfpkg = flag.String("rdfpkg", "github.com/go-fed/activity/tools/exp/ref/rdf", "Go package location for known RDF references")
	// TODO: Use this flag
	writeWellKnown = flag.Bool("write-well-known", false, "When true, also outputs well-known specifications to './ref' subdirectories (ex: XML, RDF)")
)

type list []string

func (l *list) String() string {
	return strings.Join(*l, ",")
}

func (l *list) Set(v string) error {
	vals := strings.Split(v, ",")
	*l = append(*l, vals...)
	return nil
}

func main() {
	var ref list
	var refspec list
	var refpkg list
	flag.Var(&ref, "ref", "Input JSON-LD specification of other referenced specifications. Must be the same size and in the same order as the 'refspec' and 'refpkg' flags")
	flag.Var(&refspec, "refspec", "Base URL for other referenced specifications. Must be the same size and in the same order as the 'ref' and 'refpkg' flags")
	flag.Var(&refpkg, "refpkg", "Golang package location for other referenced specifications. Must be the same size and in the same order as the 'ref' and 'refspec' flags")
	flag.Parse()
	// TODO: Flag validation

	b, err := ioutil.ReadFile(*input)
	if err != nil {
		panic(err)
	}
	var inputJSON map[string]interface{}
	err = json.Unmarshal(b, &inputJSON)
	if err != nil {
		panic(err)
	}
	p, err := rdf.ParseVocabulary(registry, inputJSON)
	if err != nil {
		panic(err)
	}
	c := &convert.Converter{
		Registry:              registry,
		VocabularyRoot:        "as",
		PropertyPackagePolicy: convert.PropertyFlatUnderRoot,
		PropertyPackageRoot:   "props",
		TypePackagePolicy:     convert.TypeFlatUnderRoot,
		TypePackageRoot:       "types",
	}
	f, err := c.Convert(p)
	if err != nil {
		panic(err)
	}
	for _, file := range f {
		file.F.ImportName(*xmlpkg, "xml")
		file.F.ImportName(*rdfpkg, "rdf")
		if e := os.MkdirAll("./"+file.Directory, 0777); e != nil {
			panic(e)
		}
		if e := file.F.Save("./" + file.Directory + file.FileName); e != nil {
			panic(e)
		}
	}
	fmt.Printf("done")
	// fmt.Printf("done\n%s\n", p)
}