Files
phs_v1.0.1.0/developtools/integration_verification/tools/fotff/vcs/gitee/compare.go
2024-09-27 19:16:49 +08:00

140 lines
3.4 KiB
Go

/*
* Copyright (c) 2022 Huawei Device Co., Ltd.
* 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 gitee
import (
"encoding/json"
"fmt"
"fotff/utils"
"net/http"
"time"
)
type CompareParam struct {
Head string
Base string
Repo string
Owner string
}
type CompareResp struct {
Commits []*Commit `json:"commits"`
}
type Commit struct {
CommitExtend `json:"-"`
URL string `json:"url"`
SHA string `json:"sha"`
Commit struct {
Committer struct {
Date string `json:"date"`
} `json:"committer"`
Message string `json:"message"`
} `json:"commit"`
Parents []struct {
SHA string `json:"sha"`
URL string `json:"url"`
} `json:"parents"`
Files []struct {
Filename string `json:"filename"`
Status string `json:"status"`
Patch string `json:"patch,omitempty"`
} `json:"files,omitempty"`
}
type CommitExtend struct {
Owner string
Repo string
}
func GetLatestMRBefore(owner, repo, branch string, before string) (ret *Commit, err error) {
branchResp, err := GetBranch(owner, repo, branch)
if err != nil {
return nil, err
}
head := branchResp.Commit
head.Owner = owner
head.Repo = repo
for head.Commit.Committer.Date > before {
if head, err = GetCommit(owner, repo, head.Parents[0].SHA); err != nil {
return nil, err
}
}
return head, nil
}
func GetBetweenTimeMRs(owner, repo, branch string, from, to time.Time) (ret []*Commit, err error) {
branchResp, err := GetBranch(owner, repo, branch)
if err != nil {
return nil, err
}
fromStr := from.UTC().Format(time.RFC3339)
toStr := to.UTC().Format(time.RFC3339)
head := branchResp.Commit
head.Owner = owner
head.Repo = repo
for head.Commit.Committer.Date > fromStr {
if head.Commit.Committer.Date < toStr {
ret = append(ret, head)
}
if head, err = GetCommit(owner, repo, head.Parents[0].SHA); err != nil {
return nil, err
}
}
return ret, nil
}
func GetBetweenMRs(param CompareParam) ([]*Commit, error) {
commits, err := GetBetweenCommits(param)
if err != nil {
return nil, err
}
var ret []*Commit
head := param.Head
for head != param.Base {
for _, commit := range commits {
if commit.SHA != head {
continue
}
commit.Owner = param.Owner
commit.Repo = param.Repo
ret = append(ret, commit)
head = commit.Parents[0].SHA
}
}
return ret, nil
}
func GetBetweenCommits(param CompareParam) ([]*Commit, error) {
url := fmt.Sprintf("https://gitee.com/api/v5/repos/%s/%s/compare/%s...%s", param.Owner, param.Repo, param.Base, param.Head)
var resp []byte
if c, found := utils.CacheGet("gitee", url); found {
resp = c.([]byte)
} else {
var err error
resp, err = utils.DoSimpleHttpReq(http.MethodGet, url, nil, nil)
if err != nil {
return nil, err
}
utils.CacheSet("gitee", url, resp)
}
var compareResp CompareResp
if err := json.Unmarshal(resp, &compareResp); err != nil {
return nil, err
}
return compareResp.Commits, nil
}