download.go 1.4 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-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. if !base.IsTextFile(buf) {
  19. if !base.IsImageFile(buf) {
  20. ctx.Resp.Header().Set("Content-Disposition", "attachment; filename=\""+path.Base(ctx.Repo.TreePath)+"\"")
  21. ctx.Resp.Header().Set("Content-Transfer-Encoding", "binary")
  22. }
  23. } else if !ctx.QueryBool("render") {
  24. ctx.Resp.Header().Set("Content-Type", "text/plain; charset=utf-8")
  25. }
  26. ctx.Resp.Write(buf)
  27. _, err := io.Copy(ctx.Resp, reader)
  28. return err
  29. }
  30. func ServeBlob(ctx *context.Context, blob *git.Blob) error {
  31. dataRc, err := blob.Data()
  32. if err != nil {
  33. return err
  34. }
  35. return ServeData(ctx, ctx.Repo.TreePath, dataRc)
  36. }
  37. func SingleDownload(ctx *context.Context) {
  38. blob, err := ctx.Repo.Commit.GetBlobByPath(ctx.Repo.TreePath)
  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. }