✨ Attachment has pool

pkg/internal/services/analyzer.go

@@ -99,7 +99,7 @@ func AnalyzeAttachment(file models.Attachment) error {
 
 	var start time.Time
 
-	// Do analyze job
+	// Do analyze jobs
 	if !file.IsAnalyzed || len(file.HashCode) == 0 {
 		destMap := viper.GetStringMap("destinations.temporary")
 

pkg/internal/services/attachments.go

@@ -28,7 +28,7 @@ func GetAttachmentByID(id uint) (models.Attachment, error) {
 	var attachment models.Attachment
 	if err := database.C.Where(models.Attachment{
 		BaseModel: models.BaseModel{ID: id},
-	}).Preload("Account").First(&attachment).Error; err != nil {
+	}).Preload("Pool").Preload("Account").First(&attachment).Error; err != nil {
 		return attachment, err
 	} else {
 		MaintainAttachmentCache()
@@ -42,7 +42,7 @@ func GetAttachmentByHash(hash string) (models.Attachment, error) {
 	var attachment models.Attachment
 	if err := database.C.Where(models.Attachment{
 		HashCode: hash,
-	}).First(&attachment).Error; err != nil {
+	}).Preload("Pool").First(&attachment).Error; err != nil {
 		return attachment, err
 	}
 	return attachment, nil
@@ -60,6 +60,7 @@ func CacheAttachment(id uint, item models.Attachment) {
 }
 
 func NewAttachmentMetadata(tx *gorm.DB, user models.Account, file *multipart.FileHeader, attachment models.Attachment) (models.Attachment, error) {
+	attachment.Rid = RandString(16)
 	attachment.Uuid = uuid.NewString()
 	attachment.Size = file.Size
 	attachment.Name = file.Filename
@@ -104,6 +105,13 @@ func TryLinkAttachment(tx *gorm.DB, og models.Attachment, hash string) (bool, er
 		return false, err
 	}
 
+	if prev.PoolID != nil && og.PoolID != nil && prev.PoolID != og.PoolID {
+		if !prev.Pool.Config.Data().AllowCrossPoolEgress || !og.Pool.Config.Data().AllowCrossPoolIngress {
+			// Pool config doesn't allow reference
+			return false, nil
+		}
+	}
+
 	prev.RefCount++
 	og.RefID = &prev.ID
 	og.Uuid = prev.Uuid

pkg/internal/services/pools.go

@@ -21,6 +21,14 @@ func GetAttachmentPool(id uint) (models.AttachmentPool, error) {
 	return pool, nil
 }
 
+func GetAttachmentPoolByAlias(alias string) (models.AttachmentPool, error) {
+	var pool models.AttachmentPool
+	if err := database.C.Where("alias = ?", alias).First(&pool).Error; err != nil {
+		return pool, err
+	}
+	return pool, nil
+}
+
 func GetAttachmentPoolWithUser(id uint, userId uint) (models.AttachmentPool, error) {
 	var pool models.AttachmentPool
 	if err := database.C.Where("id = ? AND account_id = ?", id, userId).First(&pool).Error; err != nil {

pkg/internal/services/random_id.go

@@ -0,0 +1,31 @@
+package services
+
+import (
+	"math/rand"
+	"strings"
+)
+
+const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
+const (
+	letterIdxBits = 6
+	letterIdxMask = 1<<letterIdxBits - 1
+	letterIdxMax  = 63 / letterIdxBits
+)
+
+func RandString(length int) string {
+	builder := strings.Builder{}
+	builder.Grow(length)
+	for idx, cache, remain := length-1, rand.Int63(), letterIdxMax; idx >= 0; {
+		if remain == 0 {
+			cache, remain = rand.Int63(), letterIdxMax
+		}
+		if idx := int(cache & letterIdxMask); idx < len(letterBytes) {
+			builder.WriteByte(letterBytes[idx])
+			idx--
+		}
+		cache >>= letterIdxBits
+		remain--
+	}
+
+	return builder.String()
+}