package git import ( "fmt" "io" "net/http" "gitlab.com/gitlab-org/gitaly/v14/proto/go/gitalypb" "gitlab.com/gitlab-org/gitlab-workhorse/internal/gitaly" "gitlab.com/gitlab-org/gitlab-workhorse/internal/helper" "gitlab.com/gitlab-org/gitlab-workhorse/internal/log" "gitlab.com/gitlab-org/gitlab-workhorse/internal/senddata" ) type snapshot struct { senddata.Prefix } type snapshotParams struct { GitalyServer gitaly.Server GetSnapshotRequest string } var ( SendSnapshot = &snapshot{"git-snapshot:"} ) func (s *snapshot) Inject(w http.ResponseWriter, r *http.Request, sendData string) { var params snapshotParams if err := s.Unpack(¶ms, sendData); err != nil { helper.Fail500(w, r, fmt.Errorf("SendSnapshot: unpack sendData: %v", err)) return } request := &gitalypb.GetSnapshotRequest{} if err := gitaly.UnmarshalJSON(params.GetSnapshotRequest, request); err != nil { helper.Fail500(w, r, fmt.Errorf("SendSnapshot: unmarshal GetSnapshotRequest: %v", err)) return } ctx, c, err := gitaly.NewRepositoryClient(r.Context(), params.GitalyServer) if err != nil { helper.Fail500(w, r, fmt.Errorf("SendSnapshot: gitaly.NewRepositoryClient: %v", err)) return } reader, err := c.SnapshotReader(ctx, request) if err != nil { helper.Fail500(w, r, fmt.Errorf("SendSnapshot: client.SnapshotReader: %v", err)) return } w.Header().Del("Content-Length") w.Header().Set("Content-Disposition", `attachment; filename="snapshot.tar"`) w.Header().Set("Content-Type", "application/x-tar") w.Header().Set("Content-Transfer-Encoding", "binary") w.Header().Set("Cache-Control", "private") w.WriteHeader(http.StatusOK) // Errors aren't detectable beyond this point if _, err := io.Copy(w, reader); err != nil { log.WithRequest(r).WithError(fmt.Errorf("SendSnapshot: copy gitaly output: %v", err)).Error() } }