Tasks/db/db.go

279 lines
7.0 KiB
Go
Raw Normal View History

2015-11-13 14:34:42 +05:30
package db
import (
"database/sql"
2015-11-21 12:09:15 +05:30
_ "github.com/mattn/go-sqlite3" //we want to use sqlite natively
2015-11-16 10:55:43 +09:00
"github.com/thewhitetulip/Tasks/types"
2016-01-28 21:44:40 +05:30
md "github.com/shurcooL/github_flavored_markdown"
2016-01-10 19:27:41 +05:30
"log"
2015-11-13 14:34:42 +05:30
"strings"
"time"
2015-11-13 14:34:42 +05:30
)
var database *sql.DB
var err error
func init() {
database, err = sql.Open("sqlite3", "./tasks.db")
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
}
2015-11-21 12:09:15 +05:30
//Close function closes this database connection
2015-11-13 14:34:42 +05:30
func Close() {
database.Close()
}
2015-11-21 12:09:15 +05:30
//GetTasks retrieves all the tasks depending on the
//status pending or trashed or completed
2015-11-21 18:50:51 +05:30
func GetTasks(status string) types.Context {
2015-11-13 14:34:42 +05:30
var task []types.Task
2015-11-21 18:50:51 +05:30
var context types.Context
2015-11-21 12:09:15 +05:30
var TaskID int
2015-11-13 14:34:42 +05:30
var TaskTitle string
var TaskContent string
var TaskCreated time.Time
2016-01-23 19:17:34 +05:30
var TaskPriority string
2015-11-13 14:34:42 +05:30
var getTasksql string
2015-11-21 18:50:51 +05:30
2016-01-23 19:17:34 +05:30
basicSQL := "select id, title, content, created_date, priority from task "
2015-11-14 16:26:53 +05:30
if status == "pending" {
2016-01-23 19:17:34 +05:30
getTasksql = basicSQL + " where finish_date is null and is_deleted='N' order by priority desc, created_date asc"
2015-11-21 18:50:51 +05:30
} else if status == "deleted" {
2016-01-23 19:17:34 +05:30
getTasksql = basicSQL + " where is_deleted='Y' order by priority desc, created_date asc"
2015-11-21 18:50:51 +05:30
} else if status == "completed" {
2016-01-23 19:17:34 +05:30
getTasksql = basicSQL + " where finish_date is not null order by priority desc, created_date asc"
2015-11-13 14:34:42 +05:30
}
rows, err := database.Query(getTasksql)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
defer rows.Close()
2015-11-13 18:41:30 +05:30
for rows.Next() {
2016-01-23 19:17:34 +05:30
err := rows.Scan(&TaskID, &TaskTitle, &TaskContent, &TaskCreated, &TaskPriority)
2016-01-28 21:44:40 +05:30
TaskContent = string(md.Markdown([]byte(TaskContent)))
// TaskContent = strings.Replace(TaskContent, "\n", "<br>", -1)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
TaskCreated = TaskCreated.Local()
2016-01-23 19:17:34 +05:30
a := types.Task{Id: TaskID, Title: TaskTitle, Content: TaskContent, Created: TaskCreated.Format(time.UnixDate)[0:20], Priority: TaskPriority}
2015-11-13 14:34:42 +05:30
task = append(task, a)
}
2015-11-21 18:50:51 +05:30
context = types.Context{Tasks: task, Navigation: status}
return context
2015-11-13 14:34:42 +05:30
}
2015-11-22 09:21:29 +05:30
//GetTaskByID function gets the tasks from the ID passed to the function, used to populate EditTask
2015-11-22 08:29:39 +05:30
func GetTaskByID(id int) types.Context {
var tasks []types.Task
2015-11-13 14:34:42 +05:30
var task types.Task
2016-01-23 19:17:34 +05:30
getTasksql := "select id, title, content, priority from task where id=?"
2015-11-13 14:34:42 +05:30
rows, err := database.Query(getTasksql, id)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
defer rows.Close()
2015-11-13 18:41:30 +05:30
if rows.Next() {
2016-01-23 19:17:34 +05:30
err := rows.Scan(&task.Id, &task.Title, &task.Content, &task.Priority)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2016-01-23 19:17:34 +05:30
//send email to respective people
2015-11-13 14:34:42 +05:30
}
}
2015-11-22 08:29:39 +05:30
tasks = append(tasks, task)
context := types.Context{Tasks: tasks, Navigation: "edit"}
return context
2015-11-13 14:34:42 +05:30
}
2015-11-21 12:09:15 +05:30
//TrashTask is used to delete the task
2015-11-14 16:26:53 +05:30
func TrashTask(id int) error {
2015-11-21 12:09:15 +05:30
trashSQL, err := database.Prepare("update task set is_deleted='Y',last_modified_at=datetime() where id=?")
2015-11-14 16:26:53 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-14 16:26:53 +05:30
}
tx, err := database.Begin()
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-14 16:26:53 +05:30
}
2015-11-21 12:09:15 +05:30
_, err = tx.Stmt(trashSQL).Exec(id)
2015-11-14 16:26:53 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println("doing rollback")
2015-11-14 16:26:53 +05:30
tx.Rollback()
} else {
tx.Commit()
}
return err
}
2015-11-21 12:09:15 +05:30
//CompleteTask is used to mark tasks as complete
2015-11-14 16:26:53 +05:30
func CompleteTask(id int) error {
2015-11-13 14:34:42 +05:30
stmt, err := database.Prepare("update task set is_deleted='Y', finish_date=datetime(),last_modified_at=datetime() where id=?")
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
tx, err := database.Begin()
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
_, err = tx.Stmt(stmt).Exec(id)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
tx.Rollback()
} else {
tx.Commit()
}
return err
}
2015-11-21 12:09:15 +05:30
//DeleteAll is used to empty the trash
2015-11-13 14:34:42 +05:30
func DeleteAll() error {
stmt, err := database.Prepare("delete from task where is_deleted='Y'")
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
tx, err := database.Begin()
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
_, err = tx.Stmt(stmt).Exec()
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println("doing rollback")
2015-11-13 14:34:42 +05:30
tx.Rollback()
} else {
tx.Commit()
}
return err
}
2015-11-21 12:09:15 +05:30
//RestoreTask is used to restore tasks from the Trash
2015-11-13 14:34:42 +05:30
func RestoreTask(id int) error {
2015-11-21 12:09:15 +05:30
restoreSQL, err := database.Prepare("update task set is_deleted='N',last_modified_at=datetime() where id=?")
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
tx, err := database.Begin()
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
2015-11-21 12:09:15 +05:30
_, err = tx.Stmt(restoreSQL).Exec(id)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println("doing rollback")
2015-11-13 14:34:42 +05:30
tx.Rollback()
} else {
tx.Commit()
}
return err
}
2016-01-09 10:48:11 +05:30
//RestoreTask is used to restore tasks from the Trash
2016-01-09 10:33:35 +05:30
func RestoreTaskFromComplete(id int) error {
2016-01-09 10:48:11 +05:30
restoreSQL, err := database.Prepare("update task set finish_date=null,last_modified_at=datetime() where id=?")
2016-01-09 10:33:35 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2016-01-09 10:33:35 +05:30
}
tx, err := database.Begin()
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2016-01-09 10:33:35 +05:30
}
_, err = tx.Stmt(restoreSQL).Exec(id)
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println("doing rollback")
2016-01-09 10:33:35 +05:30
tx.Rollback()
} else {
tx.Commit()
}
return err
}
2015-11-21 12:09:15 +05:30
//DeleteTask is used to delete the task from the database
2015-11-13 14:34:42 +05:30
func DeleteTask(id int) error {
deleteSQL, err := database.Prepare("delete from task where id = ?")
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
tx, err := database.Begin()
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
_, err = tx.Stmt(deleteSQL).Exec(id)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
tx.Rollback()
} else {
tx.Commit()
}
return err
}
2015-11-21 12:09:15 +05:30
//AddTask is used to add the task in the database
2016-01-23 19:17:34 +05:30
func AddTask(title, content string, taskPriority int) error {
2016-01-28 20:43:52 +05:30
restoreSQL, err := database.Prepare("insert into task(title, content, priority, created_date, last_modified_at) values(?,?,?,datetime(), datetime())")
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
tx, err := database.Begin()
2016-01-23 19:17:34 +05:30
_, err = tx.Stmt(restoreSQL).Exec(title, content, taskPriority)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
tx.Rollback()
} else {
tx.Commit()
}
return err
}
2015-11-21 12:09:15 +05:30
//UpdateTask is used to update the tasks in the database
2015-11-13 18:41:30 +05:30
func UpdateTask(id int, title string, content string) error {
2015-11-21 12:09:15 +05:30
SQL, err := database.Prepare("update task set title=?, content=? where id=?")
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
tx, err := database.Begin()
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
2015-11-21 12:09:15 +05:30
_, err = tx.Stmt(SQL).Exec(title, content, id)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
tx.Rollback()
} else {
2016-01-09 10:48:11 +05:30
log.Println(tx.Commit())
2015-11-13 14:34:42 +05:30
}
return err
}
2015-11-21 12:09:15 +05:30
//SearchTask is used to return the search results depending on the query
2015-11-21 19:42:20 +05:30
func SearchTask(query string) types.Context {
2015-11-21 16:33:34 +05:30
stmt := "select id, title, content, created_date from task where title like '%" + query + "%' or content like '%" + query + "%'"
2015-11-13 14:34:42 +05:30
var task []types.Task
2015-11-21 12:09:15 +05:30
var TaskID int
2015-11-13 14:34:42 +05:30
var TaskTitle string
var TaskContent string
2015-11-21 16:33:34 +05:30
var TaskCreated time.Time
2015-11-21 19:42:20 +05:30
var context types.Context
2015-11-13 14:34:42 +05:30
rows, err := database.Query(stmt, query, query)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
2015-11-13 18:41:30 +05:30
for rows.Next() {
2015-11-21 16:33:34 +05:30
err := rows.Scan(&TaskID, &TaskTitle, &TaskContent, &TaskCreated)
2015-11-13 18:41:30 +05:30
if err != nil {
2016-01-09 10:48:11 +05:30
log.Println(err)
2015-11-13 14:34:42 +05:30
}
TaskTitle = strings.Replace(TaskTitle, query, "<span class='highlight'>"+query+"</span>", -1)
TaskContent = strings.Replace(TaskContent, query, "<span class='highlight'>"+query+"</span>", -1)
2016-01-28 21:44:40 +05:30
TaskContent = string(md.Markdown([]byte(TaskContent)))
2015-11-21 16:33:34 +05:30
a := types.Task{Id: TaskID, Title: TaskTitle, Content: TaskContent, Created: TaskCreated.Format(time.UnixDate)[0:20]}
2015-11-13 14:34:42 +05:30
task = append(task, a)
}
2015-11-21 19:42:20 +05:30
context = types.Context{Tasks: task, Search: query}
return context
2015-11-13 14:34:42 +05:30
}