package controllers import ( "encoding/json" "html/template" "regexp" "strings" "math" "path/filepath" "strconv" "github.com/astaxie/beego" "github.com/astaxie/beego/logs" "github.com/astaxie/beego/orm" "github.com/lifei6671/mindoc/conf" "github.com/lifei6671/mindoc/models" "github.com/lifei6671/mindoc/utils" "github.com/lifei6671/mindoc/utils/filetil" "github.com/lifei6671/mindoc/utils/pagination" "gopkg.in/russross/blackfriday.v2" "io/ioutil" "os" ) type ManagerController struct { BaseController } func (c *ManagerController) Prepare() { c.BaseController.Prepare() if !c.Member.IsAdministrator() { c.Abort("403") } } func (c *ManagerController) Index() { c.TplName = "manager/index.tpl" c.Data["Model"] = models.NewDashboard().Query() } // 用户列表. func (c *ManagerController) Users() { c.Prepare() c.TplName = "manager/users.tpl" pageIndex, _ := c.GetInt("page", 0) members, totalCount, err := models.NewMember().FindToPager(pageIndex, conf.PageSize) if err != nil { c.Data["ErrorMessage"] = err.Error() return } if totalCount > 0 { pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() for _, item := range members { item.Avatar = conf.URLForWithCdnImage(item.Avatar) } } else { c.Data["PageHtml"] = "" } b, err := json.Marshal(members) if err != nil { c.Data["Result"] = template.JS("[]") } else { c.Data["Result"] = template.JS(string(b)) } } // 添加用户. func (c *ManagerController) CreateMember() { c.Prepare() account := strings.TrimSpace(c.GetString("account")) password1 := strings.TrimSpace(c.GetString("password1")) password2 := strings.TrimSpace(c.GetString("password2")) email := strings.TrimSpace(c.GetString("email")) phone := strings.TrimSpace(c.GetString("phone")) role, _ := c.GetInt("role", 1) status, _ := c.GetInt("status", 0) if ok, err := regexp.MatchString(conf.RegexpAccount, account); account == "" || !ok || err != nil { c.JsonResult(6001, "账号只能由英文字母数字组成,且在3-50个字符") } if l := strings.Count(password1, ""); password1 == "" || l > 50 || l < 6 { c.JsonResult(6002, "密码必须在6-50个字符之间") } if password1 != password2 { c.JsonResult(6003, "确认密码不正确") } if ok, err := regexp.MatchString(conf.RegexpEmail, email); !ok || err != nil || email == "" { c.JsonResult(6004, "邮箱格式不正确") } if role != 0 && role != 1 && role != 2 { role = 1 } if status != 0 && status != 1 { status = 0 } member := models.NewMember() if _, err := member.FindByAccount(account); err == nil && member.MemberId > 0 { c.JsonResult(6005, "账号已存在") } member.Account = account member.Password = password1 member.Role = conf.SystemRole(role) member.Avatar = conf.GetDefaultAvatar() member.CreateAt = c.Member.MemberId member.Email = email member.RealName = strings.TrimSpace(c.GetString("real_name", "")) if phone != "" { member.Phone = phone } if err := member.Add(); err != nil { c.JsonResult(6006, err.Error()) } c.JsonResult(0, "ok", member) } //更新用户状态. func (c *ManagerController) UpdateMemberStatus() { c.Prepare() member_id, _ := c.GetInt("member_id", 0) status, _ := c.GetInt("status", 0) if member_id <= 0 { c.JsonResult(6001, "参数错误") } if status != 0 && status != 1 { status = 0 } member := models.NewMember() if _, err := member.Find(member_id); err != nil { c.JsonResult(6002, "用户不存在") } if member.MemberId == c.Member.MemberId { c.JsonResult(6004, "不能变更自己的状态") } if member.Role == conf.MemberSuperRole { c.JsonResult(6005, "不能变更超级管理员的状态") } member.Status = status if err := member.Update(); err != nil { logs.Error("", err) c.JsonResult(6003, "用户状态设置失败") } c.JsonResult(0, "ok", member) } //变更用户权限. func (c *ManagerController) ChangeMemberRole() { c.Prepare() memberId, _ := c.GetInt("member_id", 0) role, _ := c.GetInt("role", 0) if memberId <= 0 { c.JsonResult(6001, "参数错误") } if role != int(conf.MemberAdminRole) && role != int(conf.MemberGeneralRole) { c.JsonResult(6001, "用户权限不正确") } member := models.NewMember() if _, err := member.Find(memberId); err != nil { c.JsonResult(6002, "用户不存在") } if member.MemberId == c.Member.MemberId { c.JsonResult(6004, "不能变更自己的权限") } if member.Role == conf.MemberSuperRole { c.JsonResult(6005, "不能变更超级管理员的权限") } member.Role = conf.SystemRole(role) if err := member.Update(); err != nil { c.JsonResult(6003, "用户权限设置失败") } member.ResolveRoleName() c.JsonResult(0, "ok", member) } //编辑用户信息. func (c *ManagerController) EditMember() { c.Prepare() c.TplName = "manager/edit_users.tpl" member_id, _ := c.GetInt(":id", 0) if member_id <= 0 { c.Abort("404") } member, err := models.NewMember().Find(member_id) if err != nil { beego.Error(err) c.Abort("404") } if c.Ctx.Input.IsPost() { password1 := c.GetString("password1") password2 := c.GetString("password2") email := c.GetString("email") phone := c.GetString("phone") description := c.GetString("description") member.Email = email member.Phone = phone member.Description = description member.RealName = c.GetString("real_name") if password1 != "" && password2 != password1 { c.JsonResult(6001, "确认密码不正确") } if password1 != "" && member.AuthMethod != conf.AuthMethodLDAP { member.Password = password1 } if err := member.Valid(password1 == ""); err != nil { c.JsonResult(6002, err.Error()) } if password1 != "" { password, err := utils.PasswordHash(password1) if err != nil { beego.Error(err) c.JsonResult(6003, "对用户密码加密时出错") } member.Password = password } if err := member.Update(); err != nil { c.JsonResult(6004, err.Error()) } c.JsonResult(0, "ok") } c.Data["Model"] = member } //删除一个用户,并将该用户的所有信息转移到超级管理员上. func (c *ManagerController) DeleteMember() { c.Prepare() member_id, _ := c.GetInt("id", 0) if member_id <= 0 { c.JsonResult(404, "参数错误") } member, err := models.NewMember().Find(member_id) if err != nil { beego.Error(err) c.JsonResult(500, "用户不存在") } if member.Role == conf.MemberSuperRole { c.JsonResult(500, "不能删除超级管理员") } superMember, err := models.NewMember().FindByFieldFirst("role", 0) if err != nil { beego.Error(err) c.JsonResult(5001, "未能找到超级管理员") } err = models.NewMember().Delete(member_id, superMember.MemberId) if err != nil { beego.Error(err) c.JsonResult(5002, "删除失败") } c.JsonResult(0, "ok") } //项目列表. func (c *ManagerController) Books() { c.Prepare() c.TplName = "manager/books.tpl" pageIndex, _ := c.GetInt("page", 1) books, totalCount, err := models.NewBookResult().FindToPager(pageIndex, conf.PageSize) if err != nil { c.Abort("500") } if totalCount > 0 { //html := utils.GetPagerHtml(c.Ctx.Request.RequestURI, pageIndex, 8, totalCount) pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() } else { c.Data["PageHtml"] = "" } for i, book := range books { books[i].Description = utils.StripTags(string(blackfriday.Run([]byte(book.Description)))) books[i].ModifyTime = book.ModifyTime.Local() books[i].CreateTime = book.CreateTime.Local() } c.Data["Lists"] = books } //编辑项目. func (c *ManagerController) EditBook() { c.Prepare() c.TplName = "manager/edit_book.tpl" identify := c.GetString(":key") if identify == "" { c.Abort("404") } book, err := models.NewBook().FindByFieldFirst("identify", identify) if err != nil { c.Abort("500") } if c.Ctx.Input.IsPost() { bookName := strings.TrimSpace(c.GetString("book_name")) description := strings.TrimSpace(c.GetString("description", "")) commentStatus := c.GetString("comment_status") tag := strings.TrimSpace(c.GetString("label")) orderIndex, _ := c.GetInt("order_index", 0) isDownload := strings.TrimSpace(c.GetString("is_download")) == "on" enableShare := strings.TrimSpace(c.GetString("enable_share")) == "on" isUseFirstDocument := strings.TrimSpace(c.GetString("is_use_first_document")) == "on" autoRelease := strings.TrimSpace(c.GetString("auto_release")) == "on" publisher := strings.TrimSpace(c.GetString("publisher")) historyCount, _ := c.GetInt("history_count", 0) itemId,_ := c.GetInt("itemId") if strings.Count(description, "") > 500 { c.JsonResult(6004, "项目描述不能大于500字") } if commentStatus != "open" && commentStatus != "closed" && commentStatus != "group_only" && commentStatus != "registered_only" { commentStatus = "closed" } if tag != "" { tags := strings.Split(tag, ";") if len(tags) > 10 { c.JsonResult(6005, "最多允许添加10个标签") } } if !models.NewItemsets().Exist(itemId) { c.JsonResult(6006,"项目集不存在") } book.Publisher = publisher book.HistoryCount = historyCount book.BookName = bookName book.Description = description book.CommentStatus = commentStatus book.Label = tag book.OrderIndex = orderIndex book.ItemId = itemId book.BookPassword = strings.TrimSpace(c.GetString("bPassword")) if autoRelease { book.AutoRelease = 1 } else { book.AutoRelease = 0 } if isDownload { book.IsDownload = 0 } else { book.IsDownload = 1 } if enableShare { book.IsEnableShare = 0 } else { book.IsEnableShare = 1 } if isUseFirstDocument { book.IsUseFirstDocument = 1 } else { book.IsUseFirstDocument = 0 } if err := book.Update(); err != nil { c.JsonResult(6006, "保存失败") } c.JsonResult(0, "ok") } if book.PrivateToken != "" { book.PrivateToken = conf.URLFor("DocumentController.Index", ":key", book.Identify, "token", book.PrivateToken) } bookResult := models.NewBookResult() bookResult.ToBookResult(*book) c.Data["Model"] = bookResult } // 删除项目. func (c *ManagerController) DeleteBook() { c.Prepare() bookId, _ := c.GetInt("book_id", 0) if bookId <= 0 { c.JsonResult(6001, "参数错误") } book := models.NewBook() err := book.ThoroughDeleteBook(bookId) if err == orm.ErrNoRows { c.JsonResult(6002, "项目不存在") } if err != nil { logs.Error("删除失败 -> ", err) c.JsonResult(6003, "删除失败") } c.JsonResult(0, "ok") } // CreateToken 创建访问来令牌. func (c *ManagerController) CreateToken() { c.Prepare() action := c.GetString("action") identify := c.GetString("identify") book, err := models.NewBook().FindByFieldFirst("identify", identify) if err != nil { c.JsonResult(6001, "项目不存在") } if action == "create" { if book.PrivatelyOwned == 0 { c.JsonResult(6001, "公开项目不能创建阅读令牌") } book.PrivateToken = string(utils.Krand(conf.GetTokenSize(), utils.KC_RAND_KIND_ALL)) if err := book.Update(); err != nil { logs.Error("生成阅读令牌失败 => ", err) c.JsonResult(6003, "生成阅读令牌失败") } c.JsonResult(0, "ok", conf.URLFor("DocumentController.Index", ":key", book.Identify, "token", book.PrivateToken)) } else { book.PrivateToken = "" if err := book.Update(); err != nil { logs.Error("CreateToken => ", err) c.JsonResult(6004, "删除令牌失败") } c.JsonResult(0, "ok", "") } } //项目设置. func (c *ManagerController) Setting() { c.Prepare() c.TplName = "manager/setting.tpl" options, err := models.NewOption().All() if c.Ctx.Input.IsPost() { for _, item := range options { item.OptionValue = c.GetString(item.OptionName) item.InsertOrUpdate() } c.JsonResult(0, "ok") } if err != nil { c.Abort("500") } c.Data["SITE_TITLE"] = c.Option["SITE_NAME"] for _, item := range options { c.Data[item.OptionName] = item.OptionValue } } // Transfer 转让项目. func (c *ManagerController) Transfer() { c.Prepare() account := c.GetString("account") if account == "" { c.JsonResult(6004, "接受者账号不能为空") } member, err := models.NewMember().FindByAccount(account) if err != nil { logs.Error("FindByAccount => ", err) c.JsonResult(6005, "接受用户不存在") } if member.Status != 0 { c.JsonResult(6006, "接受用户已被禁用") } if !c.Member.IsAdministrator() { c.Abort("403") } identify := c.GetString("identify") book, err := models.NewBook().FindByFieldFirst("identify", identify) if err != nil { c.JsonResult(6001, err.Error()) } rel, err := models.NewRelationship().FindFounder(book.BookId) if err != nil { beego.Error("FindFounder => ", err) c.JsonResult(6009, "查询项目创始人失败") } if member.MemberId == rel.MemberId { c.JsonResult(6007, "不能转让给自己") } err = models.NewRelationship().Transfer(book.BookId, rel.MemberId, member.MemberId) if err != nil { logs.Error("Transfer => ", err) c.JsonResult(6008, err.Error()) } c.JsonResult(0, "ok") } func (c *ManagerController) Comments() { c.Prepare() c.TplName = "manager/comments.tpl" if !c.Member.IsAdministrator() { c.Abort("403") } } //DeleteComment 标记评论为已删除 func (c *ManagerController) DeleteComment() { c.Prepare() comment_id, _ := c.GetInt("comment_id", 0) if comment_id <= 0 { c.JsonResult(6001, "参数错误") } comment := models.NewComment() if _, err := comment.Find(comment_id); err != nil { c.JsonResult(6002, "评论不存在") } comment.Approved = 3 if err := comment.Update("approved"); err != nil { c.JsonResult(6003, "删除评论失败") } c.JsonResult(0, "ok", comment) } //设置项目私有状态. func (c *ManagerController) PrivatelyOwned() { c.Prepare() status := c.GetString("status") identify := c.GetString("identify") if status != "open" && status != "close" { c.JsonResult(6003, "参数错误") } state := 0 if status == "open" { state = 0 } else { state = 1 } if !c.Member.IsAdministrator() { c.Abort("403") } book, err := models.NewBook().FindByFieldFirst("identify", identify) if err != nil { c.JsonResult(6001, err.Error()) } book.PrivatelyOwned = state logs.Info("", state, status) err = book.Update() if err != nil { logs.Error("PrivatelyOwned => ", err) c.JsonResult(6004, "保存失败") } c.JsonResult(0, "ok") } //附件列表. func (c *ManagerController) AttachList() { c.Prepare() c.TplName = "manager/attach_list.tpl" pageIndex, _ := c.GetInt("page", 1) attachList, totalCount, err := models.NewAttachment().FindToPager(pageIndex, conf.PageSize) if err != nil { c.Abort("500") } if totalCount > 0 { pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() } else { c.Data["PageHtml"] = "" } for _, item := range attachList { p := filepath.Join(conf.WorkingDirectory, item.FilePath) item.IsExist = filetil.FileExists(p) } c.Data["Lists"] = attachList } //附件详情. func (c *ManagerController) AttachDetailed() { c.Prepare() c.TplName = "manager/attach_detailed.tpl" attach_id, _ := strconv.Atoi(c.Ctx.Input.Param(":id")) if attach_id <= 0 { c.Abort("404") } attach, err := models.NewAttachmentResult().Find(attach_id) if err != nil { beego.Error("AttachDetailed => ", err) if err == orm.ErrNoRows { c.Abort("404") } else { c.Abort("500") } } attach.FilePath = filepath.Join(conf.WorkingDirectory, attach.FilePath) attach.HttpPath = conf.URLForWithCdnImage(attach.HttpPath) attach.IsExist = filetil.FileExists(attach.FilePath) c.Data["Model"] = attach } //删除附件. func (c *ManagerController) AttachDelete() { c.Prepare() attachId, _ := c.GetInt("attach_id") if attachId <= 0 { c.Abort("404") } attach, err := models.NewAttachment().Find(attachId) if err != nil { beego.Error("AttachDelete => ", err) c.JsonResult(6001, err.Error()) } attach.FilePath = filepath.Join(conf.WorkingDirectory, attach.FilePath) if err := attach.Delete(); err != nil { beego.Error("AttachDelete => ", err) c.JsonResult(6002, err.Error()) } c.JsonResult(0, "ok") } //标签列表 func (c *ManagerController) LabelList() { c.Prepare() c.TplName = "manager/label_list.tpl" pageIndex, _ := c.GetInt("page", 1) labels, totalCount, err := models.NewLabel().FindToPager(pageIndex, conf.PageSize) if err != nil { c.ShowErrorPage(50001, err.Error()) } if totalCount > 0 { pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() } else { c.Data["PageHtml"] = "" } c.Data["TotalPages"] = int(math.Ceil(float64(totalCount) / float64(conf.PageSize))) c.Data["Lists"] = labels } //删除标签 func (c *ManagerController) LabelDelete() { labelId, err := strconv.Atoi(c.Ctx.Input.Param(":id")) if err != nil { beego.Error("获取删除标签参数时出错:", err) c.JsonResult(50001, "参数错误") } if labelId <= 0 { c.JsonResult(50001, "参数错误") } label, err := models.NewLabel().FindFirst("label_id", labelId) if err != nil { beego.Error("查询标签时出错:", err) c.JsonResult(50001, "查询标签时出错:"+err.Error()) } if err := label.Delete(); err != nil { c.JsonResult(50002, "删除失败:"+err.Error()) } else { c.JsonResult(0, "ok") } } func (c *ManagerController) Config() { c.Prepare() c.TplName = "manager/config.tpl" if c.Ctx.Input.IsPost() { content := strings.TrimSpace(c.GetString("configFileTextArea")) if content == "" { c.JsonResult(500, "配置文件不能为空") } tf, err := ioutil.TempFile(os.TempDir(), "mindoc") if err != nil { beego.Error("创建临时文件失败 ->", err) c.JsonResult(5001, "创建临时文件失败") } defer tf.Close() tf.WriteString(content) err = beego.LoadAppConfig("ini", tf.Name()) if err != nil { beego.Error("加载配置文件失败 ->", err) c.JsonResult(5002, "加载配置文件失败") } err = filetil.CopyFile(tf.Name(), conf.ConfigurationFile) if err != nil { beego.Error("保存配置文件失败 ->", err) c.JsonResult(5003, "保存配置文件失败") } c.JsonResult(0, "保存成功") } c.Data["ConfigContent"] = "" if b, err := ioutil.ReadFile(conf.ConfigurationFile); err == nil { c.Data["ConfigContent"] = string(b) } } func (c *ManagerController) Team() { c.Prepare() c.TplName = "manager/team.tpl" pageIndex, _ := c.GetInt("page", 0) teams, totalCount, err := models.NewTeam().FindToPager(pageIndex, conf.PageSize) if err != nil && err != orm.ErrNoRows { c.ShowErrorPage(500, err.Error()) } if err == orm.ErrNoRows || len(teams) <= 0 { c.Data["Result"] = template.JS("[]") c.Data["PageHtml"] = "" return } if totalCount > 0 { pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() } else { c.Data["PageHtml"] = "" } b, err := json.Marshal(teams) if err != nil { c.Data["Result"] = template.JS("[]") } else { c.Data["Result"] = template.JS(string(b)) } } func (c *ManagerController) TeamCreate() { c.Prepare() teamName := c.GetString("teamName") if teamName == "" { c.JsonResult(5001, "团队名称不能为空") } team := models.NewTeam() team.MemberId = c.Member.MemberId team.TeamName = teamName if err := team.Save(); err == nil { c.JsonResult(0, "OK", team) } else { c.JsonResult(5002, err.Error()) } } func (c *ManagerController) TeamEdit() { c.Prepare() teamName := c.GetString("teamName") teamId, _ := c.GetInt("teamId") if teamName == "" { c.JsonResult(5001, "团队名称不能为空") } if teamId <= 0 { c.JsonResult(5002, "团队标识不能为空") } team, err := models.NewTeam().First(teamId) c.CheckJsonError(5003, err) team.TeamName = teamName err = team.Save() c.CheckJsonError(5004, err) c.JsonResult(0, "OK", team) } func (c *ManagerController) TeamDelete() { c.Prepare() teamId, _ := c.GetInt("teamId") if teamId <= 0 { c.JsonResult(5002, "团队标识不能为空") } err := models.NewTeam().Delete(teamId) c.CheckJsonError(5001, err) c.JsonResult(0, "OK") } func (c *ManagerController) TeamMemberList() { c.Prepare() c.TplName = "manager/team_member_list.tpl" teamId, _ := strconv.Atoi(c.Ctx.Input.Param(":id")) pageIndex, _ := c.GetInt("page", 0) if teamId <= 0 { c.ShowErrorPage(500, "参数错误") } team, err := models.NewTeam().First(teamId) if err == orm.ErrNoRows { c.ShowErrorPage(404, "团队不存在") } c.CheckErrorResult(500, err) c.Data["Model"] = team teams, totalCount, err := models.NewTeamMember().FindToPager(teamId, pageIndex, conf.PageSize) if err != nil && err != orm.ErrNoRows { c.ShowErrorPage(500, err.Error()) } if err == orm.ErrNoRows || len(teams) <= 0 { c.Data["Result"] = template.JS("[]") c.Data["PageHtml"] = "" return } if totalCount > 0 { pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() } else { c.Data["PageHtml"] = "" } b, err := json.Marshal(teams) if err != nil { beego.Error("编码 JSON 结果失败 ->", err) c.Data["Result"] = template.JS("[]") } else { c.Data["Result"] = template.JS(string(b)) } } //搜索团队用户. func (c *ManagerController) TeamSearchMember() { c.Prepare() teamId, _ := c.GetInt("teamId") keyword := strings.TrimSpace(c.GetString("q")) if teamId <= 0 { c.JsonResult(500, "参数错误") } searchResult, err := models.NewTeamMember().FindNotJoinMemberByAccount(teamId, keyword, 10) if err != nil { c.JsonResult(500, err.Error()) } c.JsonResult(0, "OK", searchResult) } func (c *ManagerController) TeamMemberAdd() { c.Prepare() teamId, _ := c.GetInt("teamId") memberId, _ := c.GetInt("memberId") roleId, _ := c.GetInt("roleId") if teamId <= 0 || memberId <= 0 || roleId <= 0 || roleId > int(conf.BookObserver) { c.JsonResult(5001, "参数不正确") } teamMember := models.NewTeamMember() teamMember.MemberId = memberId teamMember.TeamId = teamId teamMember.RoleId = conf.BookRole(roleId) if err := teamMember.Save(); err != nil { c.CheckJsonError(5001, err) } teamMember.Include() c.JsonResult(0, "OK", teamMember) } func (c *ManagerController) TeamMemberDelete() { c.Prepare() memberId, _ := c.GetInt("memberId") teamId, _ := c.GetInt("teamId") teamMember, err := models.NewTeamMember().FindFirst(teamId, memberId) if err != nil { c.JsonResult(5001, "用户不存在或已禁用") } err = teamMember.Delete(teamMember.TeamMemberId) if err != nil { c.JsonResult(5002, "删除失败") } c.JsonResult(0, "ok") } func (c *ManagerController) TeamChangeMemberRole() { c.Prepare() memberId, _ := c.GetInt("memberId") roleId, _ := c.GetInt("roleId") teamId, _ := c.GetInt("teamId") if memberId <= 0 || roleId <= 0 || teamId <= 0 || roleId > int(conf.BookObserver) { c.JsonResult(5001, "参数错误") } teamMember, err := models.NewTeamMember().ChangeRoleId(teamId, memberId, conf.BookRole(roleId)) if err != nil { c.JsonResult(5002, err.Error()) } else { c.JsonResult(0, "OK", teamMember) } } //团队项目列表. func (c *ManagerController) TeamBookList() { c.Prepare() c.TplName = "manager/team_book_list.tpl" teamId, _ := strconv.Atoi(c.Ctx.Input.Param(":id")) pageIndex, _ := c.GetInt("page", 0) if teamId <= 0 { c.JsonResult(5002, "团队标识不能为空") } team, err := models.NewTeam().First(teamId) if err == orm.ErrNoRows { c.ShowErrorPage(404, "团队不存在") } c.CheckErrorResult(500, err) c.Data["Model"] = team teams, totalCount, err := models.NewTeamRelationship().FindToPager(teamId, pageIndex, conf.PageSize) if err != nil && err != orm.ErrNoRows { c.ShowErrorPage(500, err.Error()) } if err == orm.ErrNoRows || len(teams) <= 0 { c.Data["Result"] = template.JS("[]") c.Data["PageHtml"] = "" return } if totalCount > 0 { pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() } else { c.Data["PageHtml"] = "" } b, err := json.Marshal(teams) if err != nil { beego.Error("编码 JSON 结果失败 ->", err) c.Data["Result"] = template.JS("[]") } else { c.Data["Result"] = template.JS(string(b)) } } //给团队增加项目. func (c *ManagerController) TeamBookAdd() { c.Prepare() teamId, _ := c.GetInt("teamId") bookId, _ := c.GetInt("bookId") if teamId <= 0 || bookId <= 0 { c.JsonResult(500, "参数错误") } teamRel := models.NewTeamRelationship() teamRel.BookId = bookId teamRel.TeamId = teamId err := teamRel.Save() if err != nil { c.JsonResult(5001, err.Error()) } else { teamRel.Include() c.JsonResult(0, "OK", teamRel) } } //搜索未参与的项目. func (c *ManagerController) TeamSearchBook() { c.Prepare() teamId, _ := c.GetInt("teamId") keyword := strings.TrimSpace(c.GetString("q")) if teamId <= 0 { c.JsonResult(500, "参数错误") } searchResult, err := models.NewTeamRelationship().FindNotJoinBookByName(teamId, keyword, 10) if err != nil { c.JsonResult(500, err.Error()) } c.JsonResult(0, "OK", searchResult) } //删除团队项目. func (c *ManagerController) TeamBookDelete() { c.Prepare() teamRelationshipId, _ := c.GetInt("teamRelId") if teamRelationshipId <= 0 { c.JsonResult(500, "参数错误") } err := models.NewTeamRelationship().Delete(teamRelationshipId) if err != nil { c.JsonResult(5001, "删除失败") } c.JsonResult(0, "OK") } //项目集列表. func (c *ManagerController) Itemsets() { c.Prepare() c.TplName = "manager/itemsets.tpl" pageIndex, _ := c.GetInt("page", 0) items, totalCount, err := models.NewItemsets().FindToPager(pageIndex, conf.PageSize) if err != nil && err != orm.ErrNoRows { c.ShowErrorPage(500, err.Error()) } if err == orm.ErrNoRows || len(items) <= 0 { c.Data["Lists"] = items c.Data["PageHtml"] = "" return } if totalCount > 0 { pager := pagination.NewPagination(c.Ctx.Request, totalCount, conf.PageSize, c.BaseUrl()) c.Data["PageHtml"] = pager.HtmlPages() } else { c.Data["PageHtml"] = "" } c.Data["Lists"] = items } //编辑或添加项目集. func (c *ManagerController) ItemsetsEdit() { c.Prepare() itemId, _ := c.GetInt("itemId") itemName := strings.TrimSpace(c.GetString("itemName")) itemKey := strings.TrimSpace(c.GetString("itemKey")) if itemName == "" || itemKey == "" { c.JsonResult(5001, "参数错误") } var item *models.Itemsets var err error if itemId > 0 { if item, err = models.NewItemsets().First(itemId); err != nil { if err == orm.ErrNoRows { c.JsonResult(5002, "项目集不存在") } else { c.JsonResult(5003, "查询项目集出错") } } } else { item = models.NewItemsets() } item.ItemKey = itemKey item.ItemName = itemName item.MemberId = c.Member.MemberId item.ModifyAt = c.Member.MemberId if err := item.Save(); err != nil { c.JsonResult(5004, err.Error()) } c.JsonResult(0, "OK") } //删除项目集. func (c *ManagerController) ItemsetsDelete() { c.Prepare() itemId, _ := c.GetInt("itemId") if err := models.NewItemsets().Delete(itemId); err != nil { c.JsonResult(5001, err.Error()) } c.JsonResult(0, "OK") }