download.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  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-module"
  9. "github.com/gogits/gogs/modules/base"
  10. "github.com/gogits/gogs/modules/context"
  11. )
  12. func ServeData(ctx *context.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. } else {
  26. ctx.Resp.Header().Set("Content-Type", "text/plain")
  27. }
  28. ctx.Resp.Write(buf)
  29. _, err := io.Copy(ctx.Resp, reader)
  30. return err
  31. }
  32. func ServeBlob(ctx *context.Context, blob *git.Blob) error {
  33. dataRc, err := blob.Data()
  34. if err != nil {
  35. return err
  36. }
  37. return ServeData(ctx, ctx.Repo.TreeName, dataRc)
  38. }
  39. func SingleDownload(ctx *context.Context) {
  40. blob, err := ctx.Repo.Commit.GetBlobByPath(ctx.Repo.TreeName)
  41. if err != nil {
  42. if git.IsErrNotExist(err) {
  43. ctx.Handle(404, "GetBlobByPath", nil)
  44. } else {
  45. ctx.Handle(500, "GetBlobByPath", err)
  46. }
  47. return
  48. }
  49. if err = ServeBlob(ctx, blob); err != nil {
  50. ctx.Handle(500, "ServeBlob", err)
  51. }
  52. }