fix create team, update team missing units

models/org_team.go

@@ -40,6 +40,14 @@ func (t *Team) getUnits(e Engine) (err error) {
  return err
 }
 
+// GetUnitNames returns the team units names
+func (t *Team) GetUnitNames() (res []string) {
+ for _, u := range t.Units {
+ res = append(res, Units[u.Type].NameKey)
+ }
+ return
+}
+
 // HasWriteAccess returns true if team has at least write level access mode.
 func (t *Team) HasWriteAccess() bool {
  return t.Authorize >= AccessModeWrite
@@ -367,6 +375,24 @@ func UpdateTeam(t *Team, authChanged bool) (err error) {
  return fmt.Errorf("update: %v", err)
  }
 
+ // update units for team
+ if len(t.Units) > 0 {
+ for _, unit := range t.Units {
+ unit.TeamID = t.ID
+ }
+ // Delete team-unit.
+ if _, err := sess.
+ Where("team_id=?", t.ID).
+ Delete(new(TeamUnit)); err != nil {
+ return err
+ }
+
+ if _, err = sess.Insert(&t.Units); err != nil {
+ sess.Rollback()
+ return err
+ }
+ }
+
  // Update access for team members if needed.
  if authChanged {
  if err = t.getRepositories(sess); err != nil {

models/unit.go

@@ -4,6 +4,10 @@
 
 package models
 
+import (
+ "strings"
+)
+
 // UnitType is Unit's Type
 type UnitType int
 
@@ -137,3 +141,13 @@ var (
  UnitTypeExternalWiki: UnitExternalWiki,
  }
 )
+
+// FindUnitTypes give the unit key name and return unit
+func FindUnitTypes(nameKeys ...string) (res []UnitType) {
+ for t, u := range Units {
+ if strings.EqualFold(u.NameKey, u.NameKey) {
+ res = append(res, t)
+ }
+ }
+ return
+}

routers/api/v1/convert/convert.go

@@ -198,5 +198,6 @@ func ToTeam(team *models.Team) *api.Team {
  Name: team.Name,
  Description: team.Description,
  Permission: team.Authorize.String(),
+ Units: team.GetUnitNames(),
  }
 }

routers/api/v1/org/team.go

@@ -90,6 +90,20 @@ func CreateTeam(ctx *context.APIContext, form api.CreateTeamOption) {
  Description: form.Description,
  Authorize: models.ParseAccessMode(form.Permission),
  }
+
+ unitTypes := models.FindUnitTypes(form.Units...)
+
+ if team.Authorize < models.AccessModeOwner {
+ var units = make([]*models.TeamUnit, 0, len(form.Units))
+ for _, tp := range unitTypes {
+ units = append(units, &models.TeamUnit{
+ OrgID: ctx.Org.Organization.ID,
+ Type: tp,
+ })
+ }
+ team.Units = units
+ }
+
  if err := models.NewTeam(team); err != nil {
  if models.IsErrTeamAlreadyExist(err) {
  ctx.Error(422, "", err)
@@ -128,6 +142,19 @@ func EditTeam(ctx *context.APIContext, form api.EditTeamOption) {
  team.Name = form.Name
  team.Description = form.Description
  team.Authorize = models.ParseAccessMode(form.Permission)
+ unitTypes := models.FindUnitTypes(form.Units...)
+
+ if team.Authorize < models.AccessModeOwner {
+ var units = make([]*models.TeamUnit, 0, len(form.Units))
+ for _, tp := range unitTypes {
+ units = append(units, &models.TeamUnit{
+ OrgID: ctx.Org.Organization.ID,
+ Type: tp,
+ })
+ }
+ team.Units = units
+ }
+
  if err := models.UpdateTeam(team, true); err != nil {
  ctx.Error(500, "EditTeam", err)
  return