-
Notifications
You must be signed in to change notification settings - Fork 0
/
skim2md.go
169 lines (143 loc) · 3.84 KB
/
skim2md.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
package skimmer
import (
"io"
"fmt"
"database/sql"
"strings"
"time"
)
// Skim2Md supports the skim2md cli.
type Skim2Md struct {
// AppName holds the name of the application
AppName string `json:"app_name,omitempty"`
// DbName holds the path to the SQLite3 database
DBName string `json:"db_name,omitempty"`
// Title if this is set the title will be included
// when generating the markdown of saved items
Title string `json:"title,omitempty"`
// FrontMatter, if true insert Frontmatter block in Markdown output
FrontMatter bool `json:"frontmatter,omitempty"`
// PocketButton, if true insert a "save to pocket" button for each RSS item output
PocketButton bool
out io.Writer
eout io.Writer
}
// NewSkim2Md initialized a new Skim2Md struct
func NewSkim2Md(appName string) (*Skim2Md, error) {
app := new(Skim2Md)
app.AppName = appName
return app, nil
}
func (app *Skim2Md) DisplayItem(link string, title string, description string, updated string, published string, label string, tags string) error {
// Then see about formatting things.
pressTime := published
if len(pressTime) > 10 {
pressTime = pressTime[0:10]
}
if updated != "" {
if len(updated) > 10 {
updated = updated[0:10]
}
pressTime += ", updated: " + updated
}
if strings.HasPrefix(label, `"~`) {
label = strings.Trim(label, `"~`)
}
if title == "" {
title = fmt.Sprintf("**@%s** (date: %s, from: %s)", label, pressTime, label)
} else {
title = fmt.Sprintf("## %s\n\ndate: %s, from: %s", title, pressTime, label)
}
if app.PocketButton {
fmt.Fprintf(app.out, `---
%s
%s
<span class="feed-item-link">
<a href="%s">%s</a> <a href="https://getpocket.com/save" class="pocket-btn" data-lang="en" data-save-url="%s">Save to Pocket</a>
</span>
`, title, description, link, link, link)
} else {
fmt.Fprintf(app.out, `---
%s
%s
<%s>
`, title, description, link)
}
return nil
}
// Write, display the contents from database
func (app *Skim2Md) Write(db *sql.DB) error {
timestamp := time.Now().Format("2006-01-02 15:04:05")
if app.FrontMatter {
fmt.Fprintf(app.out, "---\n")
if app.Title != "" {
fmt.Fprintf(app.out, "title: %s\n", app.Title)
}
fmt.Fprintf(app.out, "updated: %s\n", timestamp)
fmt.Fprintf(app.out, "---\n\n")
}
if app.Title != "" {
fmt.Fprintf(app.out, `# %s
(date: %s)
`, app.Title, timestamp)
} else {
fmt.Fprintf(app.out, `
(date: %s)
`, timestamp)
}
stmt := SQLDisplayItems
rows, err := db.Query(stmt, "saved")
if err != nil {
return err
}
defer rows.Close()
for rows.Next() {
var (
link string
title string
description string
updated string
published string
label string
tags string
)
if err := rows.Scan(&link, &title, &description, &updated, &published, &label, &tags); err != nil {
fmt.Fprintf(app.eout, "%s\n", err)
continue
}
if err := app.DisplayItem(link, title, description, updated, published, label, tags); err != nil {
return err
}
}
if err := rows.Err(); err != nil {
return err
}
if app.PocketButton {
fmt.Fprintf(app.out, `
<script type="text/javascript">!function(d,i){if(!d.getElementById(i)){var j=d.createElement("script");j.id=i;j.src="https://widgets.getpocket.com/v1/j/btn.js?v=1";var w=d.getElementById(i);d.body.appendChild(j);}}(document,"pocket-btn-js");</script>
`)
}
return nil
}
func (app *Skim2Md) Run(out io.Writer, eout io.Writer, args []string, frontMatter bool, pocketButton bool) error {
if len(args) < 1 {
return fmt.Errorf("missing skimmer database file")
}
if len(args) > 1 {
return fmt.Errorf("expected only one skimmer database file %+v", args)
}
app.FrontMatter = frontMatter
app.PocketButton = pocketButton
app.out = out
app.eout = eout
dsn := args[0]
db, err := sql.Open("sqlite", dsn)
if err != nil {
return err
}
defer db.Close()
if err := app.Write(db); err != nil {
return err
}
return nil
}