initial logic for generating folder/htmld sub cat

Signed-off-by: Avelino <avelinorun@gmail.com>

make_site.go

@@ -0,0 +1,99 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+	"text/template"
+
+	"github.com/PuerkitoBio/goquery"
+	"github.com/avelino/slugify"
+)
+
+type Link struct {
+	Title       string
+	Url         string
+	Description string
+}
+
+type Object struct {
+	Title       string
+	Slug        string
+	Description string
+	Items       []Link
+}
+
+func main() {
+	GenerateHTML()
+	input, err := ioutil.ReadFile("./tmpl/index.html")
+	if err != nil {
+		panic(err)
+	}
+	buf := bytes.NewBuffer(input)
+	query, err := goquery.NewDocumentFromReader(buf)
+	if err != nil {
+		panic(err)
+	}
+
+	objs := []Object{}
+	query.Find("body #content ul ul").First().Each(func(_ int, s *goquery.Selection) {
+
+		s.Find("li a").Each(func(_ int, s *goquery.Selection) {
+			selector, _ := s.Attr("href")
+			obj := makeObjById(selector, query.Find("body"))
+			objs = append(objs, obj)
+		})
+	})
+
+	makeSiteStruct(objs)
+	makeSitemap(objs)
+}
+
+func makeSiteStruct(objs []Object) {
+	for _, obj := range objs {
+		folder := fmt.Sprintf("tmpl/%s", obj.Slug)
+		fmt.Println(folder)
+
+		err := os.Mkdir(folder, 0755)
+		if err != nil {
+			log.Println(err)
+		}
+
+		t := template.Must(template.ParseFiles("tmpl/cat-tmpl.html"))
+		f, _ := os.Create(fmt.Sprintf("%s/index.html", folder))
+		t.Execute(f, obj)
+	}
+}
+
+func makeSitemap(objs []Object) {
+	t := template.Must(template.ParseFiles("tmpl/sitemap-tmpl.xml"))
+	f, _ := os.Create("tmpl/sitemap.xml")
+	t.Execute(f, objs)
+}
+
+func makeObjById(selector string, s *goquery.Selection) (obj Object) {
+	s.Find(selector).Each(func(_ int, s *goquery.Selection) {
+		desc := s.NextFiltered("p")
+		ul := desc.NextFiltered("ul")
+
+		links := []Link{}
+		ul.Find("li").Each(func(_ int, s *goquery.Selection) {
+			url, _ := s.Find("a").Attr("href")
+			link := Link{
+				Title:       s.Find("a").Text(),
+				Description: s.Text(),
+				Url:         url,
+			}
+			links = append(links, link)
+		})
+		obj = Object{
+			Slug:        slugify.Slugify(s.Text()),
+			Title:       s.Text(),
+			Description: desc.Text(),
+			Items:       links,
+		}
+	})
+	return
+}