mirror of https://github.com/jlelse/GoBlog
115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/sha1"
|
|
"crypto/sha512"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"io"
|
|
"mime"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
const assetsFolder = "templates/assets"
|
|
|
|
var assetFileNames map[string]string = map[string]string{}
|
|
var assetFiles map[string]*assetFile = map[string]*assetFile{}
|
|
|
|
type assetFile struct {
|
|
contentType string
|
|
sri string
|
|
body []byte
|
|
}
|
|
|
|
func initTemplateAssets() (err error) {
|
|
err = filepath.Walk(assetsFolder, func(path string, info os.FileInfo, err error) error {
|
|
if info.Mode().IsRegular() {
|
|
compiled, err := compileAsset(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if compiled != "" {
|
|
assetFileNames[strings.TrimPrefix(path, assetsFolder+"/")] = compiled
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func compileAsset(name string) (string, error) {
|
|
content, err := os.ReadFile(name)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
ext := path.Ext(name)
|
|
compiledExt := ext
|
|
switch ext {
|
|
case ".js":
|
|
content, err = minifier.Bytes("application/javascript", content)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
case ".css":
|
|
content, err = minifier.Bytes("text/css", content)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
default:
|
|
// Do nothing
|
|
}
|
|
// Hashes
|
|
sha1Hash := sha1.New()
|
|
sha512Hash := sha512.New()
|
|
if _, err := io.MultiWriter(sha1Hash, sha512Hash).Write(content); err != nil {
|
|
return "", err
|
|
}
|
|
// File name
|
|
compiledFileName := fmt.Sprintf("%x", sha1Hash.Sum(nil)) + compiledExt
|
|
// SRI
|
|
sriHash := fmt.Sprintf("sha512-%s", base64.StdEncoding.EncodeToString(sha512Hash.Sum(nil)))
|
|
// Create struct
|
|
assetFiles[compiledFileName] = &assetFile{
|
|
contentType: mime.TypeByExtension(compiledExt),
|
|
sri: sriHash,
|
|
body: content,
|
|
}
|
|
return compiledFileName, err
|
|
}
|
|
|
|
// Function for templates
|
|
func assetFileName(fileName string) string {
|
|
return "/" + assetFileNames[fileName]
|
|
}
|
|
|
|
func assetSRI(fileName string) string {
|
|
return assetFiles[assetFileNames[fileName]].sri
|
|
}
|
|
|
|
func allAssetPaths() []string {
|
|
var paths []string
|
|
for _, name := range assetFileNames {
|
|
paths = append(paths, "/"+name)
|
|
}
|
|
return paths
|
|
}
|
|
|
|
// Gets only called by registered paths
|
|
func serveAsset(w http.ResponseWriter, r *http.Request) {
|
|
af, ok := assetFiles[strings.TrimPrefix(r.URL.Path, "/")]
|
|
if !ok {
|
|
serve404(w, r)
|
|
return
|
|
}
|
|
w.Header().Set("Cache-Control", "public,max-age=31536000,immutable")
|
|
w.Header().Set(contentType, af.contentType+charsetUtf8Suffix)
|
|
_, _ = w.Write(af.body)
|
|
}
|