tonel.go

package main

import (
	"bytes"
	"fmt"
	"os"
	"path/filepath"
)

type tonelWriter struct {
	buff  *bytes.Buffer
	model *YamlModel
}

func writeTonelFiles(args *args) {

	model, err := ReadYamlModel(args)
	check(err, "could not read YAML model")

	outDir := filepath.Join(args.home, "build", "tonel")
	mkdir(outDir)

	model.EachClass(func(class *YamlClass) {
		var buffer bytes.Buffer
		writer := tonelWriter{
			buff:  &buffer,
			model: model,
		}
		writer.writeClass(class)

		outFile := filepath.Join(outDir, "Lca"+class.Name+".class.st")
		os.WriteFile(outFile, buffer.Bytes(), os.ModePerm)

	})

}

func (w *tonelWriter) writeClass(class *YamlClass) {

	className := "Lca" + class.Name

	// class declaration
	w.writeln("Class {")
	w.writeln("\t#name : #", className, ",")
	var super string
	if class.SuperClass != "" {
		super = "Lca" + class.SuperClass
	} else {
		super = "LcaEntity"
	}
	w.writeln("\t#superclass : #", super, ",")
	w.writeln("\t#instVars : [")
	for i, prop := range class.Props {
		propName := w.propNameOf(prop)
		if propName == "" {
			continue
		}
		if i < (len(class.Props) - 1) {
			w.writeln("\t\t'", propName, "',")
		} else {
			w.writeln("\t\t'", propName, "'")
		}
	}

	w.writeln("\t],")
	w.writeln("\t#category : #'openLCA-Model'")
	w.writeln("}")

	// accessors
	for _, prop := range class.Props {
		propName := w.propNameOf(prop)
		if propName == "" {
			continue
		}

		w.writeln()

		// getter
		w.writeln("{ #category : #accessing }")
		w.writeln(className, " >> ", propName, " [")
		w.writeln()
		w.writeln("\t^ ", propName)
		w.writeln("]")
		w.writeln()

		// setter
		typeHint := w.typeHintOf(prop)
		w.writeln("{ #category : #accessing }")
		w.writeln(className, " >> ", propName, ": ", typeHint, " [")
		w.writeln()
		w.writeln("\t", propName, " := ", typeHint)
		w.writeln("]")
	}
}

func (w *tonelWriter) propNameOf(prop *YamlProp) string {
	switch prop.Name {
	case "@type":
		return ""
	case "@id":
		return "id"
	default:
		return prop.Name
	}
}

func (w *tonelWriter) typeHintOf(prop *YamlProp) string {

	propType := prop.PropType()
	if propType.IsList() {
		return "aCollection"
	}
	if propType.IsRef() {
		return "aRef"
	}
	if !propType.IsPrimitive() {
		b0 := prop.Type[0]
		prefix := "a"
		switch b0 {
		case 'A', 'E', 'I', 'O', 'U':
			prefix = "an"
		}
		return prefix + prop.Type
	}

	switch prop.Type {
	case "string":
		return "aString"
	case "int", "integer":
		return "anInteger"
	case "double":
		return "aNumber"
	case "bool", "boolean":
		return "aBoolean"
	case "date":
		return "aDateString"
	case "dateTime":
		return "aDateTimeString"
	default:
		fmt.Println("warning: could provide a better type hint for:" + prop.Type)
		return "anObject"
	}
}

func (w *tonelWriter) writeln(xs ...string) {
	for _, x := range xs {
		w.buff.WriteString(x)
	}
	w.buff.WriteRune('\n')
}