102 lines
3.7 KiB
Go
102 lines
3.7 KiB
Go
package maker
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"gitea.loveuer.com/yizhisec/pkg3/logger"
|
|
"yizhisec.com/hsv2/forge/pkg/downloader"
|
|
"yizhisec.com/hsv2/forge/pkg/resource"
|
|
)
|
|
|
|
type YosguardOpt func(*yosguardOpt)
|
|
|
|
type yosguardOpt struct{}
|
|
|
|
func (m *maker) Yosguard(ctx context.Context, opts ...YosguardOpt) error {
|
|
const (
|
|
_config = `
|
|
AsController: true
|
|
AsGateway: false
|
|
ControllerServer:
|
|
Host: dasheng.zhsftech.debug
|
|
Port: 443
|
|
Database:
|
|
SQLite:
|
|
DBPath: /etc/yosguard/db/yosguard.db
|
|
HeartbeatDuration: 5
|
|
UUIDFilePath: /etc/yosguard/uuid
|
|
Web:
|
|
Host: __ip__
|
|
Port: 7788
|
|
`
|
|
|
|
systemdService = `
|
|
[Unit]
|
|
Description=YiZhiSec YOSGuard
|
|
After=network.target
|
|
|
|
[Service]
|
|
Type=simple
|
|
User=root
|
|
ExecStart=/usr/local/bin/yosguard web --host __host__
|
|
StandardOutput=journal
|
|
StandardError=journal
|
|
Nice=-20
|
|
Restart=always
|
|
RestartSec=15
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target`
|
|
binURL = "https://artifactory.yizhisec.com:443/artifactory/filestore/hsv2/bin/yosguard"
|
|
)
|
|
|
|
location := filepath.Join(m.workdir, "dependency", "yosguard")
|
|
|
|
logger.Info("☑️ maker.Yosguard: 开始构建 yosguard...")
|
|
logger.Debug("☑️ maker.Yosguard: 创建目录 %s", location)
|
|
if err := os.MkdirAll(location, 0755); err != nil {
|
|
logger.Debug("❌ maker.Yosguard: 创建 yosguard 目录失败: %v", err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ maker.Yosguard: 创建目录 %s 成功", location)
|
|
|
|
logger.Debug("☑️ maker.Yosguard: 下载 yosguard 二进制文件..., url = %s, dest = %s", binURL, filepath.Join(location, "yosguard"))
|
|
if err := downloader.Download(
|
|
ctx,
|
|
binURL,
|
|
filepath.Join(location, "yosguard"),
|
|
downloader.WithInsecureSkipVerify(),
|
|
downloader.WithFileMode(0755),
|
|
); err != nil {
|
|
logger.Debug("❌ maker.Yosguard: 下载 yosguard 失败, url = %s, err = %v", binURL, err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ maker.Yosguard: 下载 yosguard 成功, url = %s", binURL)
|
|
|
|
logger.Debug("☑️ maker.Yosguard: 写入 config.yml 文件..., dest = %s", filepath.Join(location, "config.yml"))
|
|
if err := os.WriteFile(filepath.Join(location, "config.yml"), []byte(_config), 0644); err != nil {
|
|
logger.Debug("❌ maker.Yosguard: 写入 config.yml 失败, dest = %s, err = %v", filepath.Join(location, "config.yml"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ maker.Yosguard: 写入 config.yml 文件成功, dest = %s", filepath.Join(location, "config.yml"))
|
|
|
|
logger.Debug("☑️ maker.Yosguard: 写入 create.sql 文件..., dest = %s", filepath.Join(location, "create.sql"))
|
|
if err := os.WriteFile(filepath.Join(location, "create.sql"), resource.SQLYosguard, 0644); err != nil {
|
|
logger.Debug("❌ maker.Yosguard: 写入 create.sql 失败, dest = %s, err = %v", filepath.Join(location, "create.sql"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ maker.Yosguard: 写入 create.sql 文件成功, dest = %s", filepath.Join(location, "create.sql"))
|
|
|
|
logger.Debug("☑️ maker.Yosguard: 写入 yosguard.service 文件..., dest = %s", filepath.Join(location, "yosguard.service"))
|
|
if err := os.WriteFile(filepath.Join(location, "yosguard.service"), []byte(systemdService), 0644); err != nil {
|
|
logger.Debug("❌ maker.Yosguard: 写入 yosguard.service 失败, dest = %s, err = %v", filepath.Join(location, "yosguard.service"), err)
|
|
return err
|
|
}
|
|
logger.Debug("✅ maker.Yosguard: 写入 yosguard.service 文件成功, dest = %s", filepath.Join(location, "yosguard.service"))
|
|
|
|
logger.Info("✅ maker.Yosguard: 构建 yosguard 成功!!!")
|
|
return nil
|
|
}
|