mirror of
				https://codeberg.org/forgejo/forgejo.git
				synced 2025-11-04 00:11:04 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			87 lines
		
	
	
	
		
			2.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			87 lines
		
	
	
	
		
			2.1 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2017 The Gitea Authors. All rights reserved.
 | 
						|
// Use of this source code is governed by a MIT-style
 | 
						|
// license that can be found in the LICENSE file.
 | 
						|
 | 
						|
package repo
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"strings"
 | 
						|
 | 
						|
	"code.gitea.io/gitea/models"
 | 
						|
	"code.gitea.io/gitea/modules/context"
 | 
						|
	"code.gitea.io/gitea/modules/log"
 | 
						|
	"code.gitea.io/gitea/modules/setting"
 | 
						|
	"code.gitea.io/gitea/modules/upload"
 | 
						|
)
 | 
						|
 | 
						|
func renderAttachmentSettings(ctx *context.Context) {
 | 
						|
	ctx.Data["RequireDropzone"] = true
 | 
						|
	ctx.Data["IsAttachmentEnabled"] = setting.AttachmentEnabled
 | 
						|
	ctx.Data["AttachmentAllowedTypes"] = setting.AttachmentAllowedTypes
 | 
						|
	ctx.Data["AttachmentMaxSize"] = setting.AttachmentMaxSize
 | 
						|
	ctx.Data["AttachmentMaxFiles"] = setting.AttachmentMaxFiles
 | 
						|
}
 | 
						|
 | 
						|
// UploadAttachment response for uploading issue's attachment
 | 
						|
func UploadAttachment(ctx *context.Context) {
 | 
						|
	if !setting.AttachmentEnabled {
 | 
						|
		ctx.Error(404, "attachment is not enabled")
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	file, header, err := ctx.Req.FormFile("file")
 | 
						|
	if err != nil {
 | 
						|
		ctx.Error(500, fmt.Sprintf("FormFile: %v", err))
 | 
						|
		return
 | 
						|
	}
 | 
						|
	defer file.Close()
 | 
						|
 | 
						|
	buf := make([]byte, 1024)
 | 
						|
	n, _ := file.Read(buf)
 | 
						|
	if n > 0 {
 | 
						|
		buf = buf[:n]
 | 
						|
	}
 | 
						|
 | 
						|
	err = upload.VerifyAllowedContentType(buf, strings.Split(setting.AttachmentAllowedTypes, ","))
 | 
						|
	if err != nil {
 | 
						|
		ctx.Error(400, err.Error())
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	attach, err := models.NewAttachment(&models.Attachment{
 | 
						|
		UploaderID: ctx.User.ID,
 | 
						|
		Name:       header.Filename,
 | 
						|
	}, buf, file)
 | 
						|
	if err != nil {
 | 
						|
		ctx.Error(500, fmt.Sprintf("NewAttachment: %v", err))
 | 
						|
		return
 | 
						|
	}
 | 
						|
 | 
						|
	log.Trace("New attachment uploaded: %s", attach.UUID)
 | 
						|
	ctx.JSON(200, map[string]string{
 | 
						|
		"uuid": attach.UUID,
 | 
						|
	})
 | 
						|
}
 | 
						|
 | 
						|
// DeleteAttachment response for deleting issue's attachment
 | 
						|
func DeleteAttachment(ctx *context.Context) {
 | 
						|
	file := ctx.Query("file")
 | 
						|
	attach, err := models.GetAttachmentByUUID(file)
 | 
						|
	if !ctx.IsSigned || (ctx.User.ID != attach.UploaderID) {
 | 
						|
		ctx.Error(403)
 | 
						|
		return
 | 
						|
	}
 | 
						|
	if err != nil {
 | 
						|
		ctx.Error(400, err.Error())
 | 
						|
		return
 | 
						|
	}
 | 
						|
	err = models.DeleteAttachment(attach, true)
 | 
						|
	if err != nil {
 | 
						|
		ctx.Error(500, fmt.Sprintf("DeleteAttachment: %v", err))
 | 
						|
		return
 | 
						|
	}
 | 
						|
	ctx.JSON(200, map[string]string{
 | 
						|
		"uuid": attach.UUID,
 | 
						|
	})
 | 
						|
}
 |