generate unions (#505)

* generate unions

* drop unused TagOwner enums

* union tpl clean up

check.go

@@ -10,11 +10,6 @@ import (
 	"github.com/relvacode/iso8601"
 )
 
-type CheckOwner struct {
-	Team TeamId `graphql:"... on Team"`
-	// User User `graphql:"... on User"` // TODO: will this be public?
-}
-
 type CheckInputConstructor func() any
 
 var CheckCreateConstructors = map[CheckType]CheckInputConstructor{

document.go

@@ -1,10 +1,5 @@
 package opslevel
 
-type ServiceDocumentSource struct {
-	IntegrationId     `graphql:"... on ApiDocIntegration"`
-	ServiceRepository `graphql:"... on ServiceRepository"`
-}
-
 type ServiceDocument struct {
 	Id         ID                    `graphql:"id" json:"id"`
 	HtmlURL    string                `graphql:"htmlUrl" json:"htmUrl,omitempty"`

gen.go

@@ -32,7 +32,6 @@ const (
 	mutationFile   string = "mutation.go"
 	payloadFile    string = "payload.go"
 	// scalarFile      string = "scalar.go" // NOTE: probably not useful
-	// unionFile       string = "union.go" // NOTE: probably not useful
 )
 
 var knownTypeIsName = []string{
@@ -285,6 +284,7 @@ func main() {
 			panic(fmt.Errorf("Unknown GraphQL type: %v", v))
 		}
 	}
+	genUnions(unions)
 	genObjects(objects)
 	genEnums(schemaAst.Enums)
 	genInputObjects(inputObjects)
@@ -310,6 +310,28 @@ func sortedMapKeys[T any](schemaMap map[string]T) []string {
 	return sortedNames
 }
 
+func genUnions(unions map[string]*types.Union) {
+	var buf bytes.Buffer
+
+	buf.WriteString(header + "\n\n")
+
+	tmpl := template.New("unions")
+	tmpl.Funcs(sprig.TxtFuncMap())
+	tmpl.Funcs(templFuncMap)
+	template.Must(tmpl.ParseFiles("./templates/unions.tpl"))
+	for _, union := range sortedMapKeys(unions) {
+		if err := tmpl.ExecuteTemplate(&buf, "union", unions[union]); err != nil {
+			panic(err)
+		}
+	}
+
+	fmt.Println("writing union.go")
+	err := os.WriteFile("union.go", buf.Bytes(), 0o644)
+	if err != nil {
+		panic(err)
+	}
+}
+
 func genEnums(schemaEnums []*types.EnumTypeDefinition) {
 	var buf bytes.Buffer
 
@@ -472,8 +494,6 @@ func run() error {
 			subSchema = objectSchema
 		// case scalarFile:
 		// 	subSchema = scalarSchema
-		// case unionFile:
-		// 	subSchema = unionSchema
 		default:
 			panic("Unknown file: " + filename)
 		}
@@ -867,21 +887,6 @@ var templates = map[string]*template.Template{
 	// func NewString(value string) *string {
 	// 	return &value
 	// }`),
-	// 	unionFile: t(header + `
-	// {{range .Types | sortByName}}{{if and (eq .Kind "UNION") (not (internal .Name))}}
-	// {{template "union_object" .}}
-	// {{end}}{{end}}
-
-	// {{- define "union_object" -}}
-	// // Union{{.Name}} {{ template "description" . }}
-	// type Union{{.Name}} interface { {{range .PossibleTypes }}
-	//
-	//	    {{.Name}}Fragment() {{.Name}}Fragment{{end}}
-	//	}
-	//
-	// {{- end -}}
-	//
-	//	`),
 }
 
 func t(text string) *template.Template {
@@ -1252,7 +1257,6 @@ func firstCharLowered(s string) string {
 var templFuncMap = template.FuncMap{
 	"internal":                            func(s string) bool { return strings.HasPrefix(s, "__") },
 	"quote":                               strconv.Quote,
-	"join":                                strings.Join,
 	"check_fragments":                     fragmentsForCheck,
 	"custom_actions_ext_action_fragments": fragmentsForCustomActionsExtAction,
 	"integration_fragments":               fragmentsForIntegration,

owner.go

@@ -5,10 +5,6 @@ type EntityOwnerTeam struct {
 	Id    ID     `json:"id"`
 }
 
-type EntityOwner struct {
-	OnTeam EntityOwnerTeam `graphql:"... on Team"`
-}
-
 func (entityOwner *EntityOwner) Alias() string {
 	return entityOwner.OnTeam.Alias
 }

tags.go

@@ -7,13 +7,6 @@ import (
 	"slices"
 )
 
-type TagOwner string
-
-const (
-	TagOwnerService    TagOwner = "Service"
-	TagOwnerRepository TagOwner = "Repository"
-)
-
 type TaggableResourceInterface interface {
 	GetTags(*Client, *PayloadVariables) (*TagConnection, error)
 	ResourceId() ID

templates/unions.tpl

@@ -0,0 +1,40 @@
+{{- define "union" }}
+// {{ title .Name }} {{.Desc | clean | endSentence}}
+  {{- if eq "CheckOwner" .Name }}
+    {{ template "check_owner" }}
+  {{- else if eq "EntityOwner" .Name }}
+    {{ template "entity_owner" }}
+  {{- else if eq "ServiceDocumentSource" .Name }}
+    {{ template "service_document_source" }}
+  {{- else }}
+type {{.Name}} struct {
+ {{ range .TypeNames }}
+    {{- if not (contains "Group" . ) }}
+  {{.}} {{. -}}
+      {{- if not (contains . (list "ApiDocIntegration" "InfrastructureResource" "ServiceRepository" | join " " )) -}}Id
+      {{- end }} `graphql:"... on {{.}}"`
+    {{- end }}
+  {{- end }}
+}
+ {{- end }}
+{{- end -}}
+
+{{- define "check_owner" -}}
+type CheckOwner struct {
+	Team TeamId `graphql:"... on Team"`
+	// User UserId `graphql:"... on User"` // TODO: will this be public?
+}
+{{ end }}
+
+{{- define "entity_owner" -}}
+type EntityOwner struct {
+	OnTeam EntityOwnerTeam `graphql:"... on Team"`
+}
+{{ end }}
+
+{{- define "service_document_source" -}}
+type ServiceDocumentSource struct {
+	IntegrationId     `graphql:"... on ApiDocIntegration"`
+	ServiceRepository `graphql:"... on ServiceRepository"`
+}
+{{ end }}