Fix error occurring when attaching a file to a Task

Extend the SQL storing the information to a file by username and creation date.
This commit is contained in:
Christian Huff 2016-05-15 22:23:24 +01:00
parent 3c5e5460e8
commit 72b36ef3e2
2 changed files with 8 additions and 4 deletions

View File

@ -11,8 +11,12 @@ import (
// AddFile is used to add the md5 of a file name which is uploaded to our application
// this will enable us to randomize the URL without worrying about the file names
func AddFile(fileName, token string) error {
err := taskQuery("insert into files values(?,?)", fileName, token)
func AddFile(fileName, token, username string) error {
userID, err := GetUserID(username)
if err != nil {
return err
}
err = taskQuery("insert into files values(?,?,?,datetime())", fileName, token, userID)
return err
}

View File

@ -72,6 +72,7 @@ func AddTaskFunc(w http.ResponseWriter, r *http.Request) {
cookie, _ := r.Cookie("csrftoken")
if formToken == cookie.Value {
username := sessions.GetCurrentUserName(r)
if handler != nil {
// this will be executed whenever a file is uploaded
r.ParseMultipartForm(32 << 20) //defined maximum size of file
@ -95,13 +96,12 @@ func AddTaskFunc(w http.ResponseWriter, r *http.Request) {
}
content = content + filelink
fileTruth := db.AddFile(handler.Filename, token)
fileTruth := db.AddFile(handler.Filename, token, username)
if fileTruth != nil {
message = "Error adding filename in db"
log.Println("error adding task to db")
}
}
username := sessions.GetCurrentUserName(r)
taskTruth := db.AddTask(title, content, category, taskPriority, username)
if taskTruth != nil {