TIL-7 Tags into pages

Signed-off-by: Chris Cummer <[email protected]>

Author: senorprogrammer

til.go

@@ -9,6 +9,7 @@ import (
 	"os"
 	"os/exec"
 	"path/filepath"
+	"regexp"
 	"sort"
 	"strings"
 	"sync"
@@ -383,25 +384,41 @@ func getTargetDir(cfg *config.Config, withDocsDir bool) (string, error) {
 
 func buildContent() {
 	pages := loadPages()
+	buildContentPages(pages)
 	tagMap := buildTagPages(pages)
 	buildIndexPage(pages, tagMap)
 }
 
+// buildContentPages loops through all the pages and tells them to save themselves
+// to disk. This process writes any auto-generated content into the pages
+func buildContentPages(pages []*Page) {
+	for _, page := range pages {
+		if page.IsContentPage() {
+			page.AppendTagsToContent()
+			page.Save()
+		}
+	}
+}
+
 // buildIndexPage creates the main index.md page that is the root of the site
 func buildIndexPage(pages []*Page, tagMap *TagMap) {
 	Info(statusIdxBuild)
 
 	content := ""
 
 	// Write the tag list into the top of the index
-	for _, tag := range tagMap.SortedTagNames() {
-		content += fmt.Sprintf(
-			"[%s](%s), ",
-			tag,
-			fmt.Sprintf("./%s", tag),
-		)
+	tagLinks := []string{}
+
+	for _, tagName := range tagMap.SortedTagNames() {
+		tags := tagMap.Get(tagName)
+		if len(tags) > 0 {
+			tagLinks = append(tagLinks, tags[0].Link())
+		}
 	}
 
+	content += strings.Join(tagLinks, ", ")
+	content += "\n"
+
 	// Write the page list into the middle of the page
 	content += pagesToHTMLUnorderedList(pages)
 	content += "\n"
@@ -690,13 +707,6 @@ func save(commitMsg string) {
 		Defeat(errors.New(errConfigValueRead))
 	}
 
-	// if commitMsg == "" {
-	// 	// The incoming commitMsg is optional (if it is set, it probably came in
-	// 	// via command line args on -save). If it isn't set, we use the default
-	// 	// from the config file instead
-	// 	commitMsg = defaultCommitMsg
-	// }
-
 	commit, err := w.Commit(commitMsg, &git.CommitOptions{
 		Author: &object.Signature{
 			Name:  defaultCommitName,
@@ -766,6 +776,47 @@ type Page struct {
 	Title    string `yaml:"title"`
 }
 
+// AppendTagsToContent programatically modifies the page content to save auto-included
+// content like the tag list
+func (page *Page) AppendTagsToContent() bool {
+	if len(page.Tags()) == 0 {
+		return true
+	}
+
+	tagLinks := []string{}
+	for _, tag := range page.Tags() {
+		if tag.Link() != "" {
+			tagLinks = append(tagLinks, tag.Link())
+		}
+	}
+
+	tagList := strings.Join(tagLinks, ", ")
+
+	// Tags
+	tagsStartStr := "<!-- TAGS:START -->"
+	tagsEndStr := "<!-- TAGS:END -->"
+
+	re := fmt.Sprintf("`%s\n.*\n%s`", tagsStartStr, tagsEndStr)
+	rg := regexp.MustCompile(re)
+
+	newContent := rg.ReplaceAllString(page.Content, tagList)
+
+	if page.Content != newContent {
+		// Swap the old content with the new content
+		page.Content = newContent
+	} else {
+		// Append the tag list to the end of the page
+		page.Content += fmt.Sprintf(
+			"\n%s\n%s\n%s\n",
+			tagsStartStr,
+			tagList,
+			tagsEndStr,
+		)
+	}
+
+	return true
+}
+
 // CreatedAt returns a time instance representing when the page was created
 func (page *Page) CreatedAt() time.Time {
 	date, err := time.Parse(time.RFC3339, page.Date)
@@ -805,6 +856,14 @@ func (page *Page) PrettyDate() string {
 	return page.CreatedAt().Format("Jan 02, 2006")
 }
 
+// Save writes the page to disk
+func (page *Page) Save() error {
+	Info(page.FilePath)
+
+	err := ioutil.WriteFile(page.FilePath, []byte(page.Content), 0644)
+	return err
+}
+
 // Tags returns a slice of tags assigned to this page
 func (page *Page) Tags() []*Tag {
 	tags := []*Tag{}
@@ -845,6 +904,19 @@ func (tag *Tag) IsValid() bool {
 	return tag.Name != ""
 }
 
+// Link returns a link string suitable for embedding in a Markdown page
+func (tag *Tag) Link() string {
+	if tag.Name == "" {
+		return ""
+	}
+
+	return fmt.Sprintf(
+		"[%s](%s)",
+		tag.Name,
+		fmt.Sprintf("./%s", tag.Name),
+	)
+}
+
 /* -------------------- TagMap -------------------- */
 
 // TagMap is a map of tag name to Tag instance