feat: custom property

Author: hui1601

internal/email/template.go

@@ -12,16 +12,13 @@ import (
 )
 
 // TemplateManager manages the loading and rendering of HTML email templates.
-// 템플릿 파일은 지정한 baseDir 하위에 존재하며, 내부 플레이스홀더 (예: {{.name}}, {{.group}}, {{.email}})
-// 의 치환 기능을 지원합니다.
 type TemplateManager struct {
 	templates map[string]*template.Template
 	baseDir   string
 	imageDir  string
 	mu        sync.RWMutex
 }
 
-// NewTemplateManager initializes and returns a new TemplateManager with the provided base directory.
 func NewTemplateManager(baseDir string, imageDir string) *TemplateManager {
 	return &TemplateManager{
 		templates: make(map[string]*template.Template),
@@ -34,18 +31,13 @@ func (tm *TemplateManager) BaseDir() string {
 	return tm.baseDir
 }
 
-// LoadTemplate loads a template file (relative to baseDir) and caches it under the given name.
-// 예: 이름 "default"로 "default.html" 파일을 로드하여 캐시에 저장합니다.
+// LoadTemplate loads a template file and caches it under the given name.
 func (tm *TemplateManager) LoadTemplate(name, filename string) error {
 	fullPath := filepath.Join(tm.baseDir, filename)
 	tmpl, err := template.New(filename).Funcs(template.FuncMap{
-		// Dummy image function that does nothing
-		"image": func(imageSrc string) string {
-			return ""
-		},
-		"imageWithSize": func(imageSrc, width, height string) string {
-			return ""
-		},
+		"image":         func(imageSrc string) string { return "" },
+		"imageWithSize": func(imageSrc, width, height string) string { return "" },
+		"property":      func(key string) string { return "" },
 	}).ParseFiles(fullPath)
 	if err != nil {
 		return fmt.Errorf("failed to parse template file %s: %v", fullPath, err)
@@ -57,7 +49,6 @@ func (tm *TemplateManager) LoadTemplate(name, filename string) error {
 }
 
 // RenderTemplate executes the cached template identified by name using the provided data.
-// 데이터는 예를 들어 map[string]interface{} 형태로 전달할 수 있습니다.
 func (tm *TemplateManager) RenderTemplate(name string, data interface{}) (string, []email.Attachment, error) {
 	tm.mu.RLock()
 	tmpl, exists := tm.templates[name]
@@ -67,7 +58,7 @@ func (tm *TemplateManager) RenderTemplate(name string, data interface{}) (string
 	}
 	var buf bytes.Buffer
 	var attachments []email.Attachment
-	tmpl.Funcs(template.FuncMap{
+	tmpl = tmpl.Funcs(template.FuncMap{
 		"image": func(imageSrc string) template.HTML {
 			imagePath := filepath.Join(tm.imageDir, imageSrc)
 			if _, err := os.Stat(imagePath); err != nil {
@@ -110,14 +101,24 @@ func (tm *TemplateManager) RenderTemplate(name string, data interface{}) (string
 			})
 			return template.HTML(fmt.Sprintf("<img src=\"cid:%s\" alt=\"%s\" width=\"%s\" height=\"%s\">", imageSrc, imageSrc, width, height))
 		},
+		"property": func(key string) string {
+			if dataMap, ok := data.(map[string]interface{}); ok {
+				if custom, ok := dataMap["custom"].(map[string]string); ok {
+					if val, exists := custom[key]; exists {
+						return val
+					}
+				}
+			}
+			return ""
+		},
 	})
+
 	if err := tmpl.Execute(&buf, data); err != nil {
 		return "", nil, fmt.Errorf("failed to execute template %s: %v", name, err)
 	}
 	return buf.String(), attachments, nil
 }
 
-// ListTemplates returns a slice of the names of the currently loaded templates.
 func (tm *TemplateManager) ListTemplates() []string {
 	tm.mu.RLock()
 	defer tm.mu.RUnlock()
@@ -128,8 +129,6 @@ func (tm *TemplateManager) ListTemplates() []string {
 	return names
 }
 
-// ExportedTemplates returns a shallow copy of the internal template map.
-// 외부에서는 이 복사본을 읽기 전용으로 활용할 수 있습니다.
 func (tm *TemplateManager) ExportedTemplates() map[string]*template.Template {
 	tm.mu.RLock()
 	defer tm.mu.RUnlock()
@@ -140,15 +139,12 @@ func (tm *TemplateManager) ExportedTemplates() map[string]*template.Template {
 	return templatesCopy
 }
 
-// DeleteTemplate removes the template identified by name from the cache.
 func (tm *TemplateManager) DeleteTemplate(name string) {
 	tm.mu.Lock()
 	defer tm.mu.Unlock()
 	delete(tm.templates, name)
 }
 
-// Templates returns the internal template map.
-// 이 메서드는 내부에서만 읽기 전용으로 사용합니다.
 func (tm *TemplateManager) Templates() map[string]*template.Template {
 	return tm.ExportedTemplates()
 }

internal/web/handlers.go

@@ -203,6 +203,9 @@ func (h *APIHandler) PreviewTemplateHandler(w http.ResponseWriter, r *http.Reque
 				return template.HTML(fmt.Sprintf("<img src=\"%s\" width=\"%s\" height=\"%s\">",
 					html.EscapeString("/api/images/"+imageSrc), html.EscapeString(width), html.EscapeString(height)))
 			},
+			"property": func(key string) template.HTML {
+				return template.HTML(fmt.Sprintf("<code>PROPERTY(%s)</code>", key))
+			},
 		}).Parse(reqData.Content)
 		if err != nil {
 			http.Error(w, fmt.Sprintf("템플릿 파싱 실패: %v", err), http.StatusBadRequest)
@@ -220,6 +223,13 @@ func (h *APIHandler) PreviewTemplateHandler(w http.ResponseWriter, r *http.Reque
 				return template.HTML(fmt.Sprintf("<img src=\"%s\" alt=\"\" style=\"max-width: 100%%; height: auto;\">",
 					html.EscapeString("/api/images/"+imageSrc)))
 			},
+			"imageWithSize": func(imageSrc, width, height string) template.HTML {
+				return template.HTML(fmt.Sprintf("<img src=\"%s\" width=\"%s\" height=\"%s\">",
+					html.EscapeString("/api/images/"+imageSrc), html.EscapeString(width), html.EscapeString(height)))
+			},
+			"property": func(key string) template.HTML {
+				return template.HTML(fmt.Sprintf("<code>PROPERTY(%s)</code>", key))
+			},
 		}).Parse(string(contentBytes))
 		if err != nil {
 			http.Error(w, fmt.Sprintf("템플릿 파싱 실패: %v", err), http.StatusInternalServerError)
@@ -277,8 +287,9 @@ func (h *APIHandler) EmailHandler(w http.ResponseWriter, r *http.Request) {
 	}
 
 	type RecipientInfo struct {
-		Name  string `json:"name"`
-		Email string `json:"email"`
+		Name   string            `json:"name"`
+		Email  string            `json:"email"`
+		Custom map[string]string `json:"custom,omitempty"`
 	}
 
 	var reqData struct {
@@ -299,9 +310,10 @@ func (h *APIHandler) EmailHandler(w http.ResponseWriter, r *http.Request) {
 	go func(recipients []RecipientInfo) {
 		for _, rec := range recipients {
 			data := map[string]interface{}{
-				"name":  rec.Name,
-				"email": rec.Email,
-				"year":  time.Now().Year(),
+				"name":   rec.Name,
+				"email":  rec.Email,
+				"year":   time.Now().Year(),
+				"custom": rec.Custom,
 			}
 			body, attachments, err := h.TemplateManager.RenderTemplate(reqData.Template, data)
 			if err != nil {