databasus/backend/internal/features/storages/repository.go

package storages

import (
	db "databasus-backend/internal/storage"

	"github.com/google/uuid"
	"gorm.io/gorm"
)

type StorageRepository struct{}

func (r *StorageRepository) Save(storage *Storage) (*Storage, error) {
	database := db.GetDb()

	err := database.Transaction(func(tx *gorm.DB) error {
		switch storage.Type {
		case StorageTypeLocal:
			if storage.LocalStorage != nil {
				storage.LocalStorage.StorageID = storage.ID
			}
		case StorageTypeS3:
			if storage.S3Storage != nil {
				storage.S3Storage.StorageID = storage.ID
			}
		case StorageTypeGoogleDrive:
			if storage.GoogleDriveStorage != nil {
				storage.GoogleDriveStorage.StorageID = storage.ID
			}
		case StorageTypeNAS:
			if storage.NASStorage != nil {
				storage.NASStorage.StorageID = storage.ID
			}
		case StorageTypeAzureBlob:
			if storage.AzureBlobStorage != nil {
				storage.AzureBlobStorage.StorageID = storage.ID
			}
		case StorageTypeFTP:
			if storage.FTPStorage != nil {
				storage.FTPStorage.StorageID = storage.ID
			}
		case StorageTypeSFTP:
			if storage.SFTPStorage != nil {
				storage.SFTPStorage.StorageID = storage.ID
			}
		case StorageTypeRclone:
			if storage.RcloneStorage != nil {
				storage.RcloneStorage.StorageID = storage.ID
			}
		}

		if storage.ID == uuid.Nil {
			if err := tx.Create(storage).
				Omit("LocalStorage", "S3Storage", "GoogleDriveStorage", "NASStorage", "AzureBlobStorage", "FTPStorage", "SFTPStorage", "RcloneStorage").
				Error; err != nil {
				return err
			}
		} else {
			if err := tx.Save(storage).
				Omit("LocalStorage", "S3Storage", "GoogleDriveStorage", "NASStorage", "AzureBlobStorage", "FTPStorage", "SFTPStorage", "RcloneStorage").
				Error; err != nil {
				return err
			}
		}

		switch storage.Type {
		case StorageTypeLocal:
			if storage.LocalStorage != nil {
				storage.LocalStorage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.LocalStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeS3:
			if storage.S3Storage != nil {
				storage.S3Storage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.S3Storage).Error; err != nil {
					return err
				}
			}
		case StorageTypeGoogleDrive:
			if storage.GoogleDriveStorage != nil {
				storage.GoogleDriveStorage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.GoogleDriveStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeNAS:
			if storage.NASStorage != nil {
				storage.NASStorage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.NASStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeAzureBlob:
			if storage.AzureBlobStorage != nil {
				storage.AzureBlobStorage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.AzureBlobStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeFTP:
			if storage.FTPStorage != nil {
				storage.FTPStorage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.FTPStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeSFTP:
			if storage.SFTPStorage != nil {
				storage.SFTPStorage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.SFTPStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeRclone:
			if storage.RcloneStorage != nil {
				storage.RcloneStorage.StorageID = storage.ID // Ensure ID is set
				if err := tx.Save(storage.RcloneStorage).Error; err != nil {
					return err
				}
			}
		}

		return nil
	})

	if err != nil {
		return nil, err
	}

	return storage, nil
}

func (r *StorageRepository) FindByID(id uuid.UUID) (*Storage, error) {
	var s Storage

	if err := db.
		GetDb().
		Preload("LocalStorage").
		Preload("S3Storage").
		Preload("GoogleDriveStorage").
		Preload("NASStorage").
		Preload("AzureBlobStorage").
		Preload("FTPStorage").
		Preload("SFTPStorage").
		Preload("RcloneStorage").
		Where("id = ?", id).
		First(&s).Error; err != nil {
		return nil, err
	}

	return &s, nil
}

func (r *StorageRepository) FindByWorkspaceID(workspaceID uuid.UUID) ([]*Storage, error) {
	var storages []*Storage

	if err := db.
		GetDb().
		Preload("LocalStorage").
		Preload("S3Storage").
		Preload("GoogleDriveStorage").
		Preload("NASStorage").
		Preload("AzureBlobStorage").
		Preload("FTPStorage").
		Preload("SFTPStorage").
		Preload("RcloneStorage").
		Where("workspace_id = ? OR is_system = TRUE", workspaceID).
		Order("name ASC").
		Find(&storages).Error; err != nil {
		return nil, err
	}

	return storages, nil
}

func (r *StorageRepository) Delete(s *Storage) error {
	return db.GetDb().Transaction(func(tx *gorm.DB) error {
		// Delete specific storage based on type
		switch s.Type {
		case StorageTypeLocal:
			if s.LocalStorage != nil {
				if err := tx.Delete(s.LocalStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeS3:
			if s.S3Storage != nil {
				if err := tx.Delete(s.S3Storage).Error; err != nil {
					return err
				}
			}
		case StorageTypeGoogleDrive:
			if s.GoogleDriveStorage != nil {
				if err := tx.Delete(s.GoogleDriveStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeNAS:
			if s.NASStorage != nil {
				if err := tx.Delete(s.NASStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeAzureBlob:
			if s.AzureBlobStorage != nil {
				if err := tx.Delete(s.AzureBlobStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeFTP:
			if s.FTPStorage != nil {
				if err := tx.Delete(s.FTPStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeSFTP:
			if s.SFTPStorage != nil {
				if err := tx.Delete(s.SFTPStorage).Error; err != nil {
					return err
				}
			}
		case StorageTypeRclone:
			if s.RcloneStorage != nil {
				if err := tx.Delete(s.RcloneStorage).Error; err != nil {
					return err
				}
			}
		}

		// Delete the main storage
		return tx.Delete(s).Error
	})
}