anyproto/anytype-heart
1
0
Fork 0
mirror of https://github.com/anyproto/anytype-heart.git synced 2025-06-11 02:13:41 +09:00
Code Activity

GO-4969 inmemory exporter: do not allow files

Browse source
This commit is contained in:
Roman Khafizianov 2025-05-06 19:15:17 +02:00
parent 62cd6b7eda
commit 718f8e4ef3
No known key found for this signature in database
GPG key ID: F07A7D55A2684852
2 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
core/block/export/export.go
View file

@ -252,6 +252,7 @@ func (e *exportContext) getStateFilters(id string) *state.Filters {
func (e *exportContext) exportObject(ctx context.Context, objectId string) (string, error) {
e.reqIds = []string{objectId}
e.includeArchive = true
e.includeFiles = false
err := e.docsForExport(ctx)
if err != nil {
return "", err
@ -268,6 +269,16 @@ func (e *exportContext) exportObject(ctx context.Context, objectId string) (stri
docNamer = newNamer()
}
inMemoryWriter := &InMemoryWriter{fn: docNamer}
details, err := e.objectStore.SpaceIndex(e.spaceId).GetDetails(objectId)
if err != nil {
return "", err
}
// do not allow file export for in-memory writer
switch model.ObjectTypeLayout(details.GetInt64(bundle.RelationKeyLayout)) {
case model.ObjectType_file, model.ObjectType_image, model.ObjectType_video, model.ObjectType_audio:
return "", fmt.Errorf("file export is not allowed for in-memory writer")
}
err = e.writeDoc(ctx, inMemoryWriter, objectId, e.docs.transformToDetailsMap())
if err != nil {
return "", err

3
core/block/export/writer.go
View file

@ -173,6 +173,9 @@ func (d *InMemoryWriter) Path() string {
func (d *InMemoryWriter) WriteFile(filename string, r io.Reader, lastModifiedDate int64) (err error) {
d.m.Lock()
defer d.m.Unlock()
if filepath.Ext(filename) != ".md" && filepath.Ext(filename) != ".json" && filepath.Ext(filename) != ".pb" {
return fmt.Errorf("unsupported file type: %s", filename)
}
if d.data == nil {
d.data = make(map[string][]byte)
}