feat(server): Add data deletion process [VIZ-1419]

server/internal/infrastructure/adapter/plugin.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 
+	"github.com/reearth/reearth/server/internal/usecase/gateway"
 	"github.com/reearth/reearth/server/internal/usecase/repo"
 	"github.com/reearth/reearth/server/pkg/id"
 	"github.com/reearth/reearth/server/pkg/plugin"
@@ -75,3 +76,10 @@ func (r *pluginRepo) Remove(ctx context.Context, p id.PluginID) error {
 	}
 	return r.writer.Remove(ctx, p)
 }
+
+func (r *pluginRepo) RemoveBySceneWithFile(ctx context.Context, sid id.SceneID, f gateway.File) error {
+	if r.writer == nil {
+		return errors.New("cannot write")
+	}
+	return r.writer.RemoveBySceneWithFile(ctx, sid, f)
+}

server/internal/infrastructure/adapter/property_schema.go

@@ -89,3 +89,10 @@ func (r *propertySchema) RemoveAll(ctx context.Context, p []id.PropertySchemaID)
 	}
 	return r.writer.RemoveAll(ctx, p)
 }
+
+func (r *propertySchema) RemoveByScene(ctx context.Context, sceneID id.SceneID) error {
+	if r.writer == nil {
+		return rerror.ErrInternalByWithContext(ctx, errors.New("writer is not set"))
+	}
+	return r.writer.RemoveByScene(ctx, sceneID)
+}

server/internal/infrastructure/fs/plugin.go

@@ -6,6 +6,7 @@ import (
 	"path/filepath"
 	"regexp"
 
+	"github.com/reearth/reearth/server/internal/usecase/gateway"
 	"github.com/reearth/reearth/server/internal/usecase/repo"
 	"github.com/reearth/reearth/server/pkg/id"
 	"github.com/reearth/reearth/server/pkg/plugin"
@@ -65,6 +66,10 @@ func (r *pluginRepo) Remove(ctx context.Context, pid id.PluginID) error {
 	return rerror.ErrInternalByWithContext(ctx, errors.New("read only"))
 }
 
+func (r *pluginRepo) RemoveBySceneWithFile(ctx context.Context, sid id.SceneID, f gateway.File) error {
+	return rerror.ErrInternalByWithContext(ctx, errors.New("read only"))
+}
+
 var translationFileNameRegexp = regexp.MustCompile(`reearth_([a-zA-Z]+(?:-[a-zA-Z]+)?).yml`)
 
 func readPluginManifest(ctx context.Context, fs afero.Fs, pid id.PluginID) (*manifest.Manifest, error) {

server/internal/infrastructure/fs/property_schema.go

@@ -80,3 +80,7 @@ func (r *propertySchema) Remove(ctx context.Context, pid id.PropertySchemaID) er
 func (r *propertySchema) RemoveAll(ctx context.Context, pid []id.PropertySchemaID) error {
 	return rerror.ErrInternalByWithContext(ctx, errors.New("read only"))
 }
+
+func (r *propertySchema) RemoveByScene(ctx context.Context, sceneID id.SceneID) error {
+	return rerror.ErrInternalByWithContext(ctx, errors.New("read only"))
+}

server/internal/infrastructure/memory/asset.go

@@ -5,6 +5,7 @@ import (
 	"sort"
 	"strings"
 
+	"github.com/reearth/reearth/server/internal/usecase/gateway"
 	"github.com/reearth/reearth/server/internal/usecase/repo"
 	"github.com/reearth/reearth/server/pkg/asset"
 	"github.com/reearth/reearth/server/pkg/id"
@@ -143,3 +144,15 @@ func (r *Asset) Remove(_ context.Context, id id.AssetID) error {
 	r.data.Delete(id)
 	return nil
 }
+
+func (r *Asset) RemoveByProjectWithFile(ctx context.Context, pid id.ProjectID, f gateway.File) error {
+	r.data.FindAll(func(id id.AssetID, a *asset.Asset) bool {
+		if r.f.CanWrite(a.Workspace()) {
+			if a.Project() != nil && *a.Project() == pid {
+				r.data.Delete(id)
+			}
+		}
+		return true
+	})
+	return nil
+}

server/internal/infrastructure/memory/plugin.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"sync"
 
+	"github.com/reearth/reearth/server/internal/usecase/gateway"
 	"github.com/reearth/reearth/server/internal/usecase/repo"
 	"github.com/reearth/reearth/server/pkg/builtin"
 	"github.com/reearth/reearth/server/pkg/id"
@@ -117,3 +118,21 @@ func (r *Plugin) Remove(ctx context.Context, id id.PluginID) error {
 
 	return nil
 }
+
+func (r *Plugin) RemoveBySceneWithFile(cts context.Context, sid id.SceneID, f gateway.File) error {
+	if !r.f.CanWrite(sid) {
+		return nil
+	}
+
+	r.lock.Lock()
+	defer r.lock.Unlock()
+
+	newData := make([]*plugin.Plugin, 0, len(r.data))
+	for _, p := range r.data {
+		if s := p.Scene(); s == nil || *s != sid {
+			newData = append(newData, p)
+		}
+	}
+	r.data = newData
+	return nil
+}

server/internal/infrastructure/memory/property_schema.go

@@ -146,3 +146,19 @@ func (r *PropertySchema) RemoveAll(ctx context.Context, ids []id.PropertySchemaI
 	}
 	return nil
 }
+
+func (r *PropertySchema) RemoveByScene(ctx context.Context, sceneID id.SceneID) error {
+	if !r.f.CanWrite(sceneID) {
+		return nil
+	}
+
+	r.lock.Lock()
+	defer r.lock.Unlock()
+
+	for lid, p := range r.data {
+		if s := p.Scene(); s != nil && *s == sceneID {
+			delete(r.data, lid)
+		}
+	}
+	return nil
+}

server/internal/infrastructure/mongo/asset.go

@@ -4,11 +4,14 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"log"
+	"net/url"
 	"regexp"
 	"strings"
 
 	"github.com/reearth/reearth/server/internal/adapter"
 	"github.com/reearth/reearth/server/internal/infrastructure/mongo/mongodoc"
+	"github.com/reearth/reearth/server/internal/usecase/gateway"
 	"github.com/reearth/reearth/server/internal/usecase/repo"
 	"github.com/reearth/reearth/server/pkg/asset"
 	"github.com/reearth/reearth/server/pkg/id"
@@ -153,6 +156,42 @@ func (r *Asset) Remove(ctx context.Context, id id.AssetID) error {
 	}))
 }
 
+func (r *Asset) RemoveByProjectWithFile(ctx context.Context, pid id.ProjectID, f gateway.File) error {
+
+	projectAssets, err := r.find(ctx, bson.M{
+		"coresupport": true,
+		"project":     pid.String(),
+	})
+	if err != nil {
+		return err
+	}
+
+	for _, a := range projectAssets {
+
+		if !r.f.CanWrite(a.Workspace()) {
+			return repo.ErrOperationDenied
+		}
+
+		aPath, err := url.Parse(a.URL())
+		if err != nil {
+			continue
+		}
+
+		err = f.RemoveAsset(ctx, aPath)
+		if err != nil {
+			log.Print(err.Error())
+		}
+
+		err = r.Remove(ctx, a.ID())
+		if err != nil {
+			log.Print(err.Error())
+		}
+
+	}
+
+	return nil
+}
+
 func (r *Asset) paginate(ctx context.Context, filter any, sort *asset.SortType, pagination *usecasex.Pagination) ([]*asset.Asset, *usecasex.PageInfo, error) {
 	var usort *usecasex.Sort
 	if sort != nil {

server/internal/infrastructure/mongo/plugin.go

@@ -3,10 +3,12 @@ package mongo
 import (
 	"context"
 	"errors"
+	"log"
 
 	"go.mongodb.org/mongo-driver/bson"
 
 	"github.com/reearth/reearth/server/internal/infrastructure/mongo/mongodoc"
+	"github.com/reearth/reearth/server/internal/usecase/gateway"
 	"github.com/reearth/reearth/server/internal/usecase/repo"
 	"github.com/reearth/reearth/server/pkg/builtin"
 	"github.com/reearth/reearth/server/pkg/id"
@@ -102,6 +104,31 @@ func (r *Plugin) Remove(ctx context.Context, id id.PluginID) error {
 	return r.client.RemoveOne(ctx, r.writeFilter(bson.M{"id": id.String()}))
 }
 
+func (r *Plugin) RemoveBySceneWithFile(ctx context.Context, sid id.SceneID, f gateway.File) error {
+	plugins, err := r.find(ctx, bson.M{"scene": sid.String()})
+	if err != nil {
+		return err
+	}
+
+	for _, pl := range plugins {
+
+		if p := builtin.GetPlugin(pl.ID()); p != nil {
+			continue
+		}
+
+		if err := f.RemovePlugin(ctx, pl.ID()); err != nil {
+			log.Print(err.Error())
+		}
+
+		if err := r.Remove(ctx, pl.ID()); err != nil {
+			log.Print(err.Error())
+		}
+
+	}
+
+	return nil
+}
+
 func (r *Plugin) find(ctx context.Context, filter any) ([]*plugin.Plugin, error) {
 	c := mongodoc.NewPluginConsumer(r.f.Readable)
 	if err := r.client.Find(ctx, filter, c); err != nil {

server/internal/infrastructure/mongo/property_schema.go

@@ -124,6 +124,10 @@ func (r *PropertySchema) RemoveAll(ctx context.Context, ids []id.PropertySchemaI
 	}))
 }
 
+func (r *PropertySchema) RemoveByScene(ctx context.Context, id id.SceneID) error {
+	return r.client.RemoveAll(ctx, r.writeFilter(bson.M{"scene": id.String()}))
+}
+
 func (r *PropertySchema) find(ctx context.Context, dst property.SchemaList, filter any) (property.SchemaList, error) {
 	c := mongodoc.NewPropertySchemaConsumer(r.f.Readable)
 	if err := r.client.Find(ctx, filter, c); err != nil {

server/internal/usecase/interactor/common.go

@@ -143,9 +143,15 @@ func (i commonSceneLock) ReleaseSceneLock(ctx context.Context, s id.SceneID) {
 }
 
 type SceneDeleter struct {
-	Scene     repo.Scene
-	SceneLock repo.SceneLock
-	Property  repo.Property
+	Scene          repo.Scene
+	SceneLock      repo.SceneLock
+	Property       repo.Property
+	PropertySchema repo.PropertySchema
+	NLSLayer       repo.NLSLayer
+	Plugin         repo.Plugin
+	Storytelling   repo.Storytelling
+	Style          repo.Style
+	File           gateway.File
 }
 
 func (d SceneDeleter) Delete(ctx context.Context, s *scene.Scene, force bool) error {
@@ -164,11 +170,36 @@ func (d SceneDeleter) Delete(ctx context.Context, s *scene.Scene, force bool) er
 		}
 	}
 
+	// Delete nlsLayer
+	if err := d.NLSLayer.RemoveByScene(ctx, s.ID()); err != nil {
+		return err
+	}
+
+	// Delete plugin
+	if err := d.Plugin.RemoveBySceneWithFile(ctx, s.ID(), d.File); err != nil {
+		return err
+	}
+
 	// Delete property
 	if err := d.Property.RemoveByScene(ctx, s.ID()); err != nil {
 		return err
 	}
 
+	// Delete propertyschema
+	if err := d.PropertySchema.RemoveByScene(ctx, s.ID()); err != nil {
+		return err
+	}
+
+	// Delete storytelling
+	if err := d.Storytelling.RemoveByScene(ctx, s.ID()); err != nil {
+		return err
+	}
+
+	// Delete style
+	if err := d.Style.RemoveByScene(ctx, s.ID()); err != nil {
+		return err
+	}
+
 	// Release scene lock
 	if err := d.SceneLock.SaveLock(ctx, s.ID(), scene.LockModeFree); err != nil {
 		return err
@@ -186,6 +217,7 @@ type ProjectDeleter struct {
 	SceneDeleter
 	File    gateway.File
 	Project repo.Project
+	Asset   repo.Asset
 }
 
 func (d ProjectDeleter) Delete(ctx context.Context, prj *project.Project, force bool, operator *usecase.Operator) error {
@@ -199,6 +231,11 @@ func (d ProjectDeleter) Delete(ctx context.Context, prj *project.Project, force
 		return err
 	}
 
+	// Delete assets
+	if err := d.Asset.RemoveByProjectWithFile(ctx, prj.ID(), d.File); err != nil {
+		return err
+	}
+
 	// Delete scene
 	if err := d.SceneDeleter.Delete(ctx, s, force); err != nil {
 		return err