forked from Shiloh/githaven
dcd9fc7ee8
Follow #21464 Make all git command arguments strictly safe. Most changes are one-to-one replacing, keep all existing logic.
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
// Copyright 2015 The Gogs Authors. All rights reserved.
|
|
// Copyright 2020 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package git
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// ArchiveType archive types
|
|
type ArchiveType int
|
|
|
|
const (
|
|
// ZIP zip archive type
|
|
ZIP ArchiveType = iota + 1
|
|
// TARGZ tar gz archive type
|
|
TARGZ
|
|
// BUNDLE bundle archive type
|
|
BUNDLE
|
|
)
|
|
|
|
// String converts an ArchiveType to string
|
|
func (a ArchiveType) String() string {
|
|
switch a {
|
|
case ZIP:
|
|
return "zip"
|
|
case TARGZ:
|
|
return "tar.gz"
|
|
case BUNDLE:
|
|
return "bundle"
|
|
}
|
|
return "unknown"
|
|
}
|
|
|
|
// CreateArchive create archive content to the target path
|
|
func (repo *Repository) CreateArchive(ctx context.Context, format ArchiveType, target io.Writer, usePrefix bool, commitID string) error {
|
|
if format.String() == "unknown" {
|
|
return fmt.Errorf("unknown format: %v", format)
|
|
}
|
|
|
|
cmd := NewCommand(ctx, "archive")
|
|
if usePrefix {
|
|
cmd.AddArguments(CmdArg("--prefix=" + filepath.Base(strings.TrimSuffix(repo.Path, ".git")) + "/"))
|
|
}
|
|
cmd.AddArguments(CmdArg("--format=" + format.String()))
|
|
cmd.AddDynamicArguments(commitID)
|
|
|
|
var stderr strings.Builder
|
|
err := cmd.Run(&RunOpts{
|
|
Dir: repo.Path,
|
|
Stdout: target,
|
|
Stderr: &stderr,
|
|
})
|
|
if err != nil {
|
|
return ConcatenateError(err, stderr.String())
|
|
}
|
|
return nil
|
|
}
|