main.go

package main

import (
	"fmt"
	"net/http"
	"os"
	"strings"

	"github.com/itchyny/github-migrator/github"
	"github.com/itchyny/github-migrator/migrator"
	"github.com/itchyny/github-migrator/repo"
)

const name = "github-migrator"

func main() {
	if err := run(os.Args[1:]); err != nil {
		fmt.Fprintf(os.Stderr, "%s: %s\n", name, err)
		os.Exit(1)
	}
}

func run(args []string) error {
	if len(args) != 2 {
		return fmt.Errorf("usage: %s <source> <target>", name)
	}
	mig, err := createMigrator(args[0], args[1])
	if err != nil {
		return err
	}
	return mig.Migrate()
}

func createGitHubClient(tokenEnv, endpointEnv, proxyEnv string) (github.Client, error) {
	token := os.Getenv(tokenEnv)
	if token == "" {
		return nil, fmt.Errorf("GitHub token not found (specify %s)", tokenEnv)
	}
	endpoint := os.Getenv(endpointEnv)
	if endpoint == "" {
		endpoint = "https://api.github.com"
	}
	proxy := os.Getenv(proxyEnv)
	cli := github.New(
		token, endpoint, proxy,
		github.ClientLogger(
			github.NewLogger(
				github.LoggerPreRequest(func(req *http.Request) {
					fmt.Printf("===> %s: %s\n", req.Method, req.URL)
				}),
				github.LoggerPostRequest(func(res *http.Response, err error) {
					if err != nil {
						var suffix string
						if res != nil {
							suffix = fmt.Sprintf(": %s: %s", res.Request.Method, res.Request.URL)
						}
						fmt.Printf("<=== %s%s\n", err, suffix)
						return
					}
					fmt.Printf("<=== %s: %s: %s\n", res.Status, res.Request.Method, res.Request.URL)
				}),
			),
		),
	)
	user, err := cli.GetLogin()
	if err != nil {
		return nil, fmt.Errorf("%s (or you may want to set %s)", err, endpointEnv)
	}
	fmt.Printf("[<>] login succeeded: %s\n", user.Login)
	return cli, nil
}

func createMigrator(sourcePath, targetPath string) (migrator.Migrator, error) {
	sourceCli, err := createGitHubClient(
		"GITHUB_MIGRATOR_SOURCE_API_TOKEN",
		"GITHUB_MIGRATOR_SOURCE_API_ENDPOINT",
		"GITHUB_MIGRATOR_SOURCE_PROXY_URL",
	)
	if err != nil {
		return nil, err
	}
	targetCli, err := createGitHubClient(
		"GITHUB_MIGRATOR_TARGET_API_TOKEN",
		"GITHUB_MIGRATOR_TARGET_API_ENDPOINT",
		"GITHUB_MIGRATOR_TARGET_PROXY_URL",
	)
	if err != nil {
		return nil, err
	}
	source := repo.New(sourceCli, sourcePath)
	target := repo.New(targetCli, targetPath)
	return migrator.New(source, target, createUserMapping()), nil
}

func createUserMapping() map[string]string {
	m := make(map[string]string)
	for _, src := range strings.Split(os.Getenv("GITHUB_MIGRATOR_USER_MAPPING"), ",") {
		xs := strings.Split(strings.TrimSpace(src), ":")
		if len(xs) == 2 && len(xs[0]) > 0 && len(xs[1]) > 0 {
			m[xs[0]] = xs[1]
		}
	}
	return m
}