basic storage package

This commit is contained in:
Nazar Kanaev 2020-06-21 21:58:51 +01:00 committed by nkanaev
parent 7a76027961
commit 6621f49770
7 changed files with 140 additions and 0 deletions

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
yarr

3
build.sh Executable file
View File

@ -0,0 +1,3 @@
#!/bin/sh
CGO_ENABLED=1 go build -tags sqlite_foreign_keys

5
go.mod Normal file
View File

@ -0,0 +1,5 @@
module github.com/nkanaev/yarr
go 1.14
require github.com/mattn/go-sqlite3 v1.14.0

11
go.sum Normal file
View File

@ -0,0 +1,11 @@
github.com/PuerkitoBio/goquery v1.5.1/go.mod h1:GsLWisAFVj4WgDibEWF4pvYnkVQBpKBKeU+7zCJoLcc=
github.com/andybalholm/cascadia v1.1.0/go.mod h1:GsXiBklL0woXo1j/WYWtSYYC4ouU9PqHO0sqidkEA4Y=
github.com/mattn/go-sqlite3 v1.14.0 h1:mLyGNKR8+Vv9CAU7PphKa2hkEqxxhn8i32J6FPj1/QA=
github.com/mattn/go-sqlite3 v1.14.0/go.mod h1:JIl7NbARA7phWnGvh0LKTyg7S9BA+6gx71ShQilpsus=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/net v0.0.0-20180218175443-cbe0f9307d01/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
golang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
golang.org/x/net v0.0.0-20200324143707-d3edc9973b7e/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=

18
main.go Normal file
View File

@ -0,0 +1,18 @@
package main
import (
"github.com/nkanaev/yarr/storage"
"log"
)
func main() {
storage, err := storage.New()
if err != nil {
log.Fatal(err)
}
/*
folder := storage.CreateFolder("foo")
storage.RenameFolder(folder.Id, "bar")
storage.ToggleFolderExpanded(folder.Id, false)
*/
}

42
storage/folder.go Normal file
View File

@ -0,0 +1,42 @@
package storage
import "fmt"
type Folder struct {
Id int64
Title string
IsExpanded bool
}
func (s *Storage) CreateFolder(title string) *Folder {
expanded := true
result, err := s.db.Exec(`
insert into folders (title, is_expanded) values (?, ?)`,
title, expanded,
)
if err != nil {
fmt.Println(err)
return nil
}
id, idErr := result.LastInsertId()
if idErr != nil {
return nil
}
return &Folder{Id: id, Title: title, IsExpanded: expanded}
}
func (s *Storage) DeleteFolder(folderId int64) bool {
_, err1 := s.db.Exec(`update feeds set folder_id = null where folder_id = ?`, folderId)
_, err2 := s.db.Exec(`delete from folders where id = ?`, folderId)
return err1 == nil && err2 == nil
}
func (s *Storage) RenameFolder(folderId int64, newTitle string) bool {
_, err := s.db.Exec(`update folders set title = ? where id = ?`, newTitle, folderId)
return err == nil
}
func (s *Storage) ToggleFolderExpanded(folderId int64, isExpanded bool) bool {
_, err := s.db.Exec(`update folders set is_expanded = ? where id = ?`, isExpanded, folderId)
return err == nil
}

60
storage/storage.go Normal file
View File

@ -0,0 +1,60 @@
package storage
import (
"database/sql"
_ "github.com/mattn/go-sqlite3"
)
var initQuery string = `
create table if not exists folders (
id integer primary key autoincrement,
title text not null,
is_expanded boolean
);
create table if not exists feeds (
id integer primary key autoincrement,
folder_id references folders(id),
title text not null,
description text,
link text,
feed_link text,
icon text
);
create index if not exists idx_feed_folder_id on feeds(folder_id);
create table if not exists items (
id string primary key,
feed_id references feeds(id),
title text,
link text,
description text,
content text,
author text,
date integer,
date_updated integer,
status integer,
image text
);
create index if not exists idx_item_feed_id on items(feed_id);
create index if not exists idx_item_status on items(status);
`
type Storage struct {
db *sql.DB
}
func New() (*Storage, error) {
path := "./storage.db"
db, err := sql.Open("sqlite3", path)
if err != nil {
return nil, err
}
_, err = db.Exec(initQuery)
if err != nil {
return nil, err
}
return &Storage{db: db}, nil
}