bookmark.go

package main

import (
	"crypto/sha1"
	"fmt"
	"log"
	"time"
)

type Bookmark struct {
	Id            string `json:"id"`
	BookmarkUrl   string `json:"url"`
	BookmarkTitle string `json:"title"`
	BookmarkGroup string `json:"group"`
	IconURL       string `json:"icon"`
	IconEncoded   string `json:"icon_encoded"`
}

type Group struct {
	Id        string `json:"id"`
	GroupName string `json:"group_name"`
}

func (bookmark *Bookmark) CreateNewGroup(dbConnection *DBConnection, groupName string, userID string) bool {
	sha1Hash := sha1.New()
	sha1Hash.Write([]byte(time.Now().String() + groupName))
	sha1HashString := sha1Hash.Sum(nil)

	groupID := fmt.Sprintf("%x", sha1HashString)

	query := "INSERT INTO groups(id, user_id, group_name) VALUES('" + groupID + "','" + userID + "','" + groupName + "')"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) UpdateExistingGroup(dbConnection *DBConnection, groupID string, groupName string, userID string) bool {
	query := "UPDATE groups SET group_name='" + groupName + "' WHERE id='" + groupID + "' AND user_id='" + userID + "'"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) DeleteBookmarkGroupByID(dbConnection *DBConnection, groupID string, userID string) bool {
	query := "DELETE FROM groups WHERE id='" + groupID + "' AND user_id='" + userID + "'"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) ListAllGroups(dbConnection *DBConnection, userID string) []*Group {
	query := "SELECT id, group_name FROM groups WHERE user_id='" + userID + "' AND user_id='" + userID + "'"

	rows, err := dbConnection.db.Query(query)

	if err != nil {
		log.Fatal(err)
		return nil
	}

	groups := make([]*Group, 0)

	for rows.Next() {

		newGroup := new(Group)

		err := rows.Scan(
			&newGroup.Id,
			&newGroup.GroupName)

		if err != nil {
			log.Fatal(err)
			return nil
		}

		groups = append(groups, newGroup)

	}

	return groups
}

func (bookmark *Bookmark) CreateNewBookmark(dbConnection *DBConnection, userID string) bool {
	sha1Hash := sha1.New()
	sha1Hash.Write([]byte(time.Now().String() + bookmark.BookmarkTitle + " " + bookmark.BookmarkUrl + " " + bookmark.BookmarkGroup))
	sha1HashString := sha1Hash.Sum(nil)

	bookmarkID := fmt.Sprintf("%x", sha1HashString)

	query := "INSERT INTO bookmarks(id, user_id, bookmark_url, bookmark_title, bookmark_icon, bookmark_icon_base64, bookmark_group) VALUES('" +
		bookmarkID + "','" + userID + "','" + bookmark.BookmarkUrl + "','" + bookmark.BookmarkTitle + "','" + bookmark.IconURL + "','" + bookmark.IconEncoded + "','" + bookmark.BookmarkGroup + "')"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) UpdateExistingBookmark(dbConnection *DBConnection, bookmarkID string, bookmarkName string, userID string) bool {
	query := "UPDATE bookmarks SET bookmark_title='" + bookmarkName + "' WHERE id='" + bookmarkID + "' AND user_id='" + userID + "'"
	fmt.Println(query)
	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) DeleteBookmarkByID(dbConnection *DBConnection, bookmarkID string, userID string) bool {
	query := "DELETE FROM bookmarks WHERE id='" + bookmarkID + "' AND user_id='" + userID + "'"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) DeleteBookmarksInGroup(dbConnection *DBConnection, groupID string, userID string) bool {
	query := "DELETE FROM bookmarks WHERE bookmark_group='" + groupID + "' AND user_id='" + userID + "'"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) DeleteBookmarksAll(dbConnection *DBConnection) bool {
	query := "DELETE FROM bookmarks"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}

func (bookmark *Bookmark) ListAllBookmarks(dbConnection *DBConnection, userID string) []*Bookmark {
	query := "SELECT b.id, b.bookmark_url, b.bookmark_title, b.bookmark_icon, b.bookmark_icon_base64, g.group_name FROM bookmarks b INNER JOIN groups g ON b.bookmark_group=g.id AND b.user_id='" + userID + "'"

	rows, err := dbConnection.db.Query(query)

	if err != nil {
		log.Fatal(err)
		return nil
	}

	bookmarks := make([]*Bookmark, 0)

	for rows.Next() {

		newBookmark := new(Bookmark)

		err := rows.Scan(
			&newBookmark.Id,
			&newBookmark.BookmarkUrl,
			&newBookmark.BookmarkTitle,
			&newBookmark.IconURL,
			&newBookmark.IconEncoded,
			&newBookmark.BookmarkGroup)

		if err != nil {
			log.Fatal(err)
			return nil
		}

		bookmarks = append(bookmarks, newBookmark)

	}

	return bookmarks
}

func (bookmark *Bookmark) ListAllBookmarksInGroup(dbConnection *DBConnection, groupID string, userID string) []*Bookmark {
	query := "SELECT id, bookmark_url, bookmark_title, bookmark_icon, bookmark_icon_base64 FROM bookmarks WHERE bookmark_group='" + groupID + "' AND user_id='" + userID + "'"

	rows, err := dbConnection.db.Query(query)

	if err != nil {
		log.Fatal(err)
		return nil
	}

	bookmarks := make([]*Bookmark, 0)

	for rows.Next() {

		newBookmark := new(Bookmark)

		err := rows.Scan(
			&newBookmark.Id,
			&newBookmark.BookmarkUrl,
			&newBookmark.BookmarkTitle,
			&newBookmark.IconURL,
			&newBookmark.IconEncoded)

		if err != nil {
			log.Fatal(err)
			return nil
		}

		bookmarks = append(bookmarks, newBookmark)

	}

	return bookmarks
}

func (bookmark *Bookmark) ChangeGroup(dbConnection *DBConnection, group *Group) bool {
	query := "UPDATE bookmarks SET bookmark_group='" + group.Id + "' WHERE id='" + bookmark.Id + "'"

	_, err := dbConnection.db.Exec(query)

	if err != nil {
		log.Fatal(err)
		return false
	}
	return true
}