76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package commands
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var exportCmd = &cobra.Command{
|
|
Use: "export",
|
|
Short: "export source code to public repo",
|
|
Run: exportCode,
|
|
}
|
|
|
|
var (
|
|
flagPush bool
|
|
flagMoveToMaster bool
|
|
)
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(exportCmd)
|
|
|
|
exportCmd.Flags().BoolVar(&flagPush, "push", false, "push to public repo")
|
|
exportCmd.Flags().BoolVar(&flagMoveToMaster, "move-to-master", true, "move to master after completing export")
|
|
}
|
|
|
|
func git(args ...string) {
|
|
log.Println("git", strings.Join(args, " "))
|
|
|
|
cmd := exec.Command("git", args...)
|
|
cmd.Stderr = os.Stderr
|
|
cmd.Stdout = os.Stdout
|
|
|
|
if err := cmd.Run(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func exportCode(cmd *cobra.Command, args []string) {
|
|
random := make([]byte, 4)
|
|
if _, err := io.ReadFull(rand.Reader, random); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpBranch := "temp_" + hex.EncodeToString(random)
|
|
|
|
git("checkout", "-b", tmpBranch)
|
|
git("reset", "public")
|
|
|
|
privateFiles := listPrivateFiles(".")
|
|
for _, f := range privateFiles {
|
|
log.Printf("rm %s", f)
|
|
if err := os.Remove(f); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
git("checkout", "public")
|
|
git("branch", "-D", tmpBranch)
|
|
|
|
git("add", "-A")
|
|
git("commit", "-m", "export public files", "--allow-empty")
|
|
|
|
if flagPush {
|
|
git("push", "public", "public:master")
|
|
}
|
|
|
|
if flagMoveToMaster {
|
|
git("checkout", "master")
|
|
}
|
|
}
|