generated from alecodes/base-template
162 lines
3.8 KiB
Go
162 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
_ "modernc.org/sqlite"
|
|
|
|
synchronizator "git.alecodes.page/alecodes/synchronizator/pkg"
|
|
)
|
|
|
|
type ReadwiseDocument struct {
|
|
Id string `json:"id"`
|
|
Url string `json:"url"`
|
|
Title string `json:"title"`
|
|
Location string `json:"location"`
|
|
SourceUrl string `json:"source_url"`
|
|
Tags map[string]ReadwiseTag `json:"tags"`
|
|
}
|
|
|
|
type ReadwiseTag struct {
|
|
Name string `json:"name"`
|
|
Type string `json:"type"`
|
|
Created int `json:"created"`
|
|
}
|
|
|
|
type ReadwiseHighlight struct {
|
|
UserBookID int `json:"user_book_id"`
|
|
Title string `json:"title"`
|
|
Author string `json:"author"`
|
|
Source string `json:"source"`
|
|
UniqueURL string `json:"unique_url"`
|
|
BookTags []HighlightTag `json:"book_tags"`
|
|
Category string `json:"category"`
|
|
DocumentNote *string `json:"document_note"`
|
|
ReadwiseURL string `json:"readwise_url"`
|
|
SourceURL string `json:"source_url"`
|
|
Highlights []HighlightItem `json:"highlights"`
|
|
}
|
|
|
|
type HighlightTag struct {
|
|
Id int `json:"id"`
|
|
Name string `json:"name"`
|
|
}
|
|
|
|
type HighlightItem struct {
|
|
ID int `json:"id"`
|
|
Text string `json:"text"`
|
|
Location int `json:"location"`
|
|
LocationType string `json:"location_type"`
|
|
Note string `json:"note"`
|
|
Color string `json:"color"`
|
|
HighlightedAt string `json:"highlighted_at"`
|
|
CreatedAt string `json:"created_at"`
|
|
UpdatedAt string `json:"updated_at"`
|
|
ExternalID string `json:"external_id"`
|
|
EndLocation *int `json:"end_location"`
|
|
URL string `json:"url"`
|
|
BookID int `json:"book_id"`
|
|
Tags []HighlightTag `json:"tags"`
|
|
IsFavorite bool `json:"is_favorite"`
|
|
IsDiscard bool `json:"is_discard"`
|
|
ReadwiseURL string `json:"readwise_url"`
|
|
}
|
|
|
|
type ReadeckBookmark struct {
|
|
Id string `json:"id"`
|
|
Url string `json:"url"`
|
|
Title string `json:"title"`
|
|
Labels []string `json:"labels"`
|
|
IsArchived bool `json:"is_archived"`
|
|
}
|
|
|
|
func readwiseToReadeck(document *ReadwiseDocument) (*synchronizator.Node, error) {
|
|
bookmark := &ReadeckBookmark{
|
|
Url: document.SourceUrl,
|
|
Title: document.Title,
|
|
}
|
|
|
|
for _, tag := range document.Tags {
|
|
bookmark.Labels = append(bookmark.Labels, tag.Name)
|
|
}
|
|
|
|
metadata, err := json.Marshal(bookmark)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
node := synchronizator.NewNode(bookmark.Title, "BOOKMARK", metadata, nil)
|
|
|
|
return node, nil
|
|
}
|
|
|
|
func drop_data(conn *sql.DB) error {
|
|
sql := `
|
|
DELETE FROM nodes WHERE name = 'READECK';
|
|
`
|
|
_, err := conn.Exec(sql)
|
|
|
|
return err
|
|
}
|
|
|
|
func main() {
|
|
start := time.Now()
|
|
|
|
defer func() {
|
|
elapsed := time.Now().Sub(start)
|
|
fmt.Printf("\n\nExecution time took: %s", elapsed)
|
|
}()
|
|
|
|
connection, err := sql.Open("sqlite", "readwise.sql")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
|
|
return
|
|
}
|
|
|
|
defer connection.Close()
|
|
|
|
opts := synchronizator.DefaultOptions
|
|
// opts.Log_level = synchronizator.DEBUG
|
|
|
|
err = drop_data(connection)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
|
|
return
|
|
}
|
|
|
|
sync, err := synchronizator.New(connection, opts)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
|
|
return
|
|
}
|
|
|
|
readeck, err := sync.NewPlatform("READECK", nil, nil)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
collection, err := readeck.GetDefaultCollection()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
readwiseDocuments, err := sync.GetCollection(3)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
err = synchronizator.ReconciliateCollections(readwiseDocuments, collection, readwiseToReadeck)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
}
|