Alexander Weaver
0f56462fbe
* Generate API for writing templates * Persist templates app logic layer * Validate templates * Extract logic, make set and delete methods * Drop post route for templates * Fix response details, wire up remainder of API * Authorize routes * Mirror some existing tests on new APIs * Generate mock for prov store * Wire up prov store mock, add tests using it * Cover cases for both storage paths * Add happy path tests and fix bugs if file contains no template section * Normalize template content with define statement * Tests for deletion * Fix linter error * Move provenance field to DTO * empty commit * ID to name * Fix in auth too
94 lines
1.9 KiB
Go
94 lines
1.9 KiB
Go
package definitions
|
|
|
|
import (
|
|
"fmt"
|
|
"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: Accepted
|
|
// 400: ValidationError
|
|
|
|
// swagger:route DELETE /api/provisioning/templates/{name} provisioning RouteDeleteTemplate
|
|
//
|
|
// Delete a template.
|
|
//
|
|
// Responses:
|
|
// 204: Accepted
|
|
|
|
type MessageTemplate struct {
|
|
Name string
|
|
Template string
|
|
Provenance models.Provenance `json:"provenance,omitempty"`
|
|
}
|
|
|
|
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")
|
|
}
|
|
|
|
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
|
|
}
|