fusions_page_create.go

package fusions_page_create

import (
	"encoding/json"
	"fmt"
	"html/template"
	"io/ioutil"
	"os"
	"strings"

	"github.com/juju/errors"
	"github.com/raid-codex/tools/common"
	"github.com/raid-codex/tools/templatefuncs"
	"github.com/raid-codex/tools/utils"
	"github.com/raid-codex/tools/utils/wp"
	"gopkg.in/alecthomas/kingpin.v2"
)

type Command struct {
	FusionFile     *string
	TemplateFolder *string
	DataDirectory  *string
}

func New(cmd *kingpin.CmdClause) *Command {
	return &Command{
		FusionFile:     cmd.Flag("fusion-file", "Filename for the fusion").Required().String(),
		TemplateFolder: cmd.Flag("template-folder", "Template folder").Required().String(),
		DataDirectory:  cmd.Flag("data-directory", "Data directory").Required().String(),
	}
}

func (c *Command) Run() {
	client := wp.GetWPClient()

	errFactory := common.InitFactory(*c.DataDirectory)
	if errFactory != nil {
		utils.Exit(1, errFactory)
	}
	tmpl, errTmpl := c.loadTemplates()
	if errTmpl != nil {
		utils.Exit(1, errTmpl)
	}
	fusion, errFusion := c.getFusion()
	if errFusion != nil {
		utils.Exit(1, errFusion)
	} else if fusion.Slug != fmt.Sprintf("fusion-%s", fusion.ChampionSlug) {
		utils.Exit(0, fmt.Errorf("skipping, since it's a child of %s", strings.Replace(fusion.Slug, fmt.Sprintf("-%s", fusion.ChampionSlug), "", -1)))
	}
	page, errPage := wp.GetPageFromSlug(client, fusion.GetPageSlug())
	if errPage != nil && !errors.IsNotFound(errPage) {
		utils.Exit(1, errPage)
	} else if errPage != nil && errors.IsNotFound(errPage) {
		errCreate := wp.CreatePage(client, fusion, *c.TemplateFolder, *c.DataDirectory, tmpl)
		if errCreate != nil {
			utils.Exit(1, errCreate)
		}
	} else {
		errUpdate := wp.UpdatePage(client, page, fusion, *c.TemplateFolder, *c.DataDirectory, tmpl)
		if errUpdate != nil {
			utils.Exit(1, errUpdate)
		}
	}
}

func (c *Command) loadTemplates() (*template.Template, error) {
	files, errFiles := ioutil.ReadDir(*c.TemplateFolder)
	if errFiles != nil {
		return nil, errFiles
	}
	templateFiles := make([]string, 0)
	for _, file := range files {
		templateFiles = append(templateFiles, fmt.Sprintf("%s/%s", *c.TemplateFolder, file.Name()))
	}
	return template.New("main.html").Funcs(templatefuncs.FuncMap).ParseFiles(templateFiles...)
}

func (c *Command) getFusion() (*common.Fusion, error) {
	file, errFile := os.Open(*c.FusionFile)
	if errFile != nil {
		return nil, errors.Annotate(errFile, "cannot open file")
	}
	defer file.Close()

	var fusion common.Fusion
	errJSON := json.NewDecoder(file).Decode(&fusion)
	if errJSON != nil {
		return nil, errors.Annotate(errJSON, "cannot unmarshal file")
	}
	return &fusion, nil
}