license/license.go

package main

import (
	"embed"
	"errors"
	"fmt"
	"path/filepath"
	"strings"

	"github.com/make-go-great/ioe-go"
)

const (
	templatesPath = "templates"
)

var ErrInvalidLicense = errors.New("invalid license")

//go:embed templates/*
var embedFS embed.FS

// Map template name with filename and args
// always use upercase for license name
var templates = map[string]templateInfo{
	"MIT": {
		templateFilename: "mit.txt",
		licenseFilename:  "LICENSE",
		args: []string{
			"[year]",
			"[fullname]",
		},
	},
	"GNU GPLv3": {
		templateFilename: "gnu_gplv3.txt",
		licenseFilename:  "COPYING",
	},
	"Apache License 2.0": {
		templateFilename: "apache_2.0.txt",
		licenseFilename:  "LICENSE",
		args: []string{
			"[yyyy]",
			"[name of copyright owner]",
		},
	},
}

type templateInfo struct {
	templateFilename string
	licenseFilename  string
	args             []string
}

func generateLicense(name string) (string, string, error) {
	if name == "" {
		return "", "", fmt.Errorf("empty license name: %w", ErrInvalidLicense)
	}

	isSupportTemplate := false
	var templateInfo templateInfo
	for templateName := range templates {
		if strings.EqualFold(templateName, name) {
			isSupportTemplate = true
			templateInfo = templates[templateName]
		}
	}

	if !isSupportTemplate {
		return "", "", fmt.Errorf("not support license %s: %w", name, ErrInvalidLicense)
	}

	// Get correct path of license
	path := filepath.Join(templatesPath, templateInfo.templateFilename)

	// Read template
	templateRaw, err := embedFS.ReadFile(path)
	if err != nil {
		return "", "", fmt.Errorf("failed to read file %s: %w", path, err)
	}

	// Replace template info args
	licenseData := string(templateRaw)
	for _, arg := range templateInfo.args {
		fmt.Printf("What is your %s: ", arg)
		value := ioe.ReadInput()

		licenseData = strings.ReplaceAll(licenseData, arg, value)
	}

	return licenseData, templateInfo.licenseFilename, nil
}