/
branch_rule.go
104 lines (90 loc) · 3.31 KB
/
branch_rule.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
// Copyright (c) 2021 Terminus, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package bundle
import (
"strconv"
"github.com/erda-project/erda/apistructs"
"github.com/erda-project/erda/bundle/apierrors"
"github.com/erda-project/erda/modules/pkg/diceworkspace"
"github.com/erda-project/erda/pkg/http/httputil"
)
// GetProjectBranchRules 查询项目分支规则
func (b *Bundle) GetProjectBranchRules(projectId uint64) ([]*apistructs.BranchRule, error) {
return b.GetBranchRules(apistructs.ProjectScope, projectId)
}
// GetAppBranchRules 查询应用分支规则
func (b *Bundle) GetAppBranchRules(appId uint64) ([]*apistructs.BranchRule, error) {
return b.GetBranchRules(apistructs.AppScope, appId)
}
// GetBranchRules 查询分支规则
func (b *Bundle) GetBranchRules(scopeType apistructs.ScopeType, scopeID uint64) ([]*apistructs.BranchRule, error) {
host, err := b.urls.DOP()
if err != nil {
return nil, err
}
hc := b.hc
var fetchResp apistructs.QueryBranchRuleResponse
resp, err := hc.Get(host).Path("/api/branch-rules").
Param("scopeId", strconv.FormatUint(scopeID, 10)).
Param("scopeType", string(scopeType)).
Header(httputil.InternalHeader, "bundle").Do().JSON(&fetchResp)
if err != nil {
return nil, apierrors.ErrInvoke.InternalError(err)
}
if !resp.IsOK() || !fetchResp.Success {
return nil, toAPIError(resp.StatusCode(), fetchResp.Error)
}
return fetchResp.Data, nil
}
func (b *Bundle) GetAllValidBranchWorkspace(appId uint64, userID string) ([]apistructs.ValidBranch, error) {
host, err := b.urls.DOP()
if err != nil {
return nil, err
}
hc := b.hc
var fetchResp apistructs.PipelineAppAllValidBranchWorkspaceResponse
resp, err := hc.Get(host).Path("/api/branch-rules/actions/app-all-valid-branch-workspaces").
Param("appId", strconv.FormatUint(appId, 10)).
Header(httputil.UserHeader, userID).
Header(httputil.InternalHeader, "bundle").Do().JSON(&fetchResp)
if err != nil {
return nil, apierrors.ErrInvoke.InternalError(err)
}
if !resp.IsOK() || !fetchResp.Success {
return nil, toAPIError(resp.StatusCode(), fetchResp.Error)
}
return fetchResp.Data, nil
}
func (b *Bundle) GetBranchWorkspaceConfig(appId uint64, branch string) (*apistructs.ValidBranch, error) {
app, err := b.GetApp(appId)
if err != nil {
return nil, err
}
return b.GetBranchWorkspaceConfigByProject(app.ProjectID, branch)
}
func (b *Bundle) GetPermissionByGitReference(branch *apistructs.ValidBranch) string {
resource := "normalBranch"
if branch.IsProtect {
resource = "protectedBranch"
}
return resource
}
func (b *Bundle) GetBranchWorkspaceConfigByProject(projectID uint64, branch string) (*apistructs.ValidBranch, error) {
rules, err := b.GetProjectBranchRules(projectID)
if err != nil {
return nil, err
}
validBranch := diceworkspace.GetValidBranchByGitReference(branch, rules)
return validBranch, nil
}