2 files changed, 65 insertions, 4 deletions
diff --git a/internal/b2/client.go b/internal/b2/client.go
index f2b6cad..5322b84 100644
--- a/internal/b2/client.go
+++ b/internal/b2/client.go
@@ -33,13 +33,14 @@ func NewB2Client(cfg *config.B2Config) (*B2Client, error) {
 
 type BlogPage struct {
 	Link     string
+	FileName string
 	Metadata *frontmatter.Metadata
 }
 
-func (c *B2Client) Scan() ([]*BlogPage, error) {
+func (c *B2Client) Scan(prefix string) ([]*BlogPage, error) {
 	filePaths := []*BlogPage{}
 
-	iter := c.bucket.List(context.Background(), b2.ListPrefix(c.prefix))
+	iter := c.bucket.List(context.Background(), b2.ListPrefix(c.prefix+prefix))
 
 	for iter.Next() {
 		obj := iter.Object()
@@ -69,10 +70,13 @@ func (c *B2Client) Scan() ([]*BlogPage, error) {
 			return nil, fmt.Errorf("failed to parse published time metadata field: %w", err)
 		}
 
-		linkParts := strings.Split(obj.Name(), ".")
+		linkParts := strings.Split(obj.Name(), "/")
+		nameParts := strings.Split(linkParts[len(linkParts)-1], ".")
+		fileName := strings.Join(nameParts[:len(linkParts)-1], ".")
 
 		filePaths = append(filePaths, &BlogPage{
-			Link: strings.Join(linkParts[:len(linkParts)-1], "."),
+			Link:     obj.Name(),
+			FileName: fileName,
 			Metadata: &frontmatter.Metadata{
 				Title:            attrs.Info["title"],
 				ShortDescription: attrs.Info["short-description"],
diff --git a/internal/templatemanager/templatemanager.go b/internal/templatemanager/templatemanager.go
new file mode 100644
index 0000000..5effcb3
--- /dev/null
+++ b/internal/templatemanager/templatemanager.go
@@ -0,0 +1,57 @@
+package templatemanager
+
+import (
+	"bytes"
+	"fmt"
+	"html/template"
+	"path/filepath"
+	"strings"
+)
+
+type TemplateManager struct {
+	templates map[string]templateManagerRender
+}
+
+type templateManagerRender struct {
+	Main string
+	Tmpl *template.Template
+}
+
+type TemplateManagerTemplates struct {
+	Name  string
+	Files []string
+}
+
+func NewTemplateManager(templates []TemplateManagerTemplates) (*TemplateManager, error) {
+	templateMap := make(map[string]templateManagerRender)
+
+	for _, tmplStruct := range templates {
+		tmpl := template.New("").Funcs(template.FuncMap{
+			"contains": strings.Contains,
+		})
+		tmpl, err := tmpl.ParseFiles(tmplStruct.Files...)
+		if err != nil {
+			return nil, err
+		}
+		templateMap[tmplStruct.Name] = templateManagerRender{
+			Main: filepath.Base(tmplStruct.Files[0]),
+			Tmpl: tmpl,
+		}
+
+	}
+
+	return &TemplateManager{
+		templates: templateMap,
+	}, nil
+}
+
+func (tm *TemplateManager) Render(name string, data interface{}) ([]byte, error) {
+	tmpl, exists := tm.templates[name]
+	if !exists {
+		return nil, fmt.Errorf("template %s is not found", name)
+	}
+
+	var buf bytes.Buffer
+	err := tmpl.Tmpl.ExecuteTemplate(&buf, tmpl.Main, data)
+	return buf.Bytes(), err
+}