golang101/golang101-embed.go

//go:build go1.16
// +build go1.16

package main

import (
	"embed"
	//"errors"
	"fmt"
	"html/template"
	"io/fs"
	"log"
	"net/http"
	"os"
	"path"
	"path/filepath"
	"time"
)

//go:embed web
//go:embed pages
var allFiles embed.FS

var staticFilesHandler = func() http.Handler {
	if wdIsGo101ProjectRoot {
		return staticFilesHandler_NonEmbedding
	}

	staticFiles, err := fs.Sub(allFiles, path.Join("web", "static"))
	if err != nil {
		panic(fmt.Sprintf("construct static file system error: %s", err))
	}

	return http.FileServer(http.FS(staticFiles))
}()

func collectPageGroups() map[string]*PageGroup {
	if wdIsGo101ProjectRoot {
		return collectPageGroups_NonEmbedding()
	}

	entries, err := fs.ReadDir(allFiles, "pages")
	if err != nil {
		panic("collect page groups (embedding) error: " + err.Error())
	}

	pageGroups := make(map[string]*PageGroup, len(entries))

	for _, e := range entries {
		if e.IsDir() {
			group, handler := e.Name(), dummyHandler
			resFiles, err := fs.Sub(allFiles, path.Join("pages", e.Name(), "res"))
			if err == nil {
				var urlGroup string
				// For history reason, fundamentals pages uses "/article/xxx" URLs.
				if group == "fundamentals" {
					urlGroup = "/article"
				} else if group != "website" {
					urlGroup = "/" + group
				}
				handler = http.StripPrefix(urlGroup+"/res/", http.FileServer(http.FS(resFiles)))
			} else if !os.IsNotExist(err) { // !errors.Is(err, os.ErrNotExist) {
				log.Println(err)
			}

			pageGroups[group] = &PageGroup{resHandler: handler}
		}
	}

	return pageGroups
}

func loadArticleFile(group, file string) ([]byte, error) {
	if wdIsGo101ProjectRoot {
		return loadArticleFile_NonEmbedding(group, file)
	}

	content, err := allFiles.ReadFile(path.Join("pages", group, file))
	if err != nil {
		return nil, err
	}
	return content, nil
}

func parseTemplate(commonPaths []string, files ...string) *template.Template {
	if wdIsGo101ProjectRoot {
		return parseTemplate_NonEmbedding(commonPaths, files...)
	}

	cp := path.Join(commonPaths...)
	ts := make([]string, len(files))
	for i, f := range files {
		ts[i] = path.Join(cp, f)
	}
	return template.Must(template.ParseFS(allFiles, ts...))
}

func updateGolang101() {
	if wdIsGo101ProjectRoot {
		updateGolang101_NonEmbedding()
		return
	}

	if _, err := os.Stat(filepath.Join(".", "golang101.go")); err == nil {
		pullGolang101Project("")
		return
	}
	if filepath.Base(os.Args[0]) == "golang101" {
		log.Println("go", "install", "go101.org/golang101@latest")
		output, err := runShellCommand(time.Minute/2, "", "go", "install", "go101.org/golang101@latest")
		if err != nil {
			log.Printf("error: %s\n%s", err, output)
		} else {
			log.Printf("done.")
		}
	}

	// no ideas how to update
}