OrgGo
/
Tasks
mirror of https://github.com/thewhitetulip/Tasks.git
1
0
Fork 1
Code Issues Projects Releases Wiki Activity

added comments

This commit is contained in:
thewhitetulip 2015-11-21 12:09:15 +05:30
parent b7d14fbcbb
commit f79e24bc0d
1 changed files with 31 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
db/db.go
View File

@ -3,7 +3,7 @@ package db
import ( import (
"database/sql" "database/sql"
"fmt" "fmt"
_ "github.com/mattn/go-sqlite3" _ "github.com/mattn/go-sqlite3" //we want to use sqlite natively
"github.com/thewhitetulip/Tasks/types" "github.com/thewhitetulip/Tasks/types"
"strings" "strings"
"time" "time"
@ -19,13 +19,16 @@ func init() {
} }
} }
//Close function closes this database connection
func Close() { func Close() {
database.Close() database.Close()
} }
//GetTasks retrieves all the tasks depending on the
//status pending or trashed or completed
func GetTasks(status string) []types.Task { func GetTasks(status string) []types.Task {
var task []types.Task var task []types.Task
var TaskId int var TaskID int
var TaskTitle string var TaskTitle string
var TaskContent string var TaskContent string
var TaskCreated time.Time var TaskCreated time.Time
@ -44,21 +47,23 @@ func GetTasks(status string) []types.Task {
} }
defer rows.Close() defer rows.Close()
for rows.Next() { for rows.Next() {
err := rows.Scan(&TaskId, &TaskTitle, &TaskContent, &TaskCreated) err := rows.Scan(&TaskID, &TaskTitle, &TaskContent, &TaskCreated)
TaskContent = strings.Replace(TaskContent, "\n", "<br>", -1)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
TaskCreated = TaskCreated.Local() TaskCreated = TaskCreated.Local()
a := types.Task{Id: TaskId, Title: TaskTitle, Content: TaskContent, Created: TaskCreated.Format(time.UnixDate)[0:20]} a := types.Task{Id: TaskID, Title: TaskTitle, Content: TaskContent, Created: TaskCreated.Format(time.UnixDate)[0:20]}
task = append(task, a) task = append(task, a)
} }
return task return task
} }
//GetTaskById function gets the tasks from the ID passed to the function
func GetTaskById(id int) types.Task { func GetTaskById(id int) types.Task {
var task types.Task var task types.Task
var TaskId int var TaskID int
var TaskTitle string var TaskTitle string
var TaskContent string var TaskContent string
getTasksql := "select id, title, content from task where id=?" getTasksql := "select id, title, content from task where id=?"
@ -69,17 +74,18 @@ func GetTaskById(id int) types.Task {
} }
defer rows.Close() defer rows.Close()
if rows.Next() { if rows.Next() {
err := rows.Scan(&TaskId, &TaskTitle, &TaskContent) err := rows.Scan(&TaskID, &TaskTitle, &TaskContent)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
task = types.Task{Id: TaskId, Title: TaskTitle, Content: TaskContent} task = types.Task{Id: TaskID, Title: TaskTitle, Content: TaskContent}
} }
return task return task
} }
//TrashTask is used to delete the task
func TrashTask(id int) error { func TrashTask(id int) error {
trashSql, err := database.Prepare("update task set is_deleted='Y',last_modified_at=datetime() where id=?") trashSQL, err := database.Prepare("update task set is_deleted='Y',last_modified_at=datetime() where id=?")
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
@ -87,7 +93,7 @@ func TrashTask(id int) error {
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
_, err = tx.Stmt(trashSql).Exec(id) _, err = tx.Stmt(trashSQL).Exec(id)
if err != nil { if err != nil {
fmt.Println("doing rollback") fmt.Println("doing rollback")
tx.Rollback() tx.Rollback()
@ -97,6 +103,7 @@ func TrashTask(id int) error {
return err return err
} }
//CompleteTask is used to mark tasks as complete
func CompleteTask(id int) error { func CompleteTask(id int) error {
stmt, err := database.Prepare("update task set is_deleted='Y', finish_date=datetime(),last_modified_at=datetime() where id=?") stmt, err := database.Prepare("update task set is_deleted='Y', finish_date=datetime(),last_modified_at=datetime() where id=?")
if err != nil { if err != nil {
@ -116,6 +123,7 @@ func CompleteTask(id int) error {
return err return err
} }
//DeleteAll is used to empty the trash
func DeleteAll() error { func DeleteAll() error {
stmt, err := database.Prepare("delete from task where is_deleted='Y'") stmt, err := database.Prepare("delete from task where is_deleted='Y'")
if err != nil { if err != nil {
@ -135,8 +143,9 @@ func DeleteAll() error {
return err return err
} }
//RestoreTask is used to restore tasks from the Trash
func RestoreTask(id int) error { func RestoreTask(id int) error {
restoreSql, err := database.Prepare("update task set is_deleted='N',last_modified_at=datetime() where id=?") restoreSQL, err := database.Prepare("update task set is_deleted='N',last_modified_at=datetime() where id=?")
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
@ -144,7 +153,7 @@ func RestoreTask(id int) error {
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
_, err = tx.Stmt(restoreSql).Exec(id) _, err = tx.Stmt(restoreSQL).Exec(id)
if err != nil { if err != nil {
fmt.Println("doing rollback") fmt.Println("doing rollback")
tx.Rollback() tx.Rollback()
@ -154,6 +163,7 @@ func RestoreTask(id int) error {
return err return err
} }
//DeleteTask is used to delete the task from the database
func DeleteTask(id int) error { func DeleteTask(id int) error {
deleteSQL, err := database.Prepare("delete from task where id = ?") deleteSQL, err := database.Prepare("delete from task where id = ?")
if err != nil { if err != nil {
@ -173,13 +183,14 @@ func DeleteTask(id int) error {
return err return err
} }
//AddTask is used to add the task in the database
func AddTask(title, content string) error { func AddTask(title, content string) error {
restoreSql, err := database.Prepare("insert into task(title, content, created_date, last_modified_at) values(?,?,datetime(), datetime())") restoreSQL, err := database.Prepare("insert into task(title, content, created_date, last_modified_at) values(?,?,datetime(), datetime())")
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
tx, err := database.Begin() tx, err := database.Begin()
_, err = tx.Stmt(restoreSql).Exec(title, content) _, err = tx.Stmt(restoreSQL).Exec(title, content)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
tx.Rollback() tx.Rollback()
@ -189,8 +200,9 @@ func AddTask(title, content string) error {
return err return err
} }
//UpdateTask is used to update the tasks in the database
func UpdateTask(id int, title string, content string) error { func UpdateTask(id int, title string, content string) error {
Sql, err := database.Prepare("update task set title=?, content=? where id=?") SQL, err := database.Prepare("update task set title=?, content=? where id=?")
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
@ -199,7 +211,7 @@ func UpdateTask(id int, title string, content string) error {
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
_, err = tx.Stmt(Sql).Exec(title, content, id) _, err = tx.Stmt(SQL).Exec(title, content, id)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
tx.Rollback() tx.Rollback()
@ -209,10 +221,11 @@ func UpdateTask(id int, title string, content string) error {
return err return err
} }
//SearchTask is used to return the search results depending on the query
func SearchTask(query string) []types.Task { func SearchTask(query string) []types.Task {
stmt := "select id, title, content from task where title like '%" + query + "%' or content like '%" + query + "%'" stmt := "select id, title, content from task where title like '%" + query + "%' or content like '%" + query + "%'"
var task []types.Task var task []types.Task
var TaskId int var TaskID int
var TaskTitle string var TaskTitle string
var TaskContent string var TaskContent string
@ -221,13 +234,13 @@ func SearchTask(query string) []types.Task {
fmt.Println(err) fmt.Println(err)
} }
for rows.Next() { for rows.Next() {
err := rows.Scan(&TaskId, &TaskTitle, &TaskContent) err := rows.Scan(&TaskID, &TaskTitle, &TaskContent)
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
} }
TaskTitle = strings.Replace(TaskTitle, query, "<span class='highlight'>"+query+"</span>", -1) TaskTitle = strings.Replace(TaskTitle, query, "<span class='highlight'>"+query+"</span>", -1)
TaskContent = strings.Replace(TaskContent, query, "<span class='highlight'>"+query+"</span>", -1) TaskContent = strings.Replace(TaskContent, query, "<span class='highlight'>"+query+"</span>", -1)
a := types.Task{Id: TaskId, Title: TaskTitle, Content: TaskContent} a := types.Task{Id: TaskID, Title: TaskTitle, Content: TaskContent}
task = append(task, a) task = append(task, a)
} }
return task return task