package builder import ( "context" "fmt" "io" "os" "path" "slices" "time" "website/internal/config" "website/internal/content" "website/internal/log" "website/internal/sitemap" "website/templates" "github.com/a-h/templ" mapset "github.com/deckarep/golang-set/v2" "github.com/pkg/errors" ) type IOConfig struct { Source string `conf:"default:.,short:s,flag:src"` Destination string `conf:"default:website,short:d,flag:dest"` Development bool `conf:"default:false,flag:dev"` } type Result struct { Hashes []string } var compressFiles = false func mkdirp(dirs ...string) error { err := os.MkdirAll(path.Join(dirs...), 0755) return errors.Wrap(err, "could not create directory") } func outputToFile(output io.Reader, pathParts ...string) error { filename := path.Join(pathParts...) // log.Debug("outputting file", "filename", filename) file, err := openFileAndVariants(filename) if err != nil { return errors.WithMessage(err, "could not open output file") } defer file.Close() if _, err := io.Copy(file, output); err != nil { return errors.WithMessage(err, "could not write output file") } return nil } func renderToFile(component templ.Component, pathParts ...string) error { filename := path.Join(pathParts...) // log.Debug("outputting file", "filename", filename) file, err := openFileAndVariants(filename) if err != nil { return errors.WithMessage(err, "could not open output file") } defer file.Close() if err := component.Render(context.TODO(), file); err != nil { return errors.WithMessage(err, "could not write output file") } return nil } func writerToFile(writer io.WriterTo, pathParts ...string) error { filename := path.Join(pathParts...) // log.Debug("outputting file", "filename", path.Join(filename...)) file, err := openFileAndVariants(filename) if err != nil { return errors.WithMessage(err, "could not open output file") } defer file.Close() if _, err := writer.WriteTo(file); err != nil { return errors.WithMessage(err, "could not write output file") } return nil } func build(outDir string, config config.Config) (*Result, error) { log.Debug("output", "dir", outDir) r := &Result{ Hashes: make([]string, 0), } privateDir := path.Join(outDir, "private") if err := mkdirp(privateDir); err != nil { return nil, errors.WithMessage(err, "could not create private directory") } publicDir := path.Join(outDir, "public") if err := mkdirp(publicDir); err != nil { return nil, errors.WithMessage(err, "could not create public directory") } err := copyRecursive("static", publicDir) if err != nil { return nil, errors.WithMessage(err, "could not copy static files") } if err := mkdirp(publicDir, "post"); err != nil { return nil, errors.WithMessage(err, "could not create post output directory") } log.Debug("reading posts") posts, tags, err := content.ReadPosts("content", "post", publicDir) if err != nil { return nil, err } sitemap := sitemap.New(config) lastMod := time.Now() if len(posts) > 0 { lastMod = posts[0].Date } for _, post := range posts { if err := mkdirp(publicDir, "post", post.Basename); err != nil { return nil, errors.WithMessage(err, "could not create directory for post") } log.Debug("rendering post", "post", post.Basename) sitemap.AddPath(post.URL, post.Date) if err := renderToFile(templates.PostPage(config, post), post.Output); err != nil { return nil, err } } if err := mkdirp(publicDir, "tags"); err != nil { return nil, errors.WithMessage(err, "could not create directory for tags") } log.Debug("rendering tags list") if err := renderToFile( templates.TagsPage(config, "tags", mapset.Sorted(tags), "/tags"), publicDir, "tags", "index.html", ); err != nil { return nil, err } sitemap.AddPath("/tags/", lastMod) for _, tag := range tags.ToSlice() { matchingPosts := []content.Post{} for _, post := range posts { if slices.Contains(post.Taxonomies.Tags, tag) { matchingPosts = append(matchingPosts, post) } } if err := mkdirp(publicDir, "tags", tag); err != nil { return nil, errors.WithMessage(err, "could not create directory") } log.Debug("rendering tags page", "tag", tag) url := "/tags/" + tag if err := renderToFile( templates.TagPage(config, tag, matchingPosts, url), publicDir, "tags", tag, "index.html", ); err != nil { return nil, err } sitemap.AddPath(url, matchingPosts[0].Date) log.Debug("rendering tags feed", "tag", tag) feed, err := renderFeed( fmt.Sprintf("%s - %s", config.Title, tag), config, matchingPosts, tag, ) if err != nil { return nil, errors.WithMessage(err, "could not render tag feed page") } if err := outputToFile(feed, publicDir, "tags", tag, "atom.xml"); err != nil { return nil, err } } log.Debug("rendering list page") if err := renderToFile(templates.ListPage(config, posts, "/post"), publicDir, "post", "index.html"); err != nil { return nil, err } sitemap.AddPath("/post/", lastMod) log.Debug("rendering feed") feed, err := renderFeed(config.Title, config, posts, "feed") if err != nil { return nil, errors.WithMessage(err, "could not render feed") } if err := outputToFile(feed, publicDir, "atom.xml"); err != nil { return nil, err } log.Debug("rendering feed styles") feedStyles, err := renderFeedStyles() if err != nil { return nil, errors.WithMessage(err, "could not render feed styles") } if err := outputToFile(feedStyles, publicDir, "feed-styles.xsl"); err != nil { return nil, err } _, err = feedStyles.Seek(0, 0) if err != nil { return nil, err } h, err := getFeedStylesHash(feedStyles) if err != nil { return nil, err } r.Hashes = append(r.Hashes, h) log.Debug("rendering homepage") _, text, err := content.GetPost(path.Join("content", "index.md")) if err != nil { return nil, err } content, err := content.RenderMarkdown(text) if err != nil { return nil, err } if err := renderToFile(templates.Homepage(config, posts, content), publicDir, "index.html"); err != nil { return nil, err } // it would be nice to set LastMod here, but using the latest post // date would be wrong as the homepage has its own content file // without a date, which could be newer sitemap.AddPath("/", time.Time{}) h, _ = getHTMLStyleHash(publicDir, "index.html") r.Hashes = append(r.Hashes, h) log.Debug("rendering 404 page") if err := renderToFile(templates.NotFound(config, "/404.html"), privateDir, "404.html"); err != nil { return nil, err } log.Debug("rendering sitemap") if err := writerToFile(sitemap, publicDir, "sitemap.xml"); err != nil { return nil, err } log.Debug("rendering robots.txt") rob, err := renderRobotsTXT(config) if err != nil { return nil, err } if err := outputToFile(rob, publicDir, "robots.txt"); err != nil { return nil, err } return r, nil } func BuildSite(ioConfig IOConfig) (*Result, error) { config, err := config.GetConfig() if err != nil { return nil, errors.WithMessage(err, "could not get config") } config.InjectLiveReload = ioConfig.Development compressFiles = !ioConfig.Development loadCSS() return build(ioConfig.Destination, *config) }