codeberg-pages-deploy/main.go
[BOT] pages deployer c63c30fa47
Fix: cleanup
2023-08-25 23:41:25 +02:00

203 lines
4.1 KiB
Go

// Copyright 2020 the Drone Authors. All rights reserved.
// Use of this source code is governed by the Blue Oak Model License
// that can be found in the LICENSE file.
package main
import (
"errors"
"fmt"
"log"
"os"
"os/exec"
"path/filepath"
"strings"
"github.com/kelseyhightower/envconfig"
"github.com/appleboy/drone-git-push/repo"
)
// Args provides plugin execution arguments.
type Args struct {
Folder string `required:"true" envconfig:"INPUT_FOLDER"`
SshKey string `envconfig:"INPUT_SSH_KEY"`
GitRemote string `envconfig:"INPUT_GIT_REMOTE"`
GitBranch string `envconfig:"INPUT_GIT_BRANCH"`
Token string `envconfig:"INPUT_TOKEN"`
Username string `envconfig:"INPUT_USERNAME"`
GithubToken string `envconfig:"GITHUB_TOKEN"`
GithubTokenActor string `envconfig:"GITHUB_ACTOR"`
GithubRepository string `envconfig:"GITHUB_REPOSITORY"`
GithubServerUrl string `envconfig:"GITHUB_SERVER_URL"`
}
const BRANCH_NAME = "pages"
func main() {
var args Args
if err := envconfig.Process("", &args); err != nil {
log.Fatalln(err)
}
if err := Exec(args); err != nil {
log.Fatalln(err)
}
}
// Exec executes the plugin.
func Exec(args Args) error {
if err := checkArgs(&args); err != nil {
return err
}
if args.SshKey != "" {
if err := repo.WriteKey(strings.ReplaceAll(args.SshKey, "\\n", "\n") + "\n"); err != nil {
return err
}
} else {
var err error
args.GitRemote, err = repo.WriteToken(args.GitRemote, args.Username, args.Token)
if err != nil {
return err
}
}
if err := writeConfig(args); err != nil {
return err
}
if err := copyFiles(args); err != nil {
return err
}
if err := initRepo(args); err != nil {
return err
}
if err := doCommit(args); err != nil {
return err
}
if err := push(args); err != nil {
return err
}
// write code here
return nil
}
func checkArgs(args *Args) error {
if args.Token == "" && args.GithubToken != "" {
args.Token = args.GithubToken
}
if args.Username == "" && args.GithubTokenActor != "" {
args.Username = args.GithubTokenActor
}
if (args.Token == "" || args.Username == "") && args.SshKey == "" {
return errors.New("(INPUT_TOKEN and INPUT_USERNAME) or INPUT_SSH_KEY is required!")
}
if folderInfo, err := os.Stat(args.Folder); os.IsNotExist(err) || !folderInfo.IsDir() {
return errors.New("INPUT_FOLDER is not a folder!")
}
if args.GitRemote == "" && (args.GithubServerUrl == "" || args.GithubRepository == "") {
return errors.New("INPUT_GIT_REMOTE is required!")
}
if args.GitRemote == "" {
args.GitRemote = fmt.Sprintf("%s/%s.git", args.GithubServerUrl, args.GithubRepository)
}
if args.GitBranch == "" {
args.GitBranch = BRANCH_NAME
}
return nil
}
func writeConfig(args Args) error {
if err := repo.GlobalName("[BOT] pages deployer").Run(); err != nil {
return err
}
if err := repo.GlobalUser("noreply@pages.bot").Run(); err != nil {
return err
}
return nil
}
func copyFiles(args Args) error {
if err := os.RemoveAll("/tmp/pages"); err != nil {
return err
}
if err := os.MkdirAll("/tmp", 777); err != nil {
return err
}
folder, err := filepath.Abs(args.Folder)
if err != nil {
return err
}
cmd := exec.Command("cp", "-r", folder, "/tmp/pages")
if err := execute(cmd); err != nil {
return err
}
if err := os.Chown("/tmp/pages", os.Getuid(), os.Getgid()); err != nil {
return err
}
if err := os.RemoveAll("/tmp/pages/.git"); err != nil {
return err
}
return os.Chdir("/tmp/pages")
}
func initRepo(args Args) error {
cmd := exec.Command(
"git",
"init", "-b", args.GitBranch, "/tmp/pages")
if err := execute(cmd); err != nil {
return err
}
return nil
}
func doCommit(args Args) error {
if err := execute(repo.Add()); err != nil {
return err
}
if err := execute(repo.ForceCommit("Update pages 🚀", true, "[BOT] pages deployer", "noreply@pages.bot")); err != nil {
return err
}
return nil
}
func push(args Args) error {
return execute(repo.RemotePushNamedBranch(args.GitRemote, args.GitBranch, args.GitBranch, true, false))
}
func execute(cmd *exec.Cmd) error {
fmt.Println("+", strings.Join(cmd.Args, " "))
cmd.Env = os.Environ()
cmd.Stderr = os.Stderr
cmd.Stdin = os.Stdin
return cmd.Run()
}