122 行
3.0 KiB
Go
122 行
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"github.com/cjslep/activity/tools/exp/convert"
|
|
"github.com/cjslep/activity/tools/exp/gen"
|
|
"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/rfc"
|
|
"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{})
|
|
mustAddOntology(&rfc.RFCOntology{Package: "rfc"})
|
|
}
|
|
|
|
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
|
|
}
|
|
|
|
type CommandLineFlags struct {
|
|
specs list
|
|
prefix *string
|
|
individual *bool
|
|
}
|
|
|
|
func NewCommandLineFlags() *CommandLineFlags {
|
|
c := &CommandLineFlags{
|
|
// TODO: Be more rigorous when applying this. Also, clear the default value I am using for convenience.
|
|
prefix: flag.String("prefix", "github.com/cjslep/activity/tools/exp/tmp", "Package prefix to use for all generated package paths. This should be the prefix in the GOPATH directory if generating in a subdirectory."),
|
|
individual: flag.Bool("individual", false, "Whether to generate types and properties in individual packages."),
|
|
}
|
|
flag.Var(&(c.specs), "spec", "Input JSON-LD specification used to generate Go code.")
|
|
flag.Parse()
|
|
if err := c.validate(); err != nil {
|
|
panic(err)
|
|
}
|
|
return c
|
|
}
|
|
|
|
func (c *CommandLineFlags) validate() error {
|
|
if len(c.specs) == 0 {
|
|
return fmt.Errorf("specs must not be empty")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
cmd := NewCommandLineFlags()
|
|
|
|
inputJSONs := make([]rdf.JSONLD, 0, len(cmd.specs))
|
|
for _, spec := range cmd.specs {
|
|
b, err := ioutil.ReadFile(spec)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var inputJSON map[string]interface{}
|
|
err = json.Unmarshal(b, &inputJSON)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
inputJSONs = append(inputJSONs, inputJSON)
|
|
}
|
|
p, err := rdf.ParseVocabularies(registry, inputJSONs)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
policy := convert.FlatUnderRoot
|
|
if *cmd.individual {
|
|
policy = convert.IndividualUnderRoot
|
|
}
|
|
c := &convert.Converter{
|
|
Registry: registry,
|
|
GenRoot: gen.NewPackageManager(*cmd.prefix, "gen"),
|
|
PackagePolicy: policy,
|
|
}
|
|
f, err := c.Convert(p)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
for _, file := range f {
|
|
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\n")
|
|
}
|