mindoc/commands/install.go

package commands

import (
	"fmt"
	"os"
	"time"

	"github.com/astaxie/beego/orm"
	"github.com/lifei6671/mindoc/conf"
	"github.com/lifei6671/mindoc/models"
	"github.com/lifei6671/mindoc/utils"
	"github.com/astaxie/beego"
)

//系统安装.
func Install() {

	fmt.Println("Initializing...")

	err := orm.RunSyncdb("default", false, true)
	if err == nil {
		initialization()
	} else {
		panic(err.Error())
		os.Exit(1)
	}
	fmt.Println("Install Successfully!")
	os.Exit(0)

}

func Version() {
	if len(os.Args) >= 2 && os.Args[1] == "version" {
		fmt.Println(conf.VERSION)
		os.Exit(0)
	}
}

//修改用户密码
func ModifyPassword(account,password string) {
	if len(os.Args) < 2 {
		fmt.Println("Parameter error.")
		os.Exit(1)
	}

	if account == "" {
		fmt.Println("Account cannot be empty.")
		os.Exit(1)
	}
	if password == "" {
		fmt.Println("Password cannot be empty.")
		os.Exit(1)
	}
	member,err := models.NewMember().FindByAccount(account)

	if err != nil {
		fmt.Println("Failed to change password:",err)
		os.Exit(1)
	}
	pwd,err := utils.PasswordHash(password)

	if err != nil {
		fmt.Println("Failed to change password:",err)
		os.Exit(1)
	}
	member.Password = pwd

	err = member.Update("password")
	if err != nil {
		fmt.Println("Failed to change password:",err)
		os.Exit(1)
	}
	fmt.Println("Successfully modified.")
	os.Exit(0)

}
//初始化数据
func initialization() {

	err := models.NewOption().Init()

	if err != nil {
		beego.Error("初始化全局配置失败 => ",err.Error())
		os.Exit(1)
	}


	InitMemberGroup()


	member, err := models.NewMember().FindByFieldFirst("account", "admin")

	//如果用户不存在，则添加用户，否则需要批量更新用户角色
	if err == orm.ErrNoRows {
		beego.Info("正在创建默认用户.")

		member.Account = "admin"
		member.Avatar = "/static/images/headimgurl.jpg"
		member.Password = "123456"
		member.AuthMethod = "local"
		member.Role = 1
		member.Email = "[email protected]"

		if err := member.Add(); err != nil {
			beego.Error("创建默认用户失败 => " + err.Error())
			os.Exit(0)
		}
		beego.Info("正在创建默认项目.")

		book := models.NewBook()

		book.MemberId = member.MemberId
		book.BookName = "MinDoc演示项目"
		book.Status = 0
		book.Description = "这是一个MinDoc演示项目，该项目是由系统初始化时自动创建。"
		book.CommentCount = 0
		book.PrivatelyOwned = 0
		book.CommentStatus = "closed"
		book.Identify = "mindoc"
		book.DocCount = 0
		book.CommentCount = 0
		book.Version = time.Now().Unix()
		book.Cover = conf.GetDefaultCover()
		book.Editor = "markdown"
		book.Theme = "default"

		if err := book.Insert(); err != nil {
			beego.Error("创建默认项目失败 => " , err)
			os.Exit(1)
		}

	}
}

//初始化用户组
func InitMemberGroup()  {
	beego.Info("正在创建用户组.")

	group := models.NewMemberGroup()

	group.GroupId = 1
	group.GroupName = "管理员组"
	group.GroupNumber = 0
	group.CreateTime = time.Now()
	group.CreateAt = 1
	group.IsEnableDelete = false

	group.InsertOrUpdate()

	if err := group.InsertOrUpdate();err != nil{
		beego.Error("创建用户组失败 ",group.GroupName,err)
		os.Exit(1)
	}


	group.GroupId = 2
	group.GroupName = "普通用户组"
	group.GroupNumber = 0
	group.CreateTime = time.Now()
	group.CreateAt = 1
	group.IsEnableDelete = false

	group.InsertOrUpdate()

	if err := group.InsertOrUpdate();err != nil{
		beego.Error("创建用户组失败 ",group.GroupName,err)
		os.Exit(1)
	}


	group.GroupId = 3
	group.GroupName = "匿名用户组"
	group.GroupNumber = 0
	group.CreateTime = time.Now()
	group.CreateAt = 1
	group.IsEnableDelete = false
	group.Resources = "1,2,3,4,5"

	if err := group.InsertOrUpdate();err != nil{
		beego.Error("创建用户组失败 ",group.GroupName,err)
		os.Exit(1)
	}

}