gickup/gitea/gitea.go
Andreas Wachter fecb4040e6
added sublogger to gitea (#178)
* added sublogger to gitea

* implemented the rest of the subloggers
2023-10-04 10:55:24 +02:00

525 lines
12 KiB
Go

package gitea
import (
"strings"
"time"
"code.gitea.io/sdk/gitea"
"github.com/cooperspencer/gickup/logger"
"github.com/cooperspencer/gickup/types"
"github.com/rs/zerolog"
)
var (
sub zerolog.Logger
)
func getOrgVisibility(visibility string) gitea.VisibleType {
switch visibility {
case "public":
return gitea.VisibleTypePublic
case "private":
return gitea.VisibleTypePrivate
case "limited":
return gitea.VisibleTypeLimited
default:
return gitea.VisibleTypePrivate
}
}
func getRepoVisibility(visibility string, private bool) bool {
switch visibility {
case "public":
return false
case "private":
return true
default:
return private
}
}
// Backup TODO.
func Backup(r types.Repo, d types.GenRepo, dry bool) bool {
orgvisibilty := getOrgVisibility(d.Visibility.Organizations)
repovisibility := getRepoVisibility(d.Visibility.Repositories, r.Private)
if d.URL == "" {
d.URL = "https://gitea.com/"
}
sub = logger.CreateSubLogger("stage", "gitea", "url", d.URL)
sub.Info().
Msgf("mirroring %s to %s", types.Blue(r.Name), d.URL)
giteaclient, err := gitea.NewClient(d.URL, gitea.SetToken(d.GetToken()))
if err != nil {
sub.Error().Msg(err.Error())
return false
}
user, _, err := giteaclient.GetMyUserInfo()
if err != nil {
sub.Error().
Msg(err.Error())
return false
}
if d.User == "" && d.CreateOrg {
d.User = r.Owner
}
if d.User != "" {
user, _, err = giteaclient.GetUserInfo(d.User)
if err != nil {
if d.CreateOrg {
org, _, err := giteaclient.CreateOrg(gitea.CreateOrgOption{
Name: d.User,
Visibility: orgvisibilty,
})
if err != nil {
sub.Error().
Msg(err.Error())
return false
}
user.ID = org.ID
user.UserName = org.UserName
} else {
sub.Error().
Msg(err.Error())
return false
}
}
}
if dry {
return true
}
repo, _, err := giteaclient.GetRepo(user.UserName, r.Name)
if err != nil {
opts := gitea.MigrateRepoOption{
RepoName: r.Name,
RepoOwner: user.UserName,
Mirror: true,
CloneAddr: r.URL,
AuthToken: r.Token,
Wiki: r.Origin.Wiki,
Private: repovisibility,
Description: r.Description,
MirrorInterval: d.MirrorInterval,
LFS: d.LFS,
}
if r.Token == "" {
opts = gitea.MigrateRepoOption{
RepoName: r.Name,
RepoOwner: user.UserName,
Mirror: true,
CloneAddr: r.URL,
AuthUsername: r.Origin.User,
AuthPassword: r.Origin.Password,
Wiki: r.Origin.Wiki,
Private: repovisibility,
Description: r.Description,
MirrorInterval: d.MirrorInterval,
LFS: d.LFS,
}
}
_, _, err := giteaclient.MigrateRepo(opts)
if err != nil {
sub.Error().
Msg(err.Error())
sub.Info().
Msgf("deleting %s again", types.Blue(r.Name))
_, err = giteaclient.DeleteRepo(user.UserName, r.Name)
if err != nil {
sub.Error().
Str("stage", "gitea").
Str("url", d.URL).
Msgf("couldn't delete %s!", types.Red(r.Name))
}
return false
}
sub.Info().
Msgf("mirrored %s to %s", types.Blue(r.Name), d.URL)
return true
}
if d.MirrorInterval != repo.MirrorInterval {
_, _, err := giteaclient.EditRepo(user.UserName, r.Name, gitea.EditRepoOption{MirrorInterval: &d.MirrorInterval})
if err != nil {
sub.Error().
Msgf("Couldn't update %s", types.Red(r.Name))
}
return false
}
if repo.Mirror {
sub.Info().
Msgf("mirror of %s already exists, syncing instead", types.Blue(r.Name))
_, err := giteaclient.MirrorSync(user.UserName, repo.Name)
if err != nil {
sub.Error().
Str("stage", "gitea").
Str("url", d.URL).
Msg(err.Error())
return false
}
sub.Info().
Str("stage", "gitea").
Str("url", d.URL).
Msgf("successfully synced %s.", types.Blue(r.Name))
}
return true
}
// Get TODO.
func Get(conf *types.Conf) ([]types.Repo, bool) {
ran := false
repos := []types.Repo{}
for _, repo := range conf.Source.Gitea {
if repo.URL == "" {
repo.URL = "https://gitea.com"
}
sub = logger.CreateSubLogger("stage", "gitea", "url", repo.URL)
err := repo.Filter.ParseDuration()
if err != nil {
sub.Error().
Msg(err.Error())
}
ran = true
if repo.User == "" {
sub.Info().
Msg("grabbing my repositories")
} else {
sub.Info().
Msgf("grabbing repositories from %s", repo.User)
}
opt := gitea.ListReposOptions{}
opt.PageSize = 50
opt.Page = 1
gitearepos := []*gitea.Repository{}
var client *gitea.Client
token := repo.GetToken()
if token != "" {
client, err = gitea.NewClient(repo.URL, gitea.SetToken(token))
} else {
client, err = gitea.NewClient(repo.URL)
}
if token != "" && repo.User == "" {
user, _, err := client.GetMyUserInfo()
if err != nil {
sub.Error().
Msg(err.Error())
continue
}
repo.User = user.UserName
}
if err != nil {
sub.Error().
Msg(err.Error())
continue
}
for {
repos, _, err := client.ListUserRepos(repo.User, opt)
if err != nil {
sub.Error().
Msg(err.Error())
continue
}
if len(repos) == 0 {
break
}
gitearepos = append(gitearepos, repos...)
opt.Page++
}
if repo.Starred {
starredrepos, _, err := client.GetStarredRepos(repo.User)
if err != nil {
sub.Error().
Msg(err.Error())
} else {
gitearepos = append(gitearepos, starredrepos...)
}
}
include := types.GetMap(repo.Include)
exclude := types.GetMap(repo.Exclude)
includeorgs := types.GetMap(repo.IncludeOrgs)
excludeorgs := types.GetMap(repo.ExcludeOrgs)
for i := range repo.Filter.Languages {
repo.Filter.Languages[i] = strings.ToLower(repo.Filter.Languages[i])
}
languages := types.GetMap(repo.Filter.Languages)
for _, r := range gitearepos {
if repo.Filter.ExcludeForks {
if r.Fork {
continue
}
}
if repo.Filter.ExcludeArchived {
if r.Archived {
continue
}
}
if len(repo.Filter.Languages) > 0 {
langs, _, err := client.GetRepoLanguages(r.Owner.UserName, r.Name)
if err != nil {
sub.Error().
Msg(err.Error())
continue
} else {
language := ""
percentage := int64(0)
for lang, percent := range langs {
if percent > percentage {
language = lang
}
}
if !languages[strings.ToLower(language)] {
continue
}
}
}
if r.Stars < repo.Filter.Stars {
continue
}
if time.Since(r.Updated) > repo.Filter.LastActivityDuration && repo.Filter.LastActivityDuration != 0 {
continue
}
if include[r.Name] {
repos = append(repos, types.Repo{
Name: r.Name,
URL: r.CloneURL,
SSHURL: r.SSHURL,
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
if r.HasWiki && repo.Wiki && types.StatRemote(r.CloneURL, r.SSHURL, repo) {
repos = append(repos, types.Repo{
Name: r.Name + ".wiki",
URL: types.DotGitRx.ReplaceAllString(r.CloneURL, ".wiki.git"),
SSHURL: types.DotGitRx.ReplaceAllString(r.SSHURL, ".wiki.git"),
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
}
continue
}
if exclude[r.Name] {
continue
}
if len(repo.Include) == 0 {
repos = append(repos, types.Repo{
Name: r.Name,
URL: r.CloneURL,
SSHURL: r.SSHURL,
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
if r.HasWiki && repo.Wiki && types.StatRemote(r.CloneURL, r.SSHURL, repo) {
repos = append(repos, types.Repo{
Name: r.Name + ".wiki",
URL: types.DotGitRx.ReplaceAllString(r.CloneURL, ".wiki.git"),
SSHURL: types.DotGitRx.ReplaceAllString(r.SSHURL, ".wiki.git"),
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
}
}
}
orgopt := gitea.ListOptions{Page: 1, PageSize: 50}
orgs := []*gitea.Organization{}
if token != "" {
for {
o, _, err := client.ListUserOrgs(repo.User, gitea.ListOrgsOptions{ListOptions: orgopt})
if err != nil {
sub.Error().
Msg(err.Error())
}
if len(o) == 0 {
break
}
orgs = append(orgs, o...)
orgopt.Page++
}
}
orgrepos := []*gitea.Repository{}
for _, org := range orgs {
orgopt.Page = 1
if excludeorgs[org.UserName] {
continue
}
for {
if len(includeorgs) > 0 {
if includeorgs[org.UserName] {
o := getOrgRepos(client, org, orgopt, repo)
if len(o) == 0 {
break
}
orgrepos = append(orgrepos, o...)
}
} else {
o := getOrgRepos(client, org, orgopt, repo)
if len(o) == 0 {
break
}
orgrepos = append(orgrepos, o...)
}
orgopt.Page++
}
}
for _, r := range orgrepos {
if repo.Filter.ExcludeForks {
if r.Fork {
continue
}
}
if repo.Filter.ExcludeArchived {
if r.Archived {
continue
}
}
if len(repo.Filter.Languages) > 0 {
langs, _, err := client.GetRepoLanguages(r.Owner.UserName, r.Name)
if err != nil {
sub.Error().
Msg(err.Error())
continue
} else {
language := ""
percentage := int64(0)
for lang, percent := range langs {
if percent > percentage {
language = lang
}
}
if !languages[strings.ToLower(language)] {
continue
}
}
}
if r.Stars < repo.Filter.Stars {
continue
}
if time.Since(r.Updated) > repo.Filter.LastActivityDuration && repo.Filter.LastActivityDuration != 0 {
continue
}
if include[r.Name] {
repos = append(repos, types.Repo{
Name: r.Name,
URL: r.CloneURL,
SSHURL: r.SSHURL,
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
if r.HasWiki && repo.Wiki && types.StatRemote(r.CloneURL, r.SSHURL, repo) {
repos = append(repos, types.Repo{
Name: r.Name + ".wiki",
URL: types.DotGitRx.ReplaceAllString(r.CloneURL, ".wiki.git"),
SSHURL: types.DotGitRx.ReplaceAllString(r.SSHURL, ".wiki.git"),
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
}
continue
}
if exclude[r.Name] {
continue
}
if len(repo.Include) == 0 {
repos = append(repos, types.Repo{
Name: r.Name,
URL: r.CloneURL,
SSHURL: r.SSHURL,
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
if r.HasWiki && repo.Wiki && types.StatRemote(r.CloneURL, r.SSHURL, repo) {
repos = append(repos, types.Repo{
Name: r.Name + ".wiki",
URL: types.DotGitRx.ReplaceAllString(r.CloneURL, ".wiki.git"),
SSHURL: types.DotGitRx.ReplaceAllString(r.SSHURL, ".wiki.git"),
Token: token,
Defaultbranch: r.DefaultBranch,
Origin: repo,
Owner: r.Owner.UserName,
Hoster: types.GetHost(repo.URL),
Description: r.Description,
Private: r.Private,
})
}
}
}
}
return repos, ran
}
func getOrgRepos(client *gitea.Client, org *gitea.Organization,
orgopt gitea.ListOptions, repo types.GenRepo,
) []*gitea.Repository {
o, _, err := client.ListOrgRepos(org.UserName,
gitea.ListOrgReposOptions{orgopt})
if err != nil {
sub.Error().Str("stage", "gitea").Str("url", repo.URL).Msg(err.Error())
}
return o
}