99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
|
package gitlabremote
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/pterm/pterm"
|
||
|
|
"github.com/xanzy/go-gitlab"
|
||
|
|
"gitlab.sweetwater.com/it/devops/tools/gitlab-project-manager/internal/remotes/projects"
|
||
|
|
"gitlab.sweetwater.com/it/devops/tools/gitlab-project-manager/internal/remotes/remote"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewGitlabApi(info *remote.RemoteInfo) (*gitlab.Client, error) {
|
||
|
|
client, err := gitlab.NewClient(info.Token, gitlab.WithBaseURL(info.Host))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return client, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *GitlabRemote) GetNumProjects(opts *remote.RemoteQueryOpts) int {
|
||
|
|
listOpts := *DefaultListOpts
|
||
|
|
listOpts.PerPage = 1
|
||
|
|
listOpts.Simple = gitlab.Ptr[bool](true)
|
||
|
|
_, resp, err := r.api.Projects.ListProjects(&listOpts)
|
||
|
|
if err != nil {
|
||
|
|
pterm.Error.Printfln("Failed getting number of GitLab projects: %s", err)
|
||
|
|
return -1
|
||
|
|
}
|
||
|
|
return resp.TotalItems
|
||
|
|
}
|
||
|
|
|
||
|
|
// Returns a list of projects along with the next page and an error
|
||
|
|
// if there was an error
|
||
|
|
func (r *GitlabRemote) ListProjects(opts *gitlab.ListProjectsOptions) (
|
||
|
|
[]*projects.Project, *gitlab.Response, error) {
|
||
|
|
pList := make([]*projects.Project, 0)
|
||
|
|
projects, resp, err := r.api.Projects.ListProjects(
|
||
|
|
opts,
|
||
|
|
gitlab.WithContext(r.info.Ctx),
|
||
|
|
)
|
||
|
|
if err == nil {
|
||
|
|
pList = append(pList, r.handleProjects(projects)...)
|
||
|
|
}
|
||
|
|
return pList, resp, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *GitlabRemote) handleProjects(gitProjects []*gitlab.Project) []*projects.Project {
|
||
|
|
// Opportunity to perform any filtering or additional lookups
|
||
|
|
// on a per-project basis
|
||
|
|
pList := make([]*projects.Project, 0, len(gitProjects))
|
||
|
|
for _, project := range gitProjects {
|
||
|
|
var owner string
|
||
|
|
if project.Owner != nil {
|
||
|
|
owner = project.Owner.Email
|
||
|
|
}
|
||
|
|
p := &projects.Project{
|
||
|
|
ID: project.ID,
|
||
|
|
Description: project.Description,
|
||
|
|
SSHURLToRepo: project.SSHURLToRepo,
|
||
|
|
HTTPURLToRepo: project.HTTPURLToRepo,
|
||
|
|
WebURL: project.WebURL,
|
||
|
|
Name: project.Name,
|
||
|
|
NameWithNamespace: project.NameWithNamespace,
|
||
|
|
Path: project.Path,
|
||
|
|
PathWithNamespace: project.PathWithNamespace,
|
||
|
|
Remote: r.info.Host,
|
||
|
|
Owner: owner,
|
||
|
|
AvatarURL: project.AvatarURL,
|
||
|
|
LastActivityAt: *project.LastActivityAt,
|
||
|
|
Readme: project.ReadmeURL,
|
||
|
|
Languages: r.GetProjectLanguages(project),
|
||
|
|
}
|
||
|
|
pList = append(pList, p)
|
||
|
|
}
|
||
|
|
return pList
|
||
|
|
}
|
||
|
|
|
||
|
|
// A nil return indicates an API error or GitLab doesn't know what
|
||
|
|
// language the project uses.
|
||
|
|
func (r *GitlabRemote) GetProjectLanguages(project *gitlab.Project) *projects.ProjectLanguages {
|
||
|
|
l, _, e := r.api.Projects.GetProjectLanguages(project.ID, gitlab.WithContext(r.info.Ctx))
|
||
|
|
if e != nil {
|
||
|
|
pterm.Error.Printfln("Failed requesting project languages: %s", e.Error())
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var pLangs projects.ProjectLanguages
|
||
|
|
pLangs = make([]*projects.ProjectLanguage, len(*l))
|
||
|
|
|
||
|
|
var i int
|
||
|
|
for name, pcnt := range *l {
|
||
|
|
pLangs[i] = &projects.ProjectLanguage{
|
||
|
|
Name: name,
|
||
|
|
Percentage: pcnt,
|
||
|
|
}
|
||
|
|
i++
|
||
|
|
}
|
||
|
|
|
||
|
|
return &pLangs
|
||
|
|
}
|