Modernize the logging and flag parsing a bit
This commit is contained in:
147
main.go
147
main.go
@@ -3,9 +3,6 @@ package main
|
||||
import (
|
||||
_ "embed"
|
||||
"flag"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"path"
|
||||
|
||||
@@ -13,149 +10,171 @@ import (
|
||||
"github.com/go-git/go-git/v5"
|
||||
"github.com/go-git/go-git/v5/config"
|
||||
"golang.design/x/clipboard"
|
||||
|
||||
logger "git.site.quack-lab.dev/dave/cylogger"
|
||||
)
|
||||
|
||||
var Error *log.Logger
|
||||
var Warning *log.Logger
|
||||
|
||||
func init() {
|
||||
log.SetFlags(log.Lmicroseconds | log.Lshortfile)
|
||||
// logFile, err := os.Create("main.log")
|
||||
// if err != nil {
|
||||
// log.Printf("Error creating log file: %v", err)
|
||||
// os.Exit(1)
|
||||
// }
|
||||
logger := io.MultiWriter(os.Stdout)
|
||||
log.SetOutput(logger)
|
||||
|
||||
Error = log.New(io.MultiWriter(os.Stderr, os.Stdout),
|
||||
fmt.Sprintf("%sERROR:%s ", "\033[0;101m", "\033[0m"),
|
||||
log.Lmicroseconds|log.Lshortfile)
|
||||
Warning = log.New(io.MultiWriter(os.Stdout),
|
||||
fmt.Sprintf("%sWarning:%s ", "\033[0;93m", "\033[0m"),
|
||||
log.Lmicroseconds|log.Lshortfile)
|
||||
}
|
||||
var Version = "1.0.0"
|
||||
|
||||
func main() {
|
||||
name := flag.String("name", "", "Name of the repository")
|
||||
flag.String("n", "", "Name of the repository")
|
||||
|
||||
private := flag.Bool("private", false, "Make the repository private")
|
||||
flag.Bool("p", false, "Make the repository private (shorthand)")
|
||||
|
||||
noinit := flag.Bool("noinit", false, "Do not add remote to the new repo")
|
||||
flag.Bool("ni", false, "Do not add remote to the new repo (shorthand)")
|
||||
|
||||
forceInit := flag.Bool("f", false, "Force assignment of the remote (deletes existing origin)")
|
||||
|
||||
remote := flag.String("remote", "origin", "Name of the remote to create for the new repository")
|
||||
flag.String("r", "origin", "Name of the remote to create for the new repository (shorthand)")
|
||||
|
||||
version := flag.Bool("v", false, "Show version")
|
||||
flag.Bool("version", false, "Show version")
|
||||
|
||||
help := flag.Bool("h", false, "Show help")
|
||||
flag.Bool("help", false, "Show help")
|
||||
|
||||
flag.Parse()
|
||||
logger.InitFlag()
|
||||
|
||||
logger.Info("Starting repo %s", Version)
|
||||
|
||||
if *version {
|
||||
logger.Info("Version: %s", Version)
|
||||
return
|
||||
}
|
||||
if *help {
|
||||
flag.PrintDefaults()
|
||||
return
|
||||
}
|
||||
|
||||
err := clipboard.Init()
|
||||
if err != nil {
|
||||
Error.Fatalf("Error initializing clipboard: %v", err)
|
||||
logger.Error("Error initializing clipboard: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
username, ok := os.LookupEnv("GITEA_USER")
|
||||
if !ok {
|
||||
Error.Fatalf("GITEA_USER environment variable is required")
|
||||
logger.Error("GITEA_USER environment variable is required")
|
||||
return
|
||||
}
|
||||
password, ok := os.LookupEnv("GITEA_PASSWORD")
|
||||
if !ok {
|
||||
Error.Fatalf("GITEA_PASSWORD environment variable is required")
|
||||
logger.Error("GITEA_PASSWORD environment variable is required")
|
||||
return
|
||||
}
|
||||
url, ok := os.LookupEnv("GITEA_URL")
|
||||
if !ok {
|
||||
Error.Fatalf("GITEA_URL environment variable is required")
|
||||
logger.Error("GITEA_URL environment variable is required")
|
||||
return
|
||||
}
|
||||
|
||||
if username == "" || password == "" {
|
||||
Error.Fatalf("GITEA_USER and GITEA_PASSWORD environment variables are required")
|
||||
logger.Error("GITEA_USER and GITEA_PASSWORD environment variables are required")
|
||||
return
|
||||
}
|
||||
|
||||
var name, remote string
|
||||
var private, noinit, forceInit bool
|
||||
flag.StringVar(&name, "name", "", "Name of the repository")
|
||||
flag.StringVar(&name, "n", "", "Name of the repository (shorthand)")
|
||||
flag.BoolVar(&private, "private", false, "Make the repository private")
|
||||
flag.BoolVar(&private, "p", false, "Make the repository private (shorthand)")
|
||||
flag.BoolVar(&noinit, "noinit", false, "Do not add remote to the new repo")
|
||||
flag.BoolVar(&noinit, "ni", false, "Do not add remote to the new repo (shorthand)")
|
||||
flag.BoolVar(&forceInit, "f", false, "Force assignment of the remote (deletes existing origin)")
|
||||
flag.StringVar(&remote, "remote", "origin", "Name of the remote to create for the new repository")
|
||||
flag.StringVar(&remote, "r", "origin", "Name of the remote to create for the new repository (shorthand)")
|
||||
flag.Parse()
|
||||
|
||||
if len(flag.Args()) > 0 {
|
||||
name = flag.Args()[0]
|
||||
if flag.NArg() > 0 {
|
||||
*name = flag.Arg(0)
|
||||
}
|
||||
|
||||
if name == "" {
|
||||
Error.Fatalf("Repository name is required")
|
||||
if *name == "" {
|
||||
logger.Error("Repository name is required")
|
||||
return
|
||||
}
|
||||
|
||||
logger.Info("Username: %s", username)
|
||||
logger.Info("URL: %s", url)
|
||||
logger.Info("Name: %s", *name)
|
||||
logger.Info("Private: %t", *private)
|
||||
logger.Info("Noinit: %t", *noinit)
|
||||
logger.Info("ForceInit: %t", *forceInit)
|
||||
logger.Info("Remote: %s", *remote)
|
||||
|
||||
client, err := gitea.NewClient(url)
|
||||
if err != nil {
|
||||
Error.Fatalf("Error creating client: %v", err)
|
||||
logger.Error("Error creating client: %v", err)
|
||||
return
|
||||
}
|
||||
client.SetBasicAuth(username, password)
|
||||
|
||||
repo, _, err := client.GetRepo(username, name)
|
||||
logger.Info("Getting repository...")
|
||||
repo, _, err := client.GetRepo(username, *name)
|
||||
if err == nil {
|
||||
log.Printf("Repository already exists at:\n%s", repo.CloneURL)
|
||||
logger.Info("Repository already exists at:\n%s", repo.CloneURL)
|
||||
clipboard.Write(clipboard.FmtText, []byte(repo.CloneURL))
|
||||
} else {
|
||||
logger.Info("Repository does not exist, creating...")
|
||||
repo, _, err = client.CreateRepo(gitea.CreateRepoOption{
|
||||
Name: name,
|
||||
Private: private,
|
||||
Name: *name,
|
||||
Private: *private,
|
||||
DefaultBranch: "master",
|
||||
})
|
||||
if err != nil {
|
||||
Error.Fatalf("Error creating repository: %v", err)
|
||||
logger.Error("Error creating repository: %v", err)
|
||||
return
|
||||
}
|
||||
log.Printf("Repository created at:\n%s", repo.CloneURL)
|
||||
logger.Info("Repository created at: %s", repo.CloneURL)
|
||||
clipboard.Write(clipboard.FmtText, []byte(repo.CloneURL))
|
||||
}
|
||||
|
||||
if !noinit {
|
||||
if !*noinit {
|
||||
logger.Info("Adding remote to local repository...")
|
||||
cwd, err := os.Getwd()
|
||||
if err != nil {
|
||||
Error.Fatalf("Error getting current working directory: %v", err)
|
||||
logger.Error("Error getting current working directory: %v", err)
|
||||
return
|
||||
}
|
||||
cwd = path.Clean(cwd)
|
||||
|
||||
var localRepo *git.Repository
|
||||
logger.Info("Checking if local repository exists at %s/.git...", cwd)
|
||||
_, err = os.Stat(path.Join(cwd, ".git"))
|
||||
if err == nil {
|
||||
localRepo, err = git.PlainOpen(cwd)
|
||||
if err != nil {
|
||||
Error.Fatalf("Error opening git repository: %v", err)
|
||||
logger.Error("Error opening git repository: %v", err)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
localRepo, err = git.PlainInit(cwd, false)
|
||||
if err != nil {
|
||||
Error.Fatalf("Error initializing git repository: %v", err)
|
||||
logger.Error("Error initializing git repository: %v", err)
|
||||
return
|
||||
}
|
||||
}
|
||||
_, err = localRepo.Remote(remote)
|
||||
logger.Info("Checking if remote %s exists in local repository...", *remote)
|
||||
_, err = localRepo.Remote(*remote)
|
||||
if err == nil {
|
||||
if forceInit {
|
||||
err = localRepo.DeleteRemote(remote)
|
||||
logger.Info("Remote %s exists in local repository", *remote)
|
||||
if *forceInit {
|
||||
logger.Info("Deleting remote %s from local repository...", *remote)
|
||||
err = localRepo.DeleteRemote(*remote)
|
||||
if err != nil {
|
||||
Error.Fatalf("Error deleting remote %s from local repository: %v", remote, err)
|
||||
logger.Error("Error deleting remote %s from local repository: %v", *remote, err)
|
||||
return
|
||||
}
|
||||
logger.Info("Remote %s deleted from local repository", *remote)
|
||||
} else {
|
||||
Error.Fatalf("Remote %s already exists in local repository", remote)
|
||||
logger.Error("Remote %s already exists in local repository", *remote)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
logger.Info("Adding remote %s to local repository...", *remote)
|
||||
_, err = localRepo.CreateRemote(&config.RemoteConfig{
|
||||
Name: remote,
|
||||
Name: *remote,
|
||||
URLs: []string{repo.CloneURL},
|
||||
})
|
||||
if err != nil {
|
||||
Error.Fatalf("Error adding remote %s to local repository: %v", remote, err)
|
||||
logger.Error("Error adding remote %s to local repository: %v", *remote, err)
|
||||
return
|
||||
}
|
||||
log.Printf("Remote added to local repository")
|
||||
logger.Info("Remote added to local repository")
|
||||
}
|
||||
logger.Info("Done")
|
||||
}
|
||||
|
Reference in New Issue
Block a user