mirror of
https://github.com/splitsh/lite.git
synced 2024-05-02 14:42:46 +02:00
116 lines
2.9 KiB
Go
116 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/splitsh/lite/splitter"
|
|
)
|
|
|
|
var (
|
|
version = "dev"
|
|
)
|
|
|
|
type prefixesFlag []*splitter.Prefix
|
|
|
|
func (p *prefixesFlag) String() string {
|
|
return fmt.Sprint(*p)
|
|
}
|
|
|
|
func (p *prefixesFlag) Set(value string) error {
|
|
parts := strings.Split(value, ":")
|
|
from := parts[0]
|
|
to := ""
|
|
excludes := make([]string, 0)
|
|
if len(parts) >= 2 {
|
|
to = strings.TrimRight(parts[1], "/")
|
|
if len(parts) > 2 {
|
|
for _, exclude := range parts[2:] {
|
|
excludes = append(excludes, exclude)
|
|
}
|
|
}
|
|
}
|
|
|
|
// value must be unique
|
|
for _, prefix := range *p {
|
|
if prefix.To == to {
|
|
return fmt.Errorf("cannot have two prefix splits under the same directory: %s -> %s vs %s -> %s", prefix.From, prefix.To, from, to)
|
|
}
|
|
}
|
|
|
|
*p = append(*p, splitter.NewPrefix(from, to, excludes))
|
|
return nil
|
|
}
|
|
|
|
var prefixes prefixesFlag
|
|
var origin, target, commit, path, gitVersion string
|
|
var scratch, debug, progress, v bool
|
|
|
|
func init() {
|
|
flag.Var(&prefixes, "prefix", "The directory(ies) to split")
|
|
flag.StringVar(&origin, "origin", "HEAD", "The branch to split (optional, defaults to the current one)")
|
|
flag.StringVar(&target, "target", "", "The branch to create when split is finished (optional)")
|
|
flag.StringVar(&commit, "commit", "", "The commit at which to start the split (optional)")
|
|
flag.StringVar(&path, "path", ".", "The repository path (optional, current directory by default)")
|
|
flag.BoolVar(&scratch, "scratch", false, "Flush the cache (optional)")
|
|
flag.BoolVar(&debug, "debug", false, "Enable the debug mode (optional)")
|
|
flag.StringVar(&gitVersion, "git", "latest", "Simulate a given version of Git (optional)")
|
|
flag.BoolVar(&progress, "progress", false, "Show progress bar (optional, cannot be enabled when debug is enabled)")
|
|
flag.BoolVar(&v, "version", false, "Show version")
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
if v {
|
|
fmt.Fprintf(os.Stderr, "splitsh-lite version %s\n", version)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if len(prefixes) == 0 {
|
|
fmt.Fprintln(os.Stderr, "You must provide the directory to split via the --prefix flag")
|
|
os.Exit(1)
|
|
}
|
|
|
|
config := &splitter.Config{
|
|
Path: path,
|
|
Origin: origin,
|
|
Prefixes: prefixes,
|
|
Target: target,
|
|
Commit: commit,
|
|
Debug: debug,
|
|
Scratch: scratch,
|
|
GitVersion: gitVersion,
|
|
}
|
|
|
|
result := &splitter.Result{}
|
|
|
|
var ticker *time.Ticker
|
|
if progress && !debug {
|
|
ticker = time.NewTicker(time.Millisecond * 50)
|
|
go func() {
|
|
for range ticker.C {
|
|
fmt.Fprintf(os.Stderr, "%d commits created, %d commits traversed\r", result.Created(), result.Traversed())
|
|
}
|
|
}()
|
|
}
|
|
|
|
if err := splitter.Split(config, result); err != nil {
|
|
fmt.Fprintln(os.Stderr, err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
if ticker != nil {
|
|
ticker.Stop()
|
|
}
|
|
|
|
fmt.Fprintf(os.Stderr, "%d commits created, %d commits traversed, in %s\n", result.Created(), result.Traversed(), result.Duration(time.Millisecond))
|
|
|
|
if result.Head() != nil {
|
|
fmt.Println(result.Head().String())
|
|
}
|
|
}
|