generation.go

package main

import (
	"bytes"
	_ "embed"
	"text/template"

	"encoding/json"
	"fmt"
	"net/http"
	"net/url"
	"os"
	"path"
	"strings"

	"github.com/fasibio/micropuzzle/configloader"
	"github.com/urfave/cli/v2"
)

//go:embed mircopuzzle-clientlib/index.ts
var clientLibFile string

func (ru *runner) GenerateType(c *cli.Context) error {
	destination := c.String(CliGenDestination)
	sourceUrl := c.String(CliGenUrl)
	var keyList []string
	var pageList map[string]configloader.Page

	if sourceUrl == "" {
		frontends, err := configloader.LoadConfig(c.String(CliMicrofrontends))
		if err != nil {
			return err
		}
		keyList = frontends.GetKeyList()
		pageList = frontends.GetPagesList()
	} else {
		u, err := url.Parse(sourceUrl)
		if err != nil {
			return err
		}
		u.Path += path.Join(u.Path, "/frontends")
		res, err := http.Get(u.String())
		if err != nil {
			return err
		}
		var frontends FrontedsManagementResult
		json.NewDecoder(res.Body).Decode(&frontends)
		defer res.Body.Close()
		keyList = frontends.Frontends
	}

	destinationContent, err := getTypeScriptContent(keyList, pageList)
	if _, err := os.Stat(destination); os.IsNotExist(err) {
		err := os.Mkdir(destination, os.ModeDir|0755)
		if err != nil {
			return err
		}
	}
	os.Remove(fmt.Sprintf("%s/index.ts", destination))
	file, err := os.Create(fmt.Sprintf("%s/%s", destination, "index.ts"))
	if err != nil {
		return err
	}
	defer file.Close()
	file.WriteString(destinationContent)

	return nil
}

func getTypeScriptContent(keyList []string, pageList configloader.Pages) (string, error) {
	destinationContent := `/**
 * Mircopuzzle AUTO-GENERATED CODE: PLEASE DO NOT MODIFY MANUALLY
 */
	
export enum MicropuzzleFrontends {
`
	for _, key := range keyList {
		destinationContent += fmt.Sprintf("\t%s=\"%s\",\n", strings.ToUpper(strings.Replace(key, ".", "_", 1)), key)
	}
	destinationContent += "}\n\n"

	pageD, err := getPageDeclation(pageList)
	if err != nil {
		return "", err
	}
	destinationContent += pageD
	destinationContent += string(clientLibFile)
	return destinationContent, err
}

func getPageDeclation(pageList configloader.Pages) (string, error) {

	templateDesign := `
export type Page = {{.GetKeyType}};
export const pageDeclarations: PageDeclarations = {
  {{range $key, $value := .}}'{{$key}}': {
    url: '{{$value.Url}}',
    title: '{{$value.Title}}',
    fragments: { {{range $key1, $value1 := $value.Fragments}}
      '{{$key1}}': '{{$value1}}',{{end}}  
    }
  },
{{end}}
}
`

	t, err := template.New("pageDeclaration").Parse(templateDesign)
	if err != nil {
		return "", err
	}
	var tpl bytes.Buffer
	err = t.Execute(&tpl, pageList)
	if err != nil {
		return "", err
	}

	return tpl.String(), nil
}