package main import ( "bytes" "errors" "fmt" cp "github.com/otiai10/copy" "os" "path/filepath" "strings" "text/template" "github.com/PuerkitoBio/goquery" "github.com/avelino/awesome-go/pkg/slug" ) type Link struct { Title string Url string Description string } type Object struct { Title string Slug string Description string Items []Link } // Source files const readmePath = "README.md" // This files should be copied 'as is' to outDir directory var staticFiles = []string{ "tmpl/assets", "tmpl/_redirects", "tmpl/robots.txt", } // TODO: embed // Templates var tplIndex = template.Must(template.ParseFiles("tmpl/tmpl.html")) var tplCategoryIndex = template.Must(template.ParseFiles("tmpl/cat-tmpl.html")) var tplSitemap = template.Must(template.ParseFiles("tmpl/sitemap-tmpl.xml")) // Output files const outDir = "out/" // NOTE: trailing slash is required var outIndexFile = filepath.Join(outDir, "index.html") var outSitemapFile = filepath.Join(outDir, "sitemap.xml") func main() { // Cleanup and re-create output directory { if err := os.RemoveAll(outDir); err != nil { panic(err) } if err := mkdirAll(outDir); err != nil { panic(err) } } err := GenerateHTML(readmePath, outIndexFile) if err != nil { panic(err) } input, err := os.ReadFile(outIndexFile) if err != nil { panic(err) } query, err := goquery.NewDocumentFromReader(bytes.NewReader(input)) if err != nil { panic(err) } objs := make(map[string]Object) query.Find("body #contents").NextFiltered("ul").Find("ul").Each(func(_ int, s *goquery.Selection) { s.Find("li a").Each(func(_ int, s *goquery.Selection) { selector, exists := s.Attr("href") if !exists { return } obj, err := makeObjByID(selector, query.Find("body")) if err != nil { return } objs[selector] = *obj }) }) if err := makeSiteStruct(objs); err != nil { // FIXME: remove all panics panic(err) } changeLinksInIndex(string(input), query, objs) makeSitemap(objs) for _, srcFilename := range staticFiles { dstFilename := filepath.Join(outDir, filepath.Base(srcFilename)) fmt.Printf("Copy static file: %s -> %s\n", srcFilename, dstFilename) if err := cp.Copy(srcFilename, dstFilename); err != nil { panic(err) } } } func mkdirAll(path string) error { _, err := os.Stat(path) // NOTE: directory is exists if err == nil { return nil } // NOTE: unknown error if !os.IsNotExist(err) { return err } // NOTE: directory is not exists // FIXME: fix rights if err := os.MkdirAll(path, 0o755); err != nil { return err } return nil } func makeSiteStruct(objs map[string]Object) error { for _, obj := range objs { categoryDir := filepath.Join(outDir, obj.Slug) if err := mkdirAll(categoryDir); err != nil { return err } // FIXME: embed templates // FIXME: parse templates once at start categoryIndexFilename := filepath.Join(categoryDir, "index.html") f, err := os.Create(categoryIndexFilename) if err != nil { return err } fmt.Printf("Write category Index file: %s\n", categoryIndexFilename) if err := tplCategoryIndex.Execute(f, obj); err != nil { return err } } return nil } func makeSitemap(objs map[string]Object) { // FIXME: handle error f, _ := os.Create(outSitemapFile) fmt.Printf("Render Sitemap to: %s\n", outSitemapFile) _ = tplSitemap.Execute(f, objs) } func makeObjByID(selector string, s *goquery.Selection) (*Object, error) { var obj Object var err error s.Find(selector).Each(func(_ int, selCatHeader *goquery.Selection) { selDescr := selCatHeader.NextFiltered("p") // FIXME: bug. this would select links from all neighboring // sub-categories until the next category. To prevent this we should // find only first ul ul := selCatHeader.NextFilteredUntil("ul", "h2") var links []Link ul.Find("li").Each(func(_ int, selLi *goquery.Selection) { selLink := selLi.Find("a") url, _ := selLink.Attr("href") link := Link{ Title: selLink.Text(), // FIXME: Title contains only title but description contains Title + description Description: selLi.Text(), Url: url, } links = append(links, link) }) // FIXME: In this case we would have an empty category in main index.html with link to 404 page. if len(links) == 0 { err = errors.New("object has no links") return } obj = Object{ Slug: slug.Generate(selCatHeader.Text()), Title: selCatHeader.Text(), Description: selDescr.Text(), Items: links, } }) if err != nil { return nil, fmt.Errorf("unable to build an object: %w", err) } return &obj, nil } func changeLinksInIndex(html string, query *goquery.Document, objs map[string]Object) { query.Find("body #content ul li ul li a").Each(func(_ int, s *goquery.Selection) { href, hrefExists := s.Attr("href") if !hrefExists { // FIXME: looks like is an error. Tag `a` in our case always // should have `href` attr. return } // do not replace links if no page has been created for it _, objExists := objs[href] if !objExists { return } // FIXME: parse url uri := strings.SplitAfter(href, "#") if len(uri) >= 2 && uri[1] != "contents" { // FIXME: use s.SetAttr html = strings.ReplaceAll( html, fmt.Sprintf(`href="%s"`, href), fmt.Sprintf(`href="%s"`, uri[1]), ) } }) fmt.Printf("Rewrite links in Index file: %s\n", outIndexFile) _ = os.WriteFile(outIndexFile, []byte(html), 0644) }