137 lines
3.3 KiB
Go
137 lines
3.3 KiB
Go
package webfilesystem
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
)
|
|
|
|
type DirectoryData struct {
|
|
MongoId primitive.ObjectID `bson:"_id"`
|
|
Parent primitive.ObjectID `bson:"parent_id"`
|
|
Children []primitive.ObjectID `bson:"children_id"`
|
|
}
|
|
|
|
func (fs *WebFileSystem) CreateDirectory(dirPath string) (primitive.ObjectID, primitive.ObjectID, error) {
|
|
splittedPath := fs.SplitPath(dirPath)
|
|
parentPath := fs.GetParentPath(dirPath)
|
|
|
|
parentDirId, err := fs.FindFile(parentPath)
|
|
if err != nil {
|
|
return primitive.NilObjectID, primitive.NilObjectID, err
|
|
}
|
|
|
|
parentDirHeader, err := fs.readFSDocs(parentDirId, nil)
|
|
if err != nil {
|
|
return primitive.NilObjectID, primitive.NilObjectID, err
|
|
}
|
|
|
|
newDir := FileHeader{
|
|
MongoId: primitive.NewObjectID(),
|
|
Name: splittedPath[len(splittedPath)-1],
|
|
Type: "directory",
|
|
Icon: "",
|
|
}
|
|
newDirData := DirectoryData{
|
|
MongoId: primitive.NewObjectID(),
|
|
Parent: parentDirHeader.MongoId,
|
|
Children: []primitive.ObjectID{},
|
|
}
|
|
|
|
headerId, dataId, err := fs.writeFileToMongo(&newDir, &newDirData)
|
|
if err != nil {
|
|
return primitive.NilObjectID, primitive.NilObjectID, err
|
|
}
|
|
|
|
err = fs.AppendChildToDirectory(headerId, parentDirId)
|
|
if err != nil {
|
|
return primitive.NilObjectID, primitive.NilObjectID, err
|
|
}
|
|
return headerId, dataId, nil
|
|
}
|
|
|
|
func (fs *WebFileSystem) AppendChildToDirectory(childId primitive.ObjectID, parentId primitive.ObjectID) error {
|
|
parentDirData := DirectoryData{}
|
|
parentDir, err := fs.readFSDocs(parentId, &parentDirData)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
//TODO Check for duplicates?
|
|
parentDirData.Children = append(parentDirData.Children, childId)
|
|
|
|
err = fs.UpdateFileData(parentDir, parentDirData)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (fs *WebFileSystem) ListDir(dirPath string) ([]*FileHeader, error) {
|
|
dirId, err := fs.FindFile(dirPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
dirData := DirectoryData{}
|
|
_, err = fs.readFSDocs(dirId, &dirData)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// println(dirId.String())
|
|
// println(dirData.MongoId.String())
|
|
|
|
children := []*FileHeader{}
|
|
for _, childID := range dirData.Children {
|
|
childFile, err := fs.readFSDocs(childID, nil)
|
|
if err != nil {
|
|
// println(err.Error())
|
|
continue
|
|
}
|
|
|
|
children = append(children, childFile)
|
|
|
|
}
|
|
return children, nil
|
|
}
|
|
|
|
func (fs *WebFileSystem) FindFile(filePath string) (primitive.ObjectID, error) {
|
|
splittedPath := fs.SplitPath(filePath)
|
|
rootDir, _, err := fs.GetRootDir()
|
|
if err != nil {
|
|
return primitive.NilObjectID, err
|
|
}
|
|
// println(rootDir.Name)
|
|
lastFileID := rootDir.MongoId
|
|
for _, pathPart := range splittedPath[1:] {
|
|
// println(pathPart)
|
|
id, err := fs.findInChildren(pathPart, lastFileID)
|
|
if err != nil {
|
|
return primitive.NilObjectID, err
|
|
}
|
|
lastFileID = id
|
|
}
|
|
return lastFileID, nil
|
|
}
|
|
|
|
func (fs *WebFileSystem) findInChildren(fileName string, parentDirId primitive.ObjectID) (primitive.ObjectID, error) {
|
|
parentDirData := DirectoryData{}
|
|
_, err := fs.readFSDocs(parentDirId, &parentDirData)
|
|
if err != nil {
|
|
return primitive.NilObjectID, err
|
|
}
|
|
|
|
for _, childId := range parentDirData.Children {
|
|
childFileHeader, err := fs.readFSDocs(childId, nil)
|
|
if err != nil {
|
|
return primitive.NilObjectID, err
|
|
}
|
|
if childFileHeader.Name == fileName {
|
|
return childId, nil
|
|
}
|
|
}
|
|
|
|
return primitive.NilObjectID, errors.New("file not found")
|
|
}
|