func MoveFile(wsConn *websocket.Conn, fileMoveRequest fileRequests.FileMoveRequest) {
session, collection := managers.GetMGoCollection("Files")
defer session.Close()
// Check that file exists
file, err := GetFileById(fileMoveRequest.BaseRequest.ResId)
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-300, fileMoveRequest.BaseRequest.Tag, nil))
return
}
file.Version++
err = collection.Update(bson.M{"_id": fileMoveRequest.BaseRequest.ResId}, bson.M{"$set": bson.M{"relative_path": fileMoveRequest.NewPath, "version": file.Version}})
if err != nil {
if mgo.IsDup(err) {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-307, fileMoveRequest.BaseRequest.Tag, nil))
return
}
managers.LogError("Error renaming file", err)
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-303, fileMoveRequest.BaseRequest.Tag, nil))
return
}
managers.SendWebSocketMessage(wsConn, baseModels.NewSuccessResponse(fileMoveRequest.BaseRequest.Tag, nil))
managers.NotifyProjectClients(file.Project, fileMoveRequest.GetNotification(), wsConn)
}
func Subscribe(wsConn *websocket.Conn, subscriptionRequest projectRequests.ProjectSubscribeRequest) {
projectId := subscriptionRequest.BaseRequest.ResId
proj, err := GetProjectById(projectId)
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-200, subscriptionRequest.BaseRequest.Tag, nil))
return
}
hasPermission := false
for key, value := range proj.Permissions {
if (key == subscriptionRequest.BaseRequest.Username || key == "*") && value >= 1 {
if !managers.WebSocketSubscribeProject(wsConn, subscriptionRequest.BaseRequest.Username, projectId) {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-206, subscriptionRequest.BaseRequest.Tag, nil))
return
}
hasPermission = true
}
}
if !hasPermission {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-208, subscriptionRequest.BaseRequest.Tag, nil))
return
}
managers.NotifyProjectClients(projectId, subscriptionRequest.GetNotification(), wsConn)
managers.SendWebSocketMessage(wsConn, baseModels.NewSuccessResponse(subscriptionRequest.BaseRequest.Tag, nil))
}
// Rename project (?)
func RenameProject(wsConn *websocket.Conn, projectRenameRequest projectRequests.ProjectRenameRequest) {
// Get new DB connection
session, collection := managers.GetMGoCollection("Projects")
defer session.Close()
// Rename the project
err := collection.Update(bson.M{"_id": projectRenameRequest.BaseRequest.ResId}, bson.M{"$set": bson.M{"name": projectRenameRequest.NewName}})
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-202, projectRenameRequest.BaseRequest.Tag, nil))
return
}
managers.SendWebSocketMessage(wsConn, baseModels.NewSuccessResponse(projectRenameRequest.BaseRequest.Tag, nil))
managers.NotifyProjectClients(projectRenameRequest.BaseRequest.ResId, projectRenameRequest.GetNotification(), wsConn)
}
// Grant permission <Level> to <User>
// - Check if user exists
// - Grants permission level to user, overwriting if necessary.
func GrantProjectPermissions(wsConn *websocket.Conn, projectGrantPermissionsRequest projectRequests.ProjectGrantPermissionsRequest) {
project, err := GetProjectById(projectGrantPermissionsRequest.BaseRequest.ResId)
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-200, projectGrantPermissionsRequest.BaseRequest.Tag, nil))
return
}
if !CheckUserHasPermissions(project, projectGrantPermissionsRequest.BaseRequest.Username, 5) {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-207, projectGrantPermissionsRequest.BaseRequest.Tag, nil))
return
}
// Make sure that there is still an owner of the project.
owner := ""
for key, value := range project.Permissions {
if value == 10 && key != projectGrantPermissionsRequest.GrantUsername {
owner = key
}
}
if owner == "" {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-205, projectGrantPermissionsRequest.BaseRequest.Tag, nil))
return
}
project.Permissions[projectGrantPermissionsRequest.GrantUsername] = projectGrantPermissionsRequest.PermissionLevel
// Get new DB connection
session, collection := managers.GetMGoCollection("Projects")
defer session.Close()
// Update permissions
err = collection.Update(bson.M{"_id": projectGrantPermissionsRequest.BaseRequest.ResId}, bson.M{"$set": bson.M{"permissions": project.Permissions}})
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-202, projectGrantPermissionsRequest.BaseRequest.Tag, nil))
return
}
managers.SendWebSocketMessage(wsConn, baseModels.NewSuccessResponse(projectGrantPermissionsRequest.BaseRequest.Tag, nil))
managers.NotifyProjectClients(projectGrantPermissionsRequest.BaseRequest.ResId, projectGrantPermissionsRequest.GetNotification(), wsConn)
}
func DeleteFile(wsConn *websocket.Conn, fileDeleteRequest fileRequests.FileDeleteRequest) {
session, collection := managers.GetMGoCollection("Files")
defer session.Close()
// Check that file exists
file, err := GetFileById(fileDeleteRequest.BaseRequest.ResId)
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-300, fileDeleteRequest.BaseRequest.Tag, nil))
return
}
err = os.Remove(file.GetPath())
err = collection.Remove(bson.M{"_id": fileDeleteRequest.BaseRequest.ResId})
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-304, fileDeleteRequest.BaseRequest.Tag, nil))
return
}
managers.SendWebSocketMessage(wsConn, baseModels.NewSuccessResponse(fileDeleteRequest.BaseRequest.Tag, nil))
managers.NotifyProjectClients(file.Project, fileDeleteRequest.GetNotification(), wsConn)
}
func CreateFile(wsConn *websocket.Conn, fileCreateRequest fileRequests.FileCreateRequest) {
file := new(File)
file.Id = managers.NewObjectIdString()
file.Name = fileCreateRequest.Name
file.RelativePath = fileCreateRequest.RelativePath
file.Version = 0
file.Project = fileCreateRequest.ProjectId
session, collection := managers.GetMGoCollection("Files")
defer session.Close()
// Create indexes
index := mgo.Index{
Key: []string{"name", "relative_path"},
Unique: true,
DropDups: true,
Background: true,
Sparse: true,
}
err := collection.EnsureIndex(index)
if err != nil {
managers.LogError("Failed to ensure file name/path index", err)
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-301, fileCreateRequest.BaseRequest.Tag, nil))
return
}
// Insert file record
err = collection.Insert(file)
if err != nil {
if mgo.IsDup(err) {
managers.LogError("Error creating file record", err)
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-305, fileCreateRequest.BaseRequest.Tag, nil))
return
}
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-301, fileCreateRequest.BaseRequest.Tag, nil))
return
}
// Write file to disk
fileCreateRequest.RelativePath = filepath.Clean(fileCreateRequest.RelativePath)
if fileCreateRequest.RelativePath[0:2] == ".." || filepath.IsAbs(fileCreateRequest.RelativePath) {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-308, fileCreateRequest.BaseRequest.Tag, nil))
return
}
err = os.MkdirAll("files/"+fileCreateRequest.ProjectId+"/"+fileCreateRequest.RelativePath, os.ModeExclusive)
if err != nil {
managers.LogError("Failed to create file directory", err)
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-301, fileCreateRequest.BaseRequest.Tag, nil))
return
}
err = ioutil.WriteFile(file.GetPath(), fileCreateRequest.FileBytes, os.ModeExclusive)
if err != nil {
managers.LogError("Failed to write file", err)
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-301, fileCreateRequest.BaseRequest.Tag, nil))
return
}
managers.SendWebSocketMessage(wsConn, baseModels.NewSuccessResponse(fileCreateRequest.BaseRequest.Tag, map[string]interface{}{"FileId": file.Id}))
managers.NotifyProjectClients(file.Project, fileCreateRequest.GetNotification(file.Id), wsConn)
}
func InsertChange(wsConn *websocket.Conn, fileChangeRequest fileRequests.FileChangeRequest) {
// Check that file exists
file, err := GetFileById(fileChangeRequest.BaseRequest.ResId)
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-300, fileChangeRequest.BaseRequest.Tag, nil))
return
}
// Check that user is on latest version, then increment. Otherwise, throw error
if fileChangeRequest.FileVersion < file.Version {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-401, fileChangeRequest.BaseRequest.Tag, nil))
return
}
fileChangeRequest.FileVersion++
fileChange := new(FileChange)
fileChange.Id = managers.NewObjectIdString()
fileChange.Changes = fileChangeRequest.Changes
fileChange.FileId = fileChangeRequest.BaseRequest.ResId
fileChange.Version = fileChangeRequest.FileVersion
fileChange.Username = fileChangeRequest.BaseRequest.Username
fileChange.Date = time.Now().UTC()
changesSession, changesCollection := managers.GetMGoCollection("Changes")
defer changesSession.Close()
index := mgo.Index{
Key: []string{"file", "version"},
Unique: true,
DropDups: true,
Background: true,
Sparse: true,
}
err = changesCollection.EnsureIndex(index)
if err != nil {
managers.LogError("Failed to ensure changes index", err)
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-400, fileChangeRequest.BaseRequest.Tag, nil))
}
err = changesCollection.Insert(fileChange)
if err != nil {
if mgo.IsDup(err) {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-401, fileChangeRequest.BaseRequest.Tag, nil))
return
}
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-400, fileChangeRequest.BaseRequest.Tag, nil))
return
}
filesSession, filesCollection := managers.GetMGoCollection("Files")
defer filesSession.Close()
err = filesCollection.Update(bson.M{"_id": fileChangeRequest.BaseRequest.ResId}, bson.M{"$set": bson.M{"version": fileChangeRequest.FileVersion}})
if err != nil {
managers.SendWebSocketMessage(wsConn, baseModels.NewFailResponse(-400, fileChangeRequest.BaseRequest.Tag, nil))
return
}
// TODO: Change when fileVersion is changed to atomic int
if fileChange.Version%100 == 0 {
scrunching.ScrunchDB(fileChange.FileId)
}
managers.SendWebSocketMessage(wsConn, baseModels.NewSuccessResponse(fileChangeRequest.BaseRequest.Tag, map[string]interface{}{"FileVersion": fileChangeRequest.FileVersion}))
managers.NotifyProjectClients(file.Project, fileChangeRequest.GetNotification(fileChangeRequest.FileVersion), wsConn)
}