root.go

package main

import (
	"fmt"
	"html/template"
	"os"
	"strings"
	"time"

	"github.com/jiro4989/taishoku/log"
	"github.com/spf13/cobra"
)

func init() {
	cobra.OnInitialize()
	RootCommand.PersistentFlags().BoolVarP(&log.DebugFlag, "debug", "X", false, "debug logging flag.")
	RootCommand.Flags().SortFlags = false
	RootCommand.Flags().Bool("todoke", false, "taishoku todoke")
	RootCommand.Flags().IntP("offset", "o", 3, "offset")
	RootCommand.Flags().String("format", "", "format [html]")
	RootCommand.Flags().IntP("year", "y", 2999, "year")
	RootCommand.Flags().IntP("month", "m", 12, "month")
	RootCommand.Flags().IntP("day", "d", 31, "day")
	RootCommand.Flags().StringP("department", "D", "ジョークコマンド開発部", "your department")
	RootCommand.Flags().StringP("team", "T", "第一課", "your team")
	RootCommand.Flags().StringP("your-name", "n", "真面目田マジメ", "your name")
	RootCommand.Flags().StringP("company", "C", "株式会社ジョークコマンド", "company name")
	RootCommand.Flags().StringP("president", "P", "代表取締役", "president")
	RootCommand.Flags().StringP("president-name", "N", "ジョーク山悪ふざけ太郎", "president name")
}

var RootCommand = &cobra.Command{
	Use:   "taishoku",
	Short: "退職届を出力する",
	Long: `
taishoku prints Japanese one's resignation ASCII Art.

Repository: https://github.com/jiro4989/taishoku
    Author: jiro4989
   Version: ` + Version + `

Example:

　　　代株　　　　二退こ　　　　　　
　　　表式　　　　〇職の　　　　　　
　　　取会　　　　一いた　　　　　　
　　　締社　　　　九たび　　退　　　
　　　役ジ　　　　年し一　　　　　　
　　　　ョ　　　　三た身　　職　　　
　　　ジ｜　　　　月く上　　　　　　
　　　ョク　　　　一この　　願　　　
　　　｜コ　　　　七こ都　　　　　　
　　　クマ　　　　日に合　　　　　　
　　　山ン　　　　　おに　　　　　　
　　　悪ド　　　　　願よ　　　　　　
　　　ふ　　　　　　いり　　　　　　
　　　ざ　　　　　　申、　　　　　　
　　　け　　真ジ　　し二　　　　　　
　　　太　　面ョ　　上九　　　　　　
　　　郎　　目｜　　げ九　　　　　　
　　　殿　　田ク　　ま九　　　　　　
　　　　　　マコ　　す年　　　　　　
　　　　　　ジマ　　。一　　　　　　
　　　　　　メン　　　二　　　　　　
　　　　　　　ド　　　月　　　　　　
　　　　　　印開　　　三　　　　　　
　　　　　　　発　　　一　　　　　　
　　　　　　　部　　　日　　　　　　
　　　　　　　　　　　を　私　　　　
　　　　　　　第　　　も　儀　　　　
　　　　　　　一　　　っ　　　　　　
　　　　　　　課　　　て　　　　　　

	`,
	Run: func(cmd *cobra.Command, args []string) {
		log.Debug("start 'taishoku'")
		f := cmd.Flags()

		offset, err := f.GetInt("offset")
		if err != nil {
			panic(err)
		}

		useTodoke, err := f.GetBool("todoke")
		if err != nil {
			panic(err)
		}

		year, err := f.GetInt("year")
		if err != nil {
			panic(err)
		}

		month, err := f.GetInt("month")
		if err != nil {
			panic(err)
		}

		day, err := f.GetInt("day")
		if err != nil {
			panic(err)
		}

		department, err := f.GetString("department")
		if err != nil {
			panic(err)
		}
		department = toZenkaku(department)

		team, err := f.GetString("team")
		if err != nil {
			panic(err)
		}
		team = toZenkaku(team)

		yourName, err := f.GetString("your-name")
		if err != nil {
			panic(err)
		}
		yourName = toZenkaku(yourName)

		company, err := f.GetString("company")
		if err != nil {
			panic(err)
		}
		company = toZenkaku(company)

		president, err := f.GetString("president")
		if err != nil {
			panic(err)
		}
		president = toZenkaku(president)

		presidentName, err := f.GetString("president-name")
		if err != nil {
			panic(err)
		}
		presidentName = toZenkaku(presidentName)

		formatType, err := f.GetString("format")
		if err != nil {
			panic(err)
		}
		formatType = strings.ToLower(formatType)

		log.Debug(fmt.Sprintf("command line option parameteres. "+
			"year=%d, month=%d, day=%d, "+
			"department=%s, team=%s, yourName=%s, "+
			"company=%s, president=%s, presidentName=%s",
			year, month, day,
			department, team, yourName,
			company, president, presidentName))

		tmplText := taishokuNegai
		if useTodoke {
			tmplText = taishokuTodoke
		}
		taishokuDate := fmt.Sprintf("%d年%d月%d日", year, month, day)
		taishokuDate = convertStringNumberToKanji(taishokuDate)
		today := time.Now().Format("2006年1月2日")
		today = convertStringNumberToKanji(today)

		switch formatType {
		case "html":
			tmplHtml := taishokuNegaiHTMLTemplate
			if useTodoke {
				tmplHtml = taishokuTodokeHTMLTemplate
			}
			tmplHtml = strings.Replace(tmplHtml, "ー", "｜", -1)
			t, err := template.New("template").Parse(tmplHtml)
			if err != nil {
				panic(err)
			}
			m := map[string]string{
				"taishokuDate":  taishokuDate,
				"today":         today,
				"department":    department,
				"team":          team,
				"yourName":      yourName,
				"company":       company,
				"president":     president,
				"presidentName": presidentName,
			}
			if err := t.Execute(os.Stdout, m); err != nil {
				panic(err)
			}
			// case "pdf":
		case "":
			text := makeTaishokuText(tmplText, taishokuDate, today, department, team, yourName, company, president, presidentName)
			printVertical(text, offset)
		default:
			log.Error("illegal format. format in [html]: format=" + formatType)
			os.Exit(1)
		}

		log.Debug("end 'taishoku'")
	},
}

// printVertical は横書きの文字列を縦書きにして標準出力する。
// padSizeで縦横のパディングをする。
func printVertical(text string, padSize int) {
	arr := strings.Split(text, "\n")
	s := padSpace(arr) // 文字列の長さを空白で埋めて均一にする

	// 一番上に空白をもうけたい
	for i := 0; i < padSize; i++ {
		fmt.Println()
	}

	pad := strings.Repeat("　", padSize)
	vertical := toVertical(s) // 横書きの文章を縦書きに変換
	for _, v := range vertical {
		fmt.Println(pad + v + pad)
	}

	// 一番下に空白をもうけたい
	for i := 0; i < padSize; i++ {
		fmt.Println()
	}
}