cmd-print.go

package main

import (
	. "github.com/spf13/cobra"
	"html"
	"io/ioutil"
	"os"
	"os/exec"
	"regexp"
	"strconv"
)

func cmdPrint(rootCmd *Command) {
	rootCmd.AddCommand(&Command{
		Use:   "print [item id]",
		Short: "Print out item",
		Long:  "Print out item",
		Run: func(cmd *Command, args []string) {
			if len(args) != 1 {
				fatal("item id is required")
			}

			itemId, err := strconv.Atoi(args[0])
			failOnError(err)

			item, err := itemRepository.ItemOfId(itemId)
			failOnError(err)

			tempDir, err := ioutil.TempDir("/tmp", "kobito")
			failOnError(err)

			// HTML に印刷JSを埋め込む
			itemHtml := item.Html + `<script>window.print();</script>`

			// タイトルを変更する
			itemHtml = regexp.MustCompile("<title>.*</title>").ReplaceAllString(itemHtml, "<title>"+html.EscapeString(item.Title)+"</title>")

			// HTML を書き込む
			ioutil.WriteFile(tempDir+"/index.html", []byte(itemHtml), os.ModePerm)

			// CSS などをコピーする
			assets := []string{
				"highlight.pack.js",
				"github.min.css",
				"markdown.css",
			}

			kobitoPath, err := kobitoPath()
			failOnError(err)

			for _, asset := range assets {
				err := os.Symlink(kobitoPath+"/Contents/Resources/"+asset, tempDir+"/"+asset)
				failOnError(err)
			}

			// ブラウザで開く
			_, err = exec.Command("open", "-a", "Safari", tempDir+"/index.html").Output()
			failOnError(err)
		},
	})
}