package controllers
|
|
import (
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/logger.git"
|
"github.com/satori/go.uuid"
|
"time"
|
"vamicro/extend/util"
|
"vamicro/system-service/models"
|
"vamicro/system-service/sys"
|
)
|
|
type SysInitController struct {
|
}
|
|
//comment 获取mqtt配置
|
//router /data/api-v/sysinit/getInitInfo
|
func (sic *SysInitController) GetInitInfo(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var conf models.SysInit
|
rows, _ := conf.Select()
|
inited := false
|
initUsername := ""
|
initReg := false
|
if rows > 0 {
|
inited = conf.InitPwd
|
initUsername = conf.InitUsername
|
}
|
if conf.UserType != "" && conf.Name != "" && conf.ProvinceId != "" && conf.CityId != "" && conf.Email != "" {
|
initReg = true
|
}
|
return &bhomeclient.Reply{Success: true, Msg: "查询成功", Data: map[string]interface{}{
|
"initPwd": inited,
|
"initRegInfo": initReg,
|
"username": initUsername,
|
}}
|
}
|
|
//初始化系统密码
|
//router /data/api-v/sysinit/savePassword
|
func (sic *SysInitController) SavePassword(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
type arg struct {
|
Username string `json:"username"`
|
Password string `json:"password"`
|
}
|
var reqBody arg
|
c.BindJSON(&reqBody)
|
if reqBody.Username == "" || reqBody.Password == "" || reqBody.Username == "basic" {
|
return &bhomeclient.Reply{Msg: "用户名或密码不能为空"}
|
}
|
//先保存用户名密码，然后将sysinit中的initPwd置为true
|
encryPwd, pe := util.Encrypt(reqBody.Password)
|
if pe != nil {
|
return &bhomeclient.Reply{Msg: pe.Error()}
|
}
|
tx := models.GetDB().Begin()
|
var err error
|
defer func() {
|
if tx != nil && err != nil {
|
tx.Rollback()
|
}
|
}()
|
|
var tmpUser models.SysUser
|
ru, _ := tmpUser.FindByName(reqBody.Username)
|
targetUID := ""
|
defHeadPic := ""
|
var dhp models.DefHeadPic
|
if headPics, err := dhp.FindAll(); err == nil && len(headPics) > 0 {
|
defHeadPic = headPics[0].Path
|
}
|
|
if ru > 0 {
|
targetUID = tmpUser.Id
|
if err = tx.Exec("update "+tmpUser.TableName()+" set password=?,headpic=?,backgroundpic=? where id=?", encryPwd, defHeadPic, models.DefaultBgPic, targetUID).Error; err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
} else {
|
targetUID = uuid.NewV4().String()
|
if err = tx.Table(tmpUser.TableName()).Create(&models.SysUser{Id: targetUID, Username: reqBody.Username, Password: encryPwd, HeadPic: defHeadPic, BackgroundPic: models.DefaultBgPic}).Error; err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
}
|
//给用户添加菜单权限
|
var sysR models.SysRole
|
_, err = sysR.FindByName("管理员")
|
if err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
|
userRole := models.SysUserRole{
|
Id: uuid.NewV4().String(),
|
UserId: targetUID,
|
RoleId: sysR.Id,
|
}
|
if err = tx.Table(userRole.TableName()).Create(&userRole).Error; err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
|
var siModel models.SysInit
|
err = tx.Table(siModel.TableName()).First(&siModel).Error
|
if err != nil {
|
if err.Error() != models.RecordNotFound {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
} else {
|
if err = tx.Table(siModel.TableName()).Create(models.SysInit{
|
InitPwd: true,
|
InitUsername: reqBody.Username,
|
InitTime: time.Now().Format("2006-01-02 15:04:05"),
|
NetName: "",
|
UserType: "",
|
PhoneNum: "",
|
Name: "",
|
ProvinceId: "",
|
CityId: "",
|
CountyId: "",
|
Email: "",
|
}).Error; err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
}
|
} else {
|
err = tx.Exec("update "+siModel.TableName()+" set initPwd=?, initUsername=?", true, reqBody.Username).Error
|
if err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
}
|
//初始化时给管理员所有的菜单权限
|
var sysMenu models.SysMenu
|
allMenus := sysMenu.FindAll()
|
err = tx.Exec("delete from "+models.SysRoleMenu{}.TableName()+" where role_id=?", sysR.Id).Error
|
if err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
for _, n := range allMenus {
|
tmpMenu := models.SysRoleMenu{
|
Id: uuid.NewV4().String(),
|
RoleId: sysR.Id,
|
MenuId: n.Id,
|
}
|
err = tx.Table(tmpMenu.TableName()).Create(&tmpMenu).Error
|
if err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
}
|
|
err = tx.Exec("delete from "+models.SysUserMenu{}.TableName()+" where user_id=?", targetUID).Error
|
if err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
|
tx.Commit()
|
return &bhomeclient.Reply{Success: true, Msg: "保存成功"}
|
}
|
|
//comment 获取网卡下拉菜单
|
//router /data/api-v/sysinit/networkList
|
func (sic *SysInitController) NetworkList(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
cards, e := sys.GetNetWorkCards()
|
if e != nil {
|
return &bhomeclient.Reply{Msg: e.Error()}
|
}
|
|
return &bhomeclient.Reply{Success: true, Data: cards}
|
}
|
|
//comment 系统第一次启动时初始化网络
|
// router /data/api-v/sysinit/initNetwork
|
func (sic *SysInitController) InitNetwork(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
type arg struct {
|
NetName string `json:"netName"` //网卡
|
Ip string `json:"ip"` //ip
|
SubMask string `json:"subMask"` //子网掩码
|
Gateway string `json:"gateway"` //网关
|
Dns string `json:"dns"` //dns
|
}
|
var reqBody arg
|
c.BindJSON(&reqBody)
|
if reqBody.NetName == "" || reqBody.Ip == "" || reqBody.SubMask == "" || reqBody.Gateway == "" || reqBody.Dns == "" {
|
return &bhomeclient.Reply{Msg: "参数有误，请检查"}
|
}
|
isOk, msg := sys.SetNetworkConfig(
|
reqBody.Ip,
|
reqBody.SubMask,
|
reqBody.Gateway,
|
reqBody.Dns,
|
reqBody.NetName)
|
logger.Debug("isOk:", isOk, "msg:", msg)
|
if !isOk {
|
return &bhomeclient.Reply{Msg: msg}
|
}
|
return &bhomeclient.Reply{Success: true, Msg: "初始化成功"}
|
}
|
|
type RegInfo struct {
|
UserType string `json:"userType"` //personal:个人 company:公司
|
Name string `json:"name"` //姓名
|
ProvinceId string `json:"provinceId"` //省份id
|
CityId string `json:"cityId"` //城市id
|
Addr string `json:"addr"` //地址
|
Email string `json:"email"` //邮箱
|
}
|
|
//comment 保存注册信息
|
//router /data/api-v/sysinit/saveRegInfo
|
func (sic *SysInitController) SaveRegInfo(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var reqBody RegInfo
|
c.BindJSON(&reqBody)
|
if reqBody.UserType == "" || reqBody.Name == "" || reqBody.ProvinceId == "" || reqBody.CityId == "" || reqBody.Email == "" || reqBody.Addr == "" {
|
return &bhomeclient.Reply{Msg: "参数有误"}
|
}
|
var siModel models.SysInit
|
i, _ := siModel.Select()
|
if i == 0 {
|
return &bhomeclient.Reply{Msg: "请先配置用户名和密码信息"}
|
}
|
siModel.UserType = reqBody.UserType
|
siModel.Name = reqBody.Name
|
siModel.ProvinceId = reqBody.ProvinceId
|
siModel.CityId = reqBody.CityId
|
siModel.Email = reqBody.Email
|
if siModel.Update() {
|
return &bhomeclient.Reply{Success: true, Msg: "保存成功"}
|
}
|
return &bhomeclient.Reply{Msg: "保存失败"}
|
}
|
|
//comment 获取注册信息
|
//router /data/api-v/sysinit/getRegInfo
|
func (sic *SysInitController) GetRegInfo(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var siModel models.SysInit
|
i, _ := siModel.Select()
|
if i > 0 {
|
return &bhomeclient.Reply{Success: true, Data: map[string]string{
|
"userType": siModel.UserType,
|
"name": siModel.Name,
|
"provinceId": siModel.ProvinceId,
|
"cityId": siModel.CityId,
|
"countyId": siModel.CountyId,
|
"email": siModel.Email,
|
}}
|
}
|
return &bhomeclient.Reply{Msg: "获取失败"}
|
}
|
