download.go 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. // Copyright 2014 The Gogs Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package repo
  5. import (
  6. "io"
  7. "path"
  8. "github.com/gogits/git-shell"
  9. "github.com/gogits/gogs/modules/base"
  10. "github.com/gogits/gogs/modules/middleware"
  11. )
  12. func ServeData(ctx *middleware.Context, name string, reader io.Reader) error {
  13. buf := make([]byte, 1024)
  14. n, _ := reader.Read(buf)
  15. if n > 0 {
  16. buf = buf[:n]
  17. }
  18. _, isTextFile := base.IsTextFile(buf)
  19. if !isTextFile {
  20. _, isImageFile := base.IsImageFile(buf)
  21. if !isImageFile {
  22. ctx.Resp.Header().Set("Content-Disposition", "attachment; filename="+path.Base(ctx.Repo.TreeName))
  23. ctx.Resp.Header().Set("Content-Transfer-Encoding", "binary")
  24. }
  25. }
  26. ctx.Resp.Write(buf)
  27. _, err := io.Copy(ctx.Resp, reader)
  28. return err
  29. }
  30. func ServeBlob(ctx *middleware.Context, blob *git.Blob) error {
  31. dataRc, err := blob.Data()
  32. if err != nil {
  33. return err
  34. }
  35. return ServeData(ctx, ctx.Repo.TreeName, dataRc)
  36. }
  37. func SingleDownload(ctx *middleware.Context) {
  38. blob, err := ctx.Repo.Commit.GetBlobByPath(ctx.Repo.TreeName)
  39. if err != nil {
  40. if git.IsErrNotExist(err) {
  41. ctx.Handle(404, "GetBlobByPath", nil)
  42. } else {
  43. ctx.Handle(500, "GetBlobByPath", err)
  44. }
  45. return
  46. }
  47. if err = ServeBlob(ctx, blob); err != nil {
  48. ctx.Handle(500, "ServeBlob", err)
  49. }
  50. }