awz3.go

package main

import (
	"log"
	"os"

	"github.com/bmaupin/go-epub"
)

func genetateAzw3File(bookProjectDir, bookVersion, coverImagePath string) {
	genetateAzw3FileForBook(bookProjectDir, bookVersion, coverImagePath, 0)

	//genetateAzw3FileForBook(bookProjectDir, bookVersion, 1)
	//genetateAzw3FileForBook(bookProjectDir, bookVersion, 2)
}

// zero bookId means all.
func genetateAzw3FileForBook(bookProjectDir, bookVersion, coverImagePath string, bookId int) {
	var e *epub.Epub
	var outFilename string
	var indexArticleTitle string
	var bookWebsite string
	var engVersion bool
	var css string

	projectName := confirmBookProjectName(bookProjectDir)
	switch projectName {
	default:
		log.Fatal("unknow book porject: ", projectName)
	case "Go101":
		if bookId == 0 {
			e = epub.NewEpub("Go 101")
			outFilename = "Go101-" + bookVersion + ".azw3"
		} else if bookId == 1 {
			e = epub.NewEpub("Go 101 (Type System)")
			outFilename = "Go101-" + bookVersion + "-types.azw3"
		} else if bookId == 2 {
			e = epub.NewEpub("Go 101 (Extended)")
			outFilename = "Go101-" + bookVersion + "-extended.azw3"
		} else {
			log.Fatal("unknown book id: ", bookId)
		}
		e.SetAuthor("Tapir Liu")
		bookWebsite = "https://go101.org"
		engVersion = true
		indexArticleTitle = "Contents"
		css = Awz3CSS
	case "Golang101":
		if bookId == 0 {
			e = epub.NewEpub("Go语言101")
			outFilename = "Golang101-" + bookVersion + ".azw3"
		} else if bookId == 1 {
			e = epub.NewEpub("Go语言101（类型系统）")
			outFilename = "Golang101" + bookVersion + "-types.azw3"
		} else if bookId == 2 {
			e = epub.NewEpub("Go语言101（扩展阅读）")
			outFilename = "Golang101-" + bookVersion + "-extended.azw3"
		} else {
			log.Fatal("unknown book id: ", bookId)
		}
		e.SetAuthor("老貘")
		bookWebsite = "https://gfw.go101.org"
		engVersion = false
		indexArticleTitle = "目录"
		css = Awz3CSS_Chinese
	}

	cssFilename := "all.css"
	tempCssFile := mustCreateTempFile("all*.css", []byte(css))
	defer os.Remove(tempCssFile)
	cssPath, err := e.AddCSS(tempCssFile, cssFilename)
	if err != nil {
		log.Fatalln("add css", cssFilename, "failed:", err)
	}

	//tempOutFilename := outFilename + "*.epub"
	//tempOutFilename = mustCreateTempFile(tempOutFilename, nil)
	//defer os.Remove(tempOutFilename)
	tempOutFilename := outFilename + ".epub"

	writeEpub_Go101(tempOutFilename, e, bookId, bookWebsite, projectName, indexArticleTitle, bookProjectDir, coverImagePath, cssPath, "azw3", engVersion)

	runShellCommand(".", "ebook-convert", tempOutFilename, outFilename)
	runShellCommand(".", "ebook-convert", tempOutFilename, outFilename+".mobi")
	log.Println("Create", outFilename, "done!")
}