f99f49dd7e
* switch branches * update user interface * switch branch * switch branch * change branch * submit 0.3.3
291 lines
8.2 KiB
Go
291 lines
8.2 KiB
Go
package route
|
||
|
||
import (
|
||
"encoding/xml"
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
"runtime"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/IceWhaleTech/CasaOS/model"
|
||
"github.com/IceWhaleTech/CasaOS/model/system_app"
|
||
"github.com/IceWhaleTech/CasaOS/pkg/config"
|
||
"github.com/IceWhaleTech/CasaOS/pkg/utils/command"
|
||
"github.com/IceWhaleTech/CasaOS/pkg/utils/encryption"
|
||
"github.com/IceWhaleTech/CasaOS/pkg/utils/env_helper"
|
||
"github.com/IceWhaleTech/CasaOS/pkg/utils/file"
|
||
"github.com/IceWhaleTech/CasaOS/pkg/utils/port"
|
||
"github.com/IceWhaleTech/CasaOS/service"
|
||
model2 "github.com/IceWhaleTech/CasaOS/service/model"
|
||
uuid "github.com/satori/go.uuid"
|
||
)
|
||
|
||
func InitFunction() {
|
||
go checkSystemApp()
|
||
Update2_3()
|
||
CheckSerialDiskMount()
|
||
|
||
CheckToken2_11()
|
||
ImportApplications()
|
||
ChangeAPIUrl()
|
||
|
||
MoveUserToDB()
|
||
}
|
||
|
||
var syncIsExistence = false
|
||
|
||
func installSyncthing(appId string) {
|
||
|
||
var appInfo model.ServerAppList
|
||
m := model.CustomizationPostData{}
|
||
var dockerImage string
|
||
var dockerImageVersion string
|
||
appInfo = service.MyService.Casa().GetServerAppInfo(appId, "system", "us_en")
|
||
dockerImage = appInfo.Image
|
||
dockerImageVersion = appInfo.ImageVersion
|
||
|
||
if len(appInfo.ImageVersion) == 0 {
|
||
dockerImageVersion = "latest"
|
||
}
|
||
|
||
if appInfo.NetworkModel != "host" {
|
||
for i := 0; i < len(appInfo.Ports); i++ {
|
||
if p, _ := strconv.Atoi(appInfo.Ports[i].ContainerPort); port.IsPortAvailable(p, appInfo.Ports[i].Protocol) {
|
||
appInfo.Ports[i].CommendPort = strconv.Itoa(p)
|
||
} else {
|
||
if appInfo.Ports[i].Protocol == "tcp" {
|
||
if p, err := port.GetAvailablePort("tcp"); err == nil {
|
||
appInfo.Ports[i].CommendPort = strconv.Itoa(p)
|
||
}
|
||
} else if appInfo.Ports[i].Protocol == "upd" {
|
||
if p, err := port.GetAvailablePort("udp"); err == nil {
|
||
appInfo.Ports[i].CommendPort = strconv.Itoa(p)
|
||
}
|
||
}
|
||
}
|
||
|
||
if appInfo.Ports[i].Type == 0 {
|
||
appInfo.PortMap = appInfo.Ports[i].CommendPort
|
||
}
|
||
}
|
||
}
|
||
|
||
for i := 0; i < len(appInfo.Devices); i++ {
|
||
if !file.CheckNotExist(appInfo.Devices[i].ContainerPath) {
|
||
appInfo.Devices[i].Path = appInfo.Devices[i].ContainerPath
|
||
}
|
||
}
|
||
if len(appInfo.Tip) > 0 {
|
||
appInfo.Tip = env_helper.ReplaceStringDefaultENV(appInfo.Tip)
|
||
}
|
||
|
||
appInfo.MaxMemory = service.MyService.System().GetMemInfo()["total"].(uint64) >> 20
|
||
|
||
id := uuid.NewV4().String()
|
||
|
||
// step：下载镜像
|
||
err := service.MyService.Docker().DockerPullImage(dockerImage+":"+dockerImageVersion, "", "")
|
||
if err != nil {
|
||
//pull image error
|
||
fmt.Println("pull image error", err, dockerImage, dockerImageVersion)
|
||
return
|
||
}
|
||
for !service.MyService.Docker().IsExistImage(dockerImage + ":" + dockerImageVersion) {
|
||
time.Sleep(time.Second)
|
||
}
|
||
|
||
m.CpuShares = 50
|
||
m.Envs = appInfo.Envs
|
||
m.Memory = int64(appInfo.MaxMemory)
|
||
m.Origin = "system"
|
||
m.PortMap = appInfo.PortMap
|
||
m.Ports = appInfo.Ports
|
||
m.Restart = "always"
|
||
m.Volumes = appInfo.Volumes
|
||
m.NetworkModel = appInfo.NetworkModel
|
||
m.Label = id
|
||
m.CustomId = id
|
||
containerId, err := service.MyService.Docker().DockerContainerCreate(dockerImage+":"+dockerImageVersion, m)
|
||
if err != nil {
|
||
fmt.Println("container create error", err)
|
||
// create container error
|
||
return
|
||
}
|
||
|
||
//step：start container
|
||
err = service.MyService.Docker().DockerContainerStart(containerId)
|
||
if err != nil {
|
||
//start container error
|
||
return
|
||
}
|
||
|
||
checkSystemApp()
|
||
}
|
||
|
||
// check if the system application is installed
|
||
func checkSystemApp() {
|
||
list := service.MyService.App().GetSystemAppList()
|
||
for _, v := range list {
|
||
info, err := service.MyService.Docker().DockerContainerInfo(v.ID)
|
||
if err != nil {
|
||
continue
|
||
}
|
||
if strings.Contains(info.Config.Image, "linuxserver/syncthing") {
|
||
if v.State != "running" {
|
||
//step：start container
|
||
service.MyService.Docker().DockerContainerStart(v.ID)
|
||
}
|
||
syncIsExistence = true
|
||
if config.SystemConfigInfo.SyncPort != v.Labels["web"] {
|
||
config.SystemConfigInfo.SyncPort = v.Labels["web"]
|
||
}
|
||
|
||
path := ""
|
||
for _, i := range info.Mounts {
|
||
if i.Destination == "/config" {
|
||
path = i.Source
|
||
break
|
||
}
|
||
}
|
||
content := file.ReadFullFile(filepath.Join(path, "config.xml"))
|
||
syncConfig := &system_app.SyncConfig{}
|
||
xml.Unmarshal(content, &syncConfig)
|
||
config.SystemConfigInfo.SyncKey = syncConfig.Key
|
||
break
|
||
}
|
||
}
|
||
if !syncIsExistence {
|
||
installSyncthing("74")
|
||
}
|
||
}
|
||
func CheckSerialDiskMount() {
|
||
// check mount point
|
||
dbList := service.MyService.Disk().GetSerialAll()
|
||
|
||
list := service.MyService.Disk().LSBLK(true)
|
||
mountPoint := make(map[string]string, len(dbList))
|
||
//remount
|
||
for _, v := range dbList {
|
||
mountPoint[v.UUID] = v.MountPoint
|
||
}
|
||
for _, v := range list {
|
||
command.ExecEnabledSMART(v.Path)
|
||
if v.Children != nil {
|
||
for _, h := range v.Children {
|
||
if len(h.MountPoint) == 0 && len(v.Children) == 1 && h.FsType == "ext4" {
|
||
if m, ok := mountPoint[h.UUID]; ok {
|
||
//mount point check
|
||
volume := m
|
||
if !file.CheckNotExist(m) {
|
||
for i := 0; file.CheckNotExist(volume); i++ {
|
||
volume = m + strconv.Itoa(i+1)
|
||
}
|
||
}
|
||
service.MyService.Disk().MountDisk(h.Path, volume)
|
||
if volume != m {
|
||
ms := model2.SerialDisk{}
|
||
ms.UUID = v.UUID
|
||
ms.MountPoint = volume
|
||
service.MyService.Disk().UpdateMountPoint(ms)
|
||
}
|
||
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
service.MyService.Disk().RemoveLSBLKCache()
|
||
command.OnlyExec("source " + config.AppInfo.ShellPath + "/helper.sh ;AutoRemoveUnuseDir")
|
||
}
|
||
func Update2_3() {
|
||
command.OnlyExec("source " + config.AppInfo.ShellPath + "/assist.sh")
|
||
|
||
}
|
||
func CheckToken2_11() {
|
||
if len(config.ServerInfo.Token) == 0 {
|
||
token := uuid.NewV4().String
|
||
config.ServerInfo.Token = token()
|
||
config.Cfg.Section("server").Key("Token").SetValue(token())
|
||
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
||
}
|
||
|
||
sysType := runtime.GOOS
|
||
if len(config.FileSettingInfo.DownloadDir) == 0 {
|
||
downloadPath := "/DATA/Downloads"
|
||
if sysType == "windows" {
|
||
downloadPath = "C:\\CasaOS\\DATA\\Downloads"
|
||
}
|
||
if sysType == "darwin" {
|
||
downloadPath = "./CasaOS/DATA/Downloads"
|
||
}
|
||
config.Cfg.Section("file").Key("DownloadDir").SetValue(downloadPath)
|
||
config.FileSettingInfo.DownloadDir = downloadPath
|
||
file.IsNotExistMkDir(config.FileSettingInfo.DownloadDir)
|
||
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
||
}
|
||
|
||
if len(config.UserInfo.Description) == 0 {
|
||
config.Cfg.Section("user").Key("Description").SetValue("nothing")
|
||
config.UserInfo.Description = "nothing"
|
||
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
||
}
|
||
if len(config.ServerInfo.Handshake) == 0 {
|
||
config.Cfg.Section("server").Key("Handshake").SetValue("socket.casaos.io")
|
||
config.ServerInfo.Handshake = "socket.casaos.io"
|
||
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
||
}
|
||
|
||
if service.MyService.System().GetSysInfo().KernelArch == "aarch64" && config.ServerInfo.USBAutoMount != "True" && strings.Contains(service.MyService.System().GetDeviceTree(), "Raspberry Pi") {
|
||
service.MyService.System().UpdateUSBAutoMount("False")
|
||
service.MyService.System().ExecUSBAutoMountShell("False")
|
||
}
|
||
|
||
// str := []string{}
|
||
// str = append(str, "ddd")
|
||
// str = append(str, "aaa")
|
||
// ddd := strings.Join(str, "|")
|
||
// config.Cfg.Section("file").Key("ShareDir").SetValue(ddd)
|
||
|
||
// config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
||
|
||
}
|
||
|
||
func ImportApplications() {
|
||
service.MyService.App().ImportApplications(true)
|
||
}
|
||
|
||
// 0.3.1
|
||
func ChangeAPIUrl() {
|
||
|
||
newAPIUrl := "https://api.casaos.io/casaos-api"
|
||
if config.ServerInfo.ServerApi == "https://api.casaos.zimaboard.com" {
|
||
config.ServerInfo.ServerApi = newAPIUrl
|
||
config.Cfg.Section("server").Key("ServerApi").SetValue(newAPIUrl)
|
||
config.Cfg.SaveTo(config.SystemConfigInfo.ConfigPath)
|
||
}
|
||
|
||
}
|
||
|
||
//0.3.3
|
||
//Transferring user data to the database
|
||
func MoveUserToDB() {
|
||
|
||
if len(config.UserInfo.UserName) > 0 && service.MyService.User().GetUserInfoByUserName(config.UserInfo.UserName).Id == 0 {
|
||
user := model2.UserDBModel{}
|
||
user.UserName = config.UserInfo.UserName
|
||
user.Email = config.UserInfo.Email
|
||
user.NickName = config.UserInfo.NickName
|
||
user.Password = encryption.GetMD5ByStr(config.UserInfo.PWD)
|
||
user.Role = "admin"
|
||
user = service.MyService.User().CreateUser(user)
|
||
if user.Id > 0 {
|
||
userPath := config.AppInfo.UserDataPath + "/" + strconv.Itoa(user.Id)
|
||
file.MkDir(userPath)
|
||
os.Rename("/casaOS/server/conf/app_order.json", userPath+"/app_order.json")
|
||
}
|
||
|
||
}
|
||
}
|