Added attachments to the releases API

models/attachment.go

@@ -12,6 +12,7 @@ import (
 	"path"
 	"time"
 
+	api "code.gitea.io/sdk/gitea"
 	"github.com/go-xorm/xorm"
 	gouuid "github.com/satori/go.uuid"
 
@@ -58,6 +59,20 @@ func (a *Attachment) IncreaseDownloadCount() error {
 	return nil
 }
 
+// GetSize gets the size of the attachment in bytes
+func (a *Attachment) GetSize() (int64, error) {
+	f, err := os.Open(a.LocalPath())
+	defer f.Close()
+	if err != nil {
+		return 0, err
+	}
+	info, err := f.Stat()
+	if err != nil {
+		return 0, err
+	}
+	return info.Size(), nil
+}
+
 // AttachmentLocalPath returns where attachment is stored in local file
 // system based on given UUID.
 func AttachmentLocalPath(uuid string) string {
@@ -69,6 +84,23 @@ func (a *Attachment) LocalPath() string {
 	return AttachmentLocalPath(a.UUID)
 }
 
+// APIFormat converts a Attachment to an api.Attachment
+func (a *Attachment) APIFormat() *api.Attachment {
+	apiAttachment := &api.Attachment{
+		ID:            a.ID,
+		Created:       a.Created,
+		Name:          a.Name,
+		UUID:          a.UUID,
+		DownloadURL:   setting.AppURL + "attachments/" + a.UUID,
+		DownloadCount: a.DownloadCount,
+	}
+	fileSize, err := a.GetSize()
+	if err == nil {
+		apiAttachment.Size = fileSize
+	}
+	return apiAttachment
+}
+
 // NewAttachment creates a new attachment object.
 func NewAttachment(name string, buf []byte, file multipart.File) (_ *Attachment, err error) {
 	attach := &Attachment{
@@ -126,6 +158,18 @@ func GetAttachmentByUUID(uuid string) (*Attachment, error) {
 	return getAttachmentByUUID(x, uuid)
 }
 
+// GetAttachmentByD returns attachment by given ID.
+func GetAttachmentByID(id int64) (*Attachment, error) {
+	attach := &Attachment{ID: id}
+	has, err := x.Get(attach)
+	if err != nil {
+		return nil, err
+	} else if !has {
+		return nil, ErrAttachmentNotExist{id, ""}
+	}
+	return attach, nil
+}
+
 func getAttachmentsByIssueID(e Engine, issueID int64) ([]*Attachment, error) {
 	attachments := make([]*Attachment, 0, 10)
 	return attachments, e.Where("issue_id = ? AND comment_id = 0", issueID).Find(&attachments)
@@ -142,6 +186,12 @@ func GetAttachmentsByCommentID(commentID int64) ([]*Attachment, error) {
 	return attachments, x.Where("comment_id=?", commentID).Find(&attachments)
 }
 
+// GetAttachmentsByReleaseID returns all attachments of a release
+func GetAttachmentsByReleaseID(releaseId int64) ([]*Attachment, error) {
+	attachments := make([]*Attachment, 0, 10)
+	return attachments, x.Where("release_id=?", releaseId).Find(&attachments)
+}
+
 // DeleteAttachment deletes the given attachment and optionally the associated file.
 func DeleteAttachment(a *Attachment, remove bool) error {
 	_, err := DeleteAttachments([]*Attachment{a}, remove)

models/release.go

@@ -71,6 +71,14 @@ func (r *Release) loadAttributes(e Engine) error {
 			return err
 		}
 	}
+	// load the attachments of this release
+	if r.Attachments == nil {
+		attachments, err := GetAttachmentsByReleaseID(r.ID)
+		if err != nil {
+			return err
+		}
+		r.Attachments = attachments
+	}
 	return nil
 }
 
@@ -97,6 +105,10 @@ func (r *Release) TarURL() string {
 
 // APIFormat convert a Release to api.Release
 func (r *Release) APIFormat() *api.Release {
+	apiAttachments := make([]*api.Attachment, len(r.Attachments))
+	for i := range r.Attachments {
+		apiAttachments[i] = r.Attachments[i].APIFormat()
+	}
 	return &api.Release{
 		ID:           r.ID,
 		TagName:      r.TagName,
@@ -110,6 +122,7 @@ func (r *Release) APIFormat() *api.Release {
 		CreatedAt:    r.Created,
 		PublishedAt:  r.Created,
 		Publisher:    r.Publisher.APIFormat(),
+		Attachments:  apiAttachments,
 	}
 }
 

routers/api/v1/api.go

@@ -420,9 +420,15 @@ func RegisterRoutes(m *macaron.Macaron) {
 				m.Group("/releases", func() {
 					m.Combo("").Get(repo.ListReleases).
 						Post(bind(api.CreateReleaseOption{}), repo.CreateRelease)
-					m.Combo("/:id").Get(repo.GetRelease).
-						Patch(bind(api.EditReleaseOption{}), repo.EditRelease).
-						Delete(repo.DeleteRelease)
+					m.Group("/:id", func() {
+						m.Combo("").Get(repo.GetRelease).
+							Patch(bind(api.EditReleaseOption{}), repo.EditRelease).
+							Delete(repo.DeleteRelease)
+						m.Group("/assets", func() {
+							m.Combo("").Get(repo.ListReleaseAttachments)
+							m.Combo("/:assetId").Get(repo.GetReleaseAttachment)
+						})
+					})
 				})
 				m.Post("/mirror-sync", repo.MirrorSync)
 				m.Get("/editorconfig/:filename", context.RepoRef(), repo.GetEditorconfig)

routers/api/v1/repo/release.go

@@ -32,6 +32,56 @@ func GetRelease(ctx *context.APIContext) {
 	ctx.JSON(200, release.APIFormat())
 }
 
+// ListReleaseAttachments get all the attachments of a release
+func ListReleaseAttachments(ctx *context.APIContext) {
+	id := ctx.ParamsInt64(":id")
+	release, err := models.GetReleaseByID(id)
+	if err != nil {
+		ctx.Error(500, "GetReleaseByID", err)
+		return
+	}
+	if release.RepoID != ctx.Repo.Repository.ID {
+		ctx.Status(404)
+		return
+	}
+	// load the attachments of this release
+	attachments, err := models.GetAttachmentsByReleaseID(id)
+	if err != nil {
+		ctx.Error(500, "GetAttachmentsByReleaseID", err)
+		return
+	}
+	// build the attachment list
+	apiAttachments := make([]*api.Attachment, len(attachments))
+	for i := range attachments {
+		apiAttachments[i] = attachments[i].APIFormat()
+	}
+	ctx.JSON(200, apiAttachments)
+}
+
+// GetReleaseAttachment get a single attachment of a release
+func GetReleaseAttachment(ctx *context.APIContext) {
+	id := ctx.ParamsInt64(":id")
+	attachmentId := ctx.ParamsInt64(":assetId")
+	release, err := models.GetReleaseByID(id)
+	if err != nil {
+		ctx.Error(500, "GetReleaseByID", err)
+		return
+	}
+	if release.RepoID != ctx.Repo.Repository.ID {
+		ctx.Status(404)
+		return
+	}
+	// load the attachments of this release
+	attachment, err := models.GetAttachmentByID(attachmentId)
+	// if the attachment was not found, or it was found but is not associated with this release, then throw 404
+	if err != nil || id != attachment.ReleaseID {
+		ctx.Status(404)
+		return
+	}
+
+	ctx.JSON(200, attachment.APIFormat())
+}
+
 // ListReleases list a repository's releases
 func ListReleases(ctx *context.APIContext) {
 	releases, err := models.GetReleasesByRepoID(ctx.Repo.Repository.ID, 1, 2147483647)