Sofia Papagiannaki
111 lines
2.3 KiB
Go
111 lines
2.3 KiB
Go
package definitions
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
)
|
|
|
|
// swagger:route GET /api/provisioning/templates provisioning RouteGetTemplates
|
|
//
|
|
// Get all message templates.
|
|
//
|
|
// Responses:
|
|
// 200: MessageTemplate
|
|
// 400: ValidationError
|
|
|
|
// swagger:route GET /api/provisioning/templates/{name} provisioning RouteGetTemplate
|
|
//
|
|
// Get a message template.
|
|
//
|
|
// Responses:
|
|
// 200: MessageTemplate
|
|
// 404: NotFound
|
|
|
|
// swagger:route PUT /api/provisioning/templates/{name} provisioning RoutePutTemplate
|
|
//
|
|
// Updates an existing template.
|
|
//
|
|
// Consumes:
|
|
// - application/json
|
|
//
|
|
// Responses:
|
|
// 202: Ack
|
|
// 400: ValidationError
|
|
|
|
// swagger:route DELETE /api/provisioning/templates/{name} provisioning RouteDeleteTemplate
|
|
//
|
|
// Delete a template.
|
|
//
|
|
// Responses:
|
|
// 204: Ack
|
|
|
|
// swagger:parameters RouteGetTemplate RoutePutTemplate RouteDeleteTemplate
|
|
type RouteGetTemplateParam struct {
|
|
// Template Name
|
|
// in:path
|
|
Name string `json:"name"`
|
|
}
|
|
|
|
// swagger:model
|
|
type MessageTemplate struct {
|
|
Name string
|
|
Template string
|
|
Provenance models.Provenance `json:"provenance,omitempty"`
|
|
}
|
|
|
|
// swagger:model
|
|
type MessageTemplates []MessageTemplate
|
|
|
|
type MessageTemplateContent struct {
|
|
Template string
|
|
}
|
|
|
|
// swagger:parameters RoutePutTemplate
|
|
type MessageTemplatePayload struct {
|
|
// in:body
|
|
Body MessageTemplateContent
|
|
}
|
|
|
|
func (t *MessageTemplate) ResourceType() string {
|
|
return "template"
|
|
}
|
|
|
|
func (t *MessageTemplate) ResourceID() string {
|
|
return t.Name
|
|
}
|
|
|
|
func (t *MessageTemplate) Validate() error {
|
|
if t.Name == "" {
|
|
return fmt.Errorf("template must have a name")
|
|
}
|
|
if t.Template == "" {
|
|
return fmt.Errorf("template must have content")
|
|
}
|
|
|
|
_, err := template.New("").Parse(t.Template)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid template: %w", err)
|
|
}
|
|
|
|
content := strings.TrimSpace(t.Template)
|
|
found, err := regexp.MatchString(`\{\{\s*define`, content)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to match regex: %w", err)
|
|
}
|
|
if !found {
|
|
lines := strings.Split(content, "\n")
|
|
for i, s := range lines {
|
|
lines[i] = " " + s
|
|
}
|
|
content = strings.Join(lines, "\n")
|
|
content = fmt.Sprintf("{{ define \"%s\" }}\n%s\n{{ end }}", t.Name, content)
|
|
}
|
|
t.Template = content
|
|
|
|
return nil
|
|
}
|