04f8d0079b
* 优化图标和LOGO * 修改登录页面动画的速度为3 * 增加对websocket的异常处理 * 修复了用户组和用户名唯一判断错误的问题 * 提示版本号 * 修复readme错别字 * 修复单词拼写错误的问题 * 修复代码格式 * 修改Windows资产属性名称 * Docker 打包流程增加 upx 压缩 * 升级依赖文件，修改sqlite驱动为 github.com/glebarez/sqlite * 修复第一次查询「授权令牌」的错误 * 移除无关代码 * 修改docker打包脚本 * 增加打包脚本 * 增加微信群 * 修复单词拼写错误的问题 * 修复代码格式 * 修改Windows资产属性名称 * Docker 打包流程增加 upx 压缩 * 修改docker打包脚本 * - 替换 sqlite 驱动为 github.com/glebarez/sqlite - 修复数据库锁定的问题 - 修复部分代码不完善的问题 - 修复策略显示不完整的问题 - 修复编辑文件换行符的问题 - 优化guacd连接
133 lines
2.8 KiB
Go
133 lines
2.8 KiB
Go
package app
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"next-terminal/server/cli"
|
|
"next-terminal/server/config"
|
|
"next-terminal/server/constant"
|
|
"next-terminal/server/service"
|
|
"next-terminal/server/sshd"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
)
|
|
|
|
var app *App
|
|
|
|
type App struct {
|
|
Server *echo.Echo
|
|
}
|
|
|
|
func newApp() *App {
|
|
return &App{}
|
|
}
|
|
|
|
func init() {
|
|
setupCache()
|
|
app = newApp()
|
|
}
|
|
|
|
func (app App) InitDBData() (err error) {
|
|
if err := service.PropertyService.DeleteDeprecatedProperty(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.GatewayService.ReConnectAll(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.PropertyService.InitProperties(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.UserService.InitUser(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.JobService.InitJob(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.UserService.FixUserOnlineState(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.SessionService.FixSessionState(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.SessionService.EmptyPassword(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.CredentialService.EncryptAll(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.AssetService.EncryptAll(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.StorageService.InitStorages(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 修复数据
|
|
if err := service.AssetService.FixSshMode(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.SessionService.FixSshMode(); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (app App) ReloadData() error {
|
|
if err := service.SecurityService.ReloadAccessSecurity(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.UserService.ReloadToken(); err != nil {
|
|
return err
|
|
}
|
|
if err := service.AccessTokenService.Reload(); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Run() error {
|
|
|
|
fmt.Printf(constant.AppBanner, constant.AppVersion)
|
|
|
|
if err := app.InitDBData(); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := app.ReloadData(); err != nil {
|
|
panic(err)
|
|
}
|
|
app.Server = setupRoutes()
|
|
|
|
if config.GlobalCfg.Debug {
|
|
jsonBytes, err := json.MarshalIndent(config.GlobalCfg, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Printf("当前配置为: %v\n", string(jsonBytes))
|
|
}
|
|
|
|
_cli := cli.NewCli()
|
|
|
|
if config.GlobalCfg.ResetPassword != "" {
|
|
return _cli.ResetPassword(config.GlobalCfg.ResetPassword)
|
|
}
|
|
if config.GlobalCfg.ResetTotp != "" {
|
|
return _cli.ResetTotp(config.GlobalCfg.ResetTotp)
|
|
}
|
|
|
|
if config.GlobalCfg.NewEncryptionKey != "" {
|
|
return _cli.ChangeEncryptionKey(config.GlobalCfg.EncryptionKey, config.GlobalCfg.NewEncryptionKey)
|
|
}
|
|
|
|
if config.GlobalCfg.Sshd.Enable {
|
|
go sshd.Sshd.Serve()
|
|
}
|
|
|
|
if config.GlobalCfg.Server.Cert != "" && config.GlobalCfg.Server.Key != "" {
|
|
return app.Server.StartTLS(config.GlobalCfg.Server.Addr, config.GlobalCfg.Server.Cert, config.GlobalCfg.Server.Key)
|
|
} else {
|
|
return app.Server.Start(config.GlobalCfg.Server.Addr)
|
|
}
|
|
}
|