main.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"time"
)

func main() {
	var args []string
	if len(os.Args) > 1 {
		args = os.Args[1:]
	} else {
		fmt.Println(`Usage of ./gitlab-migrate:
setup                  Initial setup
login                  Validate credentials
<pid> <destination>    Migrate <pid> to <destination>`)
		os.Exit(1)
	}

	switch args[0] {
	case "login":
		login()
		os.Exit(0)
	case "setup":
		setup()
		os.Exit(0)
	}

	if len(args) >= 3 {
		log.Println("Too many arguments")
	} else if len(args) == 2 {
		migrate(args[0], args[1])
		os.Exit(0)
	} else if len(args) == 1 {
		migrate(args[0], "")
	}
}

type scheduleResp struct {
	Message string `json:"message"`
}

const notFoundError = "404 Project Not Found"

// migrate performs the migration of the project from the source gitlab
// server to the destination server, for the given user or at the
// specified namespace.
func migrate(pid, dst string) {

	c := fetchCredentials()
	log.Println("Scheduling export")
	resp, err := scheduleExport(c.ExportURI, c.ExportToken, pid)
	if err != nil {
		log.Fatal(err)
	}

	bs, err := ioutil.ReadAll(resp.Body)

	var m scheduleResp
	json.Unmarshal(bs, &m)

	if m.Message == notFoundError {
		log.Println("Project with PID", pid, "not found.")
		os.Exit(1)
	}

	var r *statusResp
	var filename string
	// consider not looping forever
	for {
		// retrieve the export status
		r, _, err = exportStatus(c.ExportURI, c.ExportToken, pid)
		if err != nil {
			log.Fatal(err)
		}
		log.Println("Export status:", r.ExportStatus)
		if r.ExportStatus == "finished" {

			t := time.Now()
			filename = "./" + t.Format("01-02-2006") + "-" + r.Path + ".tar.gz"
			log.Println("Downloading", filename)

			// Download the project
			_, err := exportDownload(c.ExportURI, c.ExportToken, pid, filename)
			if err != nil {
				log.Fatal(err)
			}
			break
		}
		time.Sleep(10 * time.Second)
	}

	// if the dst is not empty, create the namespace
	// otherwise create the project in the token's user's projects.
	if dst != "" {
		// create the groups
		log.Println("Creating groups")
		gid := newGroup(c.ImportURI, c.ImportToken, dst)
		// import the project
		log.Println("Importing project")
		_ = importFile(c.ImportURI, c.ImportToken, gid, r.Path, filename)
	} else {
		log.Println("Importing project")
		_ = importFile(c.ImportURI, c.ImportToken, dst, r.Path, filename)
	}

	log.Println("Import complete")
}