213 lines
6.4 KiB
Go
213 lines
6.4 KiB
Go
package maker
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"yizhisec.com/hsv2/forge/internal/opt"
|
|
"yizhisec.com/hsv2/forge/pkg/logger"
|
|
"yizhisec.com/hsv2/forge/pkg/model"
|
|
"yizhisec.com/hsv2/forge/pkg/resource"
|
|
)
|
|
|
|
type SeafileOpt func(*seafileOpt)
|
|
type seafileOpt struct {
|
|
DBHost string
|
|
DBPassword string
|
|
AdminEmail string
|
|
AdminPassword string
|
|
ServerHostname string
|
|
Storage string
|
|
}
|
|
|
|
func WithSeafileStorage(storage string) SeafileOpt {
|
|
return func(o *seafileOpt) {
|
|
if opt.StorageSizeReg.MatchString(storage) {
|
|
o.Storage = storage
|
|
}
|
|
}
|
|
}
|
|
|
|
func WithSeafileDBHost(host string) SeafileOpt {
|
|
return func(o *seafileOpt) {
|
|
if o.DBHost == "" {
|
|
o.DBHost = host
|
|
}
|
|
}
|
|
}
|
|
|
|
func WithSeafileDBPassword(password string) SeafileOpt {
|
|
return func(o *seafileOpt) {
|
|
if o.DBPassword == "" {
|
|
o.DBPassword = password
|
|
}
|
|
}
|
|
}
|
|
|
|
func WithSeafileAdminEmail(email string) SeafileOpt {
|
|
return func(o *seafileOpt) {
|
|
if opt.EmailReg.MatchString(email) {
|
|
o.AdminEmail = email
|
|
}
|
|
}
|
|
}
|
|
|
|
func WithSeafileAdminPassword(password string) SeafileOpt {
|
|
return func(o *seafileOpt) {
|
|
if o.AdminPassword == "" {
|
|
o.AdminPassword = password
|
|
}
|
|
}
|
|
}
|
|
|
|
func WithSeafileHostname(hostname string) SeafileOpt {
|
|
return func(o *seafileOpt) {
|
|
if o.ServerHostname == "" {
|
|
o.ServerHostname = hostname
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *maker) Seafile(ctx context.Context, opts ...SeafileOpt) error {
|
|
const (
|
|
_config = `
|
|
ControllerServer:
|
|
UserManagement: user-service.hsv2:9013
|
|
Database:
|
|
Mysql:
|
|
Address: %s:3306
|
|
DBName: backup_server
|
|
Password: %s
|
|
SeafileDBName: seafile_db
|
|
UserName: root
|
|
Log:
|
|
Dir: ./log
|
|
Level: 1
|
|
Name: hs_backup_seafile
|
|
SeafileServer:
|
|
Admin: %s
|
|
AdminPassword: %s
|
|
BackupDir: /seafile/backup_data
|
|
Host: seafile-service
|
|
Port: 80
|
|
StorageDir: /seafile/storage
|
|
Sentry:
|
|
DSN: https://fd7149f063c211eda2b50242ac15001c@sentry.yizhisec.com:13443/7
|
|
TracesSampleRate: 1
|
|
Web:
|
|
Host: 0.0.0.0
|
|
Mode: release
|
|
Port: 9027
|
|
WorkDir: /yizhisec/hs_backup_seafile/workspace
|
|
YosGuard:
|
|
Host: 172.17.0.1
|
|
Port: 7788`
|
|
_upsert = `#!/bin/bash
|
|
|
|
kubectl create configmap config-backup-seafile --namespace seafile --from-file=config.yml=./config.yml --dry-run=client -o yaml | kubectl apply -f -
|
|
kubectl create configmap nginx-seafile --namespace hsv2 --from-file=seafile.conf=./nginx.conf --dry-run=client -o yaml | kubectl apply -f -
|
|
|
|
kubectl apply -f deployment.yaml
|
|
kubectl rollout restart deployment backup-seafile-deployment -n seafile`
|
|
)
|
|
|
|
var (
|
|
err error
|
|
o = &seafileOpt{
|
|
DBHost: "mysql-cluster-mysql-master.db-mysql",
|
|
DBPassword: "L0hMysql.",
|
|
AdminEmail: "admin@yizhisec.com",
|
|
AdminPassword: "asecret",
|
|
ServerHostname: "cloud.hybridscope.com",
|
|
Storage: "50Gi",
|
|
}
|
|
workdir = filepath.Join(m.workdir, "dependency", "seafile")
|
|
)
|
|
|
|
for _, fn := range opts {
|
|
fn(o)
|
|
}
|
|
|
|
logger.Info("☑️ maker.Seafile: 开始构建 seafile 依赖, dir = %s", workdir)
|
|
|
|
// 1. 准备工作目录
|
|
logger.Debug("☑️ make.Seafile: 准备工作目录: %s", workdir)
|
|
if err = os.MkdirAll(workdir, 0755); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备工作目录: %s 失败, err = %v", workdir, err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ make.Seafile: 准备工作目录: %s 成功", workdir)
|
|
|
|
// 2. seafile yaml
|
|
logger.Debug("☑️ make.Seafile: 准备 seafile yaml")
|
|
bs := []byte(fmt.Sprintf(resource.YAMLSeafile, o.DBHost, o.DBPassword, o.AdminEmail, o.AdminPassword, o.ServerHostname, o.Storage))
|
|
if err = os.WriteFile(filepath.Join(workdir, "seafile.yaml"), bs, 0644); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备 seafile yaml: %s 失败, err = %v", filepath.Join(workdir, "seafile.yaml"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ make.Seafile: 准备 seafile yaml 成功")
|
|
|
|
// 3. backup-seafile deployment
|
|
logger.Debug("☑️ make.Seafile: 准备 backup-seafile deployment")
|
|
bs = []byte(resource.YAMLBackupSeafile)
|
|
if err = os.WriteFile(filepath.Join(workdir, "deployment.yaml"), bs, 0644); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备 backup-seafile deployment: %s 失败, err = %v", filepath.Join(workdir, "deployment.yaml"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ make.Seafile: 准备 backup-seafile deployment 成功")
|
|
|
|
// 4. config.yml
|
|
logger.Debug("☑️ make.Seafile: 准备 config.yml")
|
|
bs = []byte(fmt.Sprintf(_config, o.DBHost, o.DBPassword, o.AdminEmail, o.AdminPassword))
|
|
if err = os.WriteFile(filepath.Join(workdir, "config.yml"), bs, 0644); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备 config.yml: %s 失败, err = %v", filepath.Join(workdir, "config.yml"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ make.Seafile: 准备 config.yml 成功")
|
|
|
|
// 5. seafile.conf
|
|
logger.Debug("☑️ make.Seafile: 准备 seafile.conf")
|
|
bs = resource.NGINXSeafile
|
|
if err = os.WriteFile(filepath.Join(workdir, "seafile.conf"), bs, 0644); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备 seafile.conf: %s 失败, err = %v", filepath.Join(workdir, "seafile.conf"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ make.Seafile: 准备 seafile.conf 成功")
|
|
|
|
// 6. upsert.sh
|
|
logger.Debug("☑️ make.Seafile: 准备 upsert.sh")
|
|
bs = []byte(_upsert)
|
|
if err = os.WriteFile(filepath.Join(workdir, "upsert.sh"), bs, 0755); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备 upsert.sh: %s 失败, err = %v", filepath.Join(workdir, "upsert.sh"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ make.Seafile: 准备 upsert.sh 成功")
|
|
|
|
// 7. prepare images
|
|
logger.Debug("☑️ make.Seafile: 准备 images")
|
|
imgDir := filepath.Join(m.workdir, "dependency", "image")
|
|
if err = os.MkdirAll(imgDir, 0755); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备 images 目录: %s 失败, err = %v", imgDir, err)
|
|
return err
|
|
}
|
|
var images = []*model.Image{
|
|
{Name: "hub.yizhisec.com/hybridscope/hs_backup_seafile:latest", Fallback: "", Save: "seafile.backup_seafile.tar", Force: true},
|
|
{Name: "hub.yizhisec.com/product/hybridscope/memcached:latest", Fallback: "", Save: "seafile.memcached.tar"},
|
|
{Name: "hub.yizhisec.com/product/hybridscope/seafile-mc:latest", Fallback: "", Save: "seafile.seafile_mc.tar"},
|
|
}
|
|
for _, img := range images {
|
|
img.Save = filepath.Join(imgDir, img.Save)
|
|
if err = m.Image(ctx, img.Name, WithImageSave(img.Save), WithImageForcePull(img.Force)); err != nil {
|
|
logger.Error("❌ make.Seafile: 准备 image: %s 失败, err = %v", img.Name, err)
|
|
return err
|
|
}
|
|
}
|
|
logger.Debug("✅ make.Seafile: 准备 images 成功")
|
|
|
|
logger.Info("✅ maker.Seafile: 构建 seafile 依赖成功!!!")
|
|
|
|
return nil
|
|
}
|