-
Notifications
You must be signed in to change notification settings - Fork 38
/
asset.go
124 lines (106 loc) · 3.16 KB
/
asset.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package interactor
import (
"context"
"net/url"
"path"
"github.com/reearth/reearth/server/internal/usecase"
"github.com/reearth/reearth/server/internal/usecase/gateway"
"github.com/reearth/reearth/server/internal/usecase/interfaces"
"github.com/reearth/reearth/server/internal/usecase/repo"
"github.com/reearth/reearth/server/pkg/asset"
"github.com/reearth/reearth/server/pkg/id"
"github.com/reearth/reearthx/account/accountdomain"
"github.com/reearth/reearthx/usecasex"
)
type Asset struct {
repos *repo.Container
gateways *gateway.Container
}
func NewAsset(r *repo.Container, g *gateway.Container) interfaces.Asset {
return &Asset{
repos: r,
gateways: g,
}
}
func (i *Asset) Fetch(ctx context.Context, assets []id.AssetID, operator *usecase.Operator) ([]*asset.Asset, error) {
return i.repos.Asset.FindByIDs(ctx, assets)
}
func (i *Asset) FindByWorkspace(ctx context.Context, tid accountdomain.WorkspaceID, keyword *string, sort *asset.SortType, p *usecasex.Pagination, operator *usecase.Operator) ([]*asset.Asset, *usecasex.PageInfo, error) {
return Run2(
ctx, operator, i.repos,
Usecase().WithReadableWorkspaces(tid),
func(ctx context.Context) ([]*asset.Asset, *usecasex.PageInfo, error) {
return i.repos.Asset.FindByWorkspace(ctx, tid, repo.AssetFilter{
Sort: sort,
Keyword: keyword,
Pagination: p,
})
},
)
}
func (i *Asset) Create(ctx context.Context, inp interfaces.CreateAssetParam, operator *usecase.Operator) (result *asset.Asset, err error) {
if inp.File == nil {
return nil, interfaces.ErrFileNotIncluded
}
ws, err := i.repos.Workspace.FindByID(ctx, inp.WorkspaceID)
if err != nil {
return nil, err
}
if !operator.IsWritableWorkspace(ws.ID()) {
return nil, interfaces.ErrOperationDenied
}
url, size, err := i.gateways.File.UploadAsset(ctx, inp.File)
if err != nil {
return nil, err
}
// enforce policy
if policyID := operator.Policy(ws.Policy()); policyID != nil {
p, err := i.repos.Policy.FindByID(ctx, *policyID)
if err != nil {
return nil, err
}
s, err := i.repos.Asset.TotalSizeByWorkspace(ctx, ws.ID())
if err != nil {
return nil, err
}
if err := p.EnforceAssetStorageSize(s + size); err != nil {
_ = i.gateways.File.RemoveAsset(ctx, url)
return nil, err
}
}
a, err := asset.New().
NewID().
Workspace(inp.WorkspaceID).
Name(path.Base(inp.File.Path)).
Size(size).
URL(url.String()).
Build()
if err != nil {
return nil, err
}
if err := i.repos.Asset.Save(ctx, a); err != nil {
return nil, err
}
return a, nil
}
func (i *Asset) Remove(ctx context.Context, aid id.AssetID, operator *usecase.Operator) (result id.AssetID, err error) {
return Run1(
ctx, operator, i.repos,
Usecase().Transaction(),
func(ctx context.Context) (id.AssetID, error) {
asset, err := i.repos.Asset.FindByID(ctx, aid)
if err != nil {
return aid, err
}
if ok := operator.IsWritableWorkspace(asset.Workspace()); !ok {
return aid, interfaces.ErrOperationDenied
}
if url, _ := url.Parse(asset.URL()); url != nil {
if err := i.gateways.File.RemoveAsset(ctx, url); err != nil {
return aid, err
}
}
return aid, i.repos.Asset.Remove(ctx, aid)
},
)
}