mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
"github.com/Rain-kl/Wavelet/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ListTemplates returns all templates ordered by system flag and creation time.
|
|
func ListTemplates(ctx context.Context) ([]model.Template, error) {
|
|
var templates []model.Template
|
|
if err := db.DB(ctx).Order("is_system DESC, created_at DESC").Find(&templates).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return templates, nil
|
|
}
|
|
|
|
// GetTemplateByKey loads a template by its key.
|
|
func GetTemplateByKey(ctx context.Context, key string) (model.Template, error) {
|
|
var tmpl model.Template
|
|
if err := db.DB(ctx).Where("key = ?", key).First(&tmpl).Error; err != nil {
|
|
return model.Template{}, err
|
|
}
|
|
return tmpl, nil
|
|
}
|
|
|
|
// TemplateExistsByKey reports whether a template key is already taken.
|
|
func TemplateExistsByKey(ctx context.Context, key string) (bool, error) {
|
|
var existing model.Template
|
|
err := db.DB(ctx).Where("key = ?", key).First(&existing).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return false, nil
|
|
}
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// CreateTemplate persists a new template.
|
|
func CreateTemplate(ctx context.Context, tmpl *model.Template) error {
|
|
return db.DB(ctx).Create(tmpl).Error
|
|
}
|
|
|
|
// SaveTemplate updates an existing template.
|
|
func SaveTemplate(ctx context.Context, tmpl *model.Template) error {
|
|
return db.DB(ctx).Save(tmpl).Error
|
|
}
|
|
|
|
// DeleteTemplate removes a template record.
|
|
func DeleteTemplate(ctx context.Context, tmpl *model.Template) error {
|
|
return db.DB(ctx).Delete(tmpl).Error
|
|
}
|