gocontentful/main.go

package main

import (
	"context"
	"encoding/json"
	"flag"
	"fmt"
	"os"
	"os/user"
	"path/filepath"
	"regexp"
	"strings"

	"github.com/foomo/gocontentful/config"
	"github.com/foomo/gocontentful/erm"
)

var VERSION = "latest"

type contentfulRc struct {
	ManagementToken string `json:"managementToken"`
}

var Usage = func() {
	fmt.Printf("\nSYNOPSIS\n")
	fmt.Printf("     gocontentful -spaceid SpaceID -cmakey CMAKey [-contenttypes firsttype,secondtype...lasttype] path/to/target/package\n\n")
	flag.PrintDefaults()
	fmt.Printf("\nNotes:\n")
	fmt.Println("- The last segment of the path/to/target/package will be used as package name")
	fmt.Println("- The -cmakey parameter can be omitted if you logged in with the Contentful CLI")
	fmt.Println()
}

func usageError(comment string) {
	fmt.Println("ERROR:", comment)
	Usage()
	os.Exit(1)
}

func fatal(infos ...interface{}) {
	fmt.Println(infos...)
	os.Exit(1)
}

func getCmaKeyFromRcFile() string {
	currentUser, errGetUser := user.Current()
	if errGetUser != nil {
		return ""
	}
	contentfulRcBytes, errReadFile := os.ReadFile(currentUser.HomeDir + "/.contentfulrc.json")
	if errReadFile != nil {
		return ""
	}
	var contentfulConfig contentfulRc
	errUnmarshal := json.Unmarshal(contentfulRcBytes, &contentfulConfig)
	if errUnmarshal != nil {
		return ""
	}
	return contentfulConfig.ManagementToken
}

func main() {
	// Get parameters from cmd line flags
	flagConfigFile := flag.String("configfile", "", "Full path to configuration file")
	flagSpaceID := flag.String("spaceid", "", "Contentful space ID")
	flagCMAKey := flag.String("cmakey", "", "[Optional] Contentful CMA key")
	flagEnvironment := flag.String("environment", "", "[Optional] Contentful space environment")
	flagGenerateFromExport := flag.String("exportfile", "", "Space export file to generate the API from")
	flagContentTypes := flag.String("contenttypes", "", "[Optional] Content type IDs to parse, comma separated")
	flagVersion := flag.Bool("version", false, "Print version and exit")
	flagHelp := flag.Bool("help", false, "Print version and exit")
	flag.Parse()

	if *flagVersion {
		fmt.Println(VERSION)
		os.Exit(0)
	}

	if *flagHelp {
		Usage()
		os.Exit(0)
	}
	var conf *config.Config
	var err error
	if *flagConfigFile != "" {
		conf, err = config.LoadConfigFromYAML(*flagConfigFile)
		if err != nil {
			fatal(err)
		}
		if conf.RequireVersion != "" && conf.RequireVersion != VERSION && conf.RequireVersion != strings.Trim(VERSION, "v") {
			fatal("Required version mismatch. Want: " + conf.RequireVersion + "  Have: " + VERSION)
		}
	} else {
		conf = &config.Config{
			SpaceID:     *flagSpaceID,
			Environment: *flagEnvironment,
			ExportFile:  *flagGenerateFromExport,
		}
		if *flagContentTypes != "" {
			conf.ContentTypes = strings.Split(*flagContentTypes, ",")
		}
	}
	cmaKey := *flagCMAKey
	if cmaKey == "" && *flagGenerateFromExport == "" {
		cmaKey = getCmaKeyFromRcFile()
	}
	if conf.ExportFile == "" && conf.SpaceID == "" ||
		conf.ExportFile != "" && conf.SpaceID != "" {
		byt, errMarshal := json.MarshalIndent(conf, "", "  ")
		if errMarshal != nil {
			fatal(errMarshal)
		}
		fmt.Println(string(byt))
		usageError("Please provide either a Contentful Space ID and CMA access token or an export file name")
	}
	var path string
	if len(flag.Args()) != 1 && conf.PathTargetPackage == "" {
		usageError("Missing arg path/to/target/package")
	}
	if conf.PathTargetPackage != "" {
		path = conf.PathTargetPackage
	} else {
		path = flag.Arg(0)
	}
	packageName := filepath.Base(path)
	fmt.Println("output path:", path)
	fmt.Println("packageName:", packageName)

	matched, err := regexp.MatchString(`[a-z].{2,}`, packageName)
	if !matched || err != nil {
		usageError("Please specify the package name correctly (only small caps letters)")
	}

	fmt.Printf("Contentful API Generator %s starting...\n\n", VERSION)

	var cleanContentTypes []string
	if len(conf.ContentTypes) > 0 {
		for _, contentType := range conf.ContentTypes {
			cleanContentTypes = append(cleanContentTypes, strings.TrimSpace(contentType))
		}
	}

	err = erm.GenerateAPI(context.Background(), filepath.Dir(path), packageName, conf.SpaceID, cmaKey, conf.Environment, conf.ExportFile, cleanContentTypes, VERSION)
	if err != nil {
		fatal("Something went horribly wrong...", err)
	}
	fmt.Println("ALL DONE!")
}