2 files changed, 59 insertions, 2 deletions
diff --git a/internal/b2/client.go b/internal/b2/client.go
index f2b6cad..1a97e5f 100644
--- a/internal/b2/client.go
+++ b/internal/b2/client.go
@@ -36,10 +36,10 @@ type BlogPage struct {
 	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()
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
+}