mugit: internal/git/gitx/archive.go (b883855)

mugit/internal/git/gitx/archive.go (view raw)

Oleksandr Smirnov gitx: refactor once again..., 3 months ago

package gitx

import (
	"context"
	"fmt"
	"io"
	"regexp"
	"strings"
)

// ArchiveTar generates a tarball of a git ref.
func ArchiveTar(ctx context.Context, repoDir, ref string, out io.Writer) error {
	if !isValidRef(ref) {
		return fmt.Errorf("invalid ref: %s", ref)
	}

	if err := gitCmd(ctx, cmdOpts{
		Cmd:     []string{"archive", "--format=tar.gz", ref},
		RepoDir: repoDir,
		Stdout:  out,
	}); err != nil {
		return fmt.Errorf("git archive %s: %w", ref, err)
	}

	return nil
}

var isValidRefRe = regexp.MustCompile(`^[a-zA-Z0-9._/-]+$`)

func isValidRef(ref string) bool {
	if ref == "" || strings.Contains(ref, "..") {
		return false
	}
	return isValidRefRe.MatchString(ref)
}

1	package gitx
2
3	import (
4	"context"
5	"fmt"
6	"io"
7	"regexp"
8	"strings"
9	)
10
11	// ArchiveTar generates a tarball of a git ref.
12	func ArchiveTar(ctx context.Context, repoDir, ref string, out io.Writer) error {
13	if !isValidRef(ref) {
14	return fmt.Errorf("invalid ref: %s", ref)
15	}
16
17	if err := gitCmd(ctx, cmdOpts{
18	Cmd: []string{"archive", "--format=tar.gz", ref},
19	RepoDir: repoDir,
20	Stdout: out,
21	}); err != nil {
22	return fmt.Errorf("git archive %s: %w", ref, err)
23	}
24
25	return nil
26	}
27
28	var isValidRefRe = regexp.MustCompile(`^[a-zA-Z0-9._/-]+$`)
29
30	func isValidRef(ref string) bool {
31	if ref == "" \|\| strings.Contains(ref, "..") {
32	return false
33	}
34	return isValidRefRe.MatchString(ref)
35	}