package gogs

import (
	"gickup/types"

	"github.com/gogs/go-gogs-client"
	"github.com/rs/zerolog/log"
)

func Backup(r types.Repo, d types.GenRepo, dry bool) {
	log.Info().Str("stage", "gogs").Str("url", d.Url).Msgf("mirroring %s to %s", types.Blue(r.Name), d.Url)
	gogsclient := gogs.NewClient(d.Url, d.Token)

	user, err := gogsclient.GetSelfInfo()
	if err != nil {
		log.Fatal().Str("stage", "gogs").Str("url", d.Url).Msg(err.Error())
	}
	if !dry {
		repo, err := gogsclient.GetRepo(user.UserName, r.Name)
		if err != nil {
			opts := gogs.MigrateRepoOption{RepoName: r.Name, UID: int(user.ID), Mirror: true, CloneAddr: r.Url, AuthUsername: r.Token}
			if r.Token == "" {
				opts = gogs.MigrateRepoOption{RepoName: r.Name, UID: int(user.ID), Mirror: true, CloneAddr: r.Url, AuthUsername: r.Origin.User, AuthPassword: r.Origin.Password}
			}
			_, err := gogsclient.MigrateRepo(opts)
			if err != nil {
				log.Fatal().Str("stage", "gogs").Str("url", d.Url).Msg(err.Error())
			}
		} else {
			if repo.Mirror {
				log.Info().Str("stage", "gogs").Str("url", d.Url).Msgf("mirror of %s already exists, syncing instead", types.Blue(r.Name))
				err := gogsclient.MirrorSync(user.UserName, repo.Name)
				if err != nil {
					log.Fatal().Str("stage", "gogs").Str("url", d.Url).Msg(err.Error())
				}
				log.Info().Str("stage", "gogs").Str("url", d.Url).Msgf("successfully synced %s.", types.Blue(r.Name))
			}
		}
	}
}

func Get(conf *types.Conf) []types.Repo {
	repos := []types.Repo{}
	for _, repo := range conf.Source.Gogs {
		log.Info().Str("stage", "gogs").Str("url", repo.Url).Msgf("grabbing repositories from %s", repo.User)
		client := gogs.NewClient(repo.Url, repo.Token)
		gogsrepos, err := client.ListUserRepos(repo.User)
		if err != nil {
			log.Fatal().Str("stage", "gogs").Str("url", repo.Url).Msg(err.Error())
		}

		include := types.GetMap(repo.Include)
		exclude := types.GetMap(repo.Exclude)
		excludeorgs := types.GetMap(repo.ExcludeOrgs)

		for _, r := range gogsrepos {
			if include[r.Name] {
				repos = append(repos, types.Repo{Name: r.Name, Url: r.CloneURL, SshUrl: r.SSHURL, Token: repo.Token, Defaultbranch: r.DefaultBranch, Origin: repo})
				continue
			}
			if exclude[r.Name] {
				continue
			}
			if len(include) == 0 {
				repos = append(repos, types.Repo{Name: r.Name, Url: r.CloneURL, SshUrl: r.SSHURL, Token: repo.Token, Defaultbranch: r.DefaultBranch, Origin: repo})
			}
		}
		orgs, err := client.ListUserOrgs(repo.User)
		if err != nil {
			log.Fatal().Str("stage", "gogs").Str("url", repo.Url).Msg(err.Error())
		}

		orgrepos := []*gogs.Repository{}
		for _, org := range orgs {
			if excludeorgs[org.UserName] {
				continue
			}
			o, err := client.ListOrgRepos(org.UserName)
			if err != nil {
				log.Fatal().Str("stage", "gogs").Str("url", repo.Url).Msg(err.Error())
			}
			if len(o) == 0 {
				break
			}
			orgrepos = append(orgrepos, o...)
		}
		for _, r := range orgrepos {
			if include[r.Name] {
				repos = append(repos, types.Repo{Name: r.Name, Url: r.CloneURL, SshUrl: r.SSHURL, Token: repo.Token, Defaultbranch: r.DefaultBranch, Origin: repo})
				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: repo.Token, Defaultbranch: r.DefaultBranch, Origin: repo})
			}
		}
	}
	return repos
}