aboutsummaryrefslogtreecommitdiffstats
path: root/src/markam-convert.scm
blob: b18bd1f40cb8da7c188d99e9eeccd0101946c26c (plain)
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
;;; markam -- Store/retrieve/manage bookmarks
;; Copyright (C) 2012, 2013 Tom Willemsen <tom at ryuslash dot org>

;; This file is part of Markam

;; Markam is free software: you can redistribute it and/or modify it
;; under the terms of the GNU General Public License as published by
;; the Free Software Foundation, either version 3 of the License, or
;; (at your option) any later version.
;;
;; Markam is distributed in the hope that it will be useful, but
;; WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
;; General Public License for more details.
;;
;; You should have received a copy of the GNU General Public License
;; along with Markam. If not, see <http://www.gnu.org/licenses/>.

;;; Commentary:

;; Convert any old database file to one usable with this version of
;; markam.  If you're just starting to use markam there should be
;; no need to use this.

;;; Code:

(declare (uses paths
               common))
#+chicken-script
(begin
  (load "paths")
  (load "common"))

(require-extension sqlite3)
(require-library srfi-4)

(define (blob->seconds blob)
  ;; Convert BLOB to an integer representing the seconds since
  ;; 01-01-1970 (standard unix timstamp).  This has to be done because
  ;; in the previous version of markam I stored the timestamp as a
  ;; blob and CHICKEN doesn't seem to like blobs much.
  (if (u32vector? blob)
      (u32vector-ref (blob->u32vector blob) 0)
      0))

(define (convert-table db name converter #!optional select)
  ;; Convert NAME from DB using CONVERTER for each specific row.  The
  ;; optional SELECT should be an SQL query that will be used to fetch
  ;; the rows to convert.
  (let ((count (first-result db (string-append "SELECT COUNT(*) FROM " name)))
        (progress 0))
    (for-each-row (lambda args
                    (apply converter args)
                    (set! progress (+ progress 1))
                    (format #t "Converted ~s of ~s ~a rows\r~!" progress count name))
                  db (or select (string-append "SELECT * FROM " name)))
    (newline)))

(define (convert)
  ;; Convert an old database into a new one and then replace the old
  ;; with the new.
  (let ((old-db (open-database (data-file "linkwave.db" "linkwave")))
        (new-db (open-database (data-file "markam.db"))))
    (execute new-db "CREATE TABLE bookmark (url VARCHAR(255) UNIQUE, date INTEGER, name VARCHAR(255), description TEXT)")
    (execute new-db "CREATE TABLE tag (name VARCHAR(255) UNIQUE)")
    (execute new-db "CREATE TABLE bookmark_tag (bookmark_id INTEGER REFERENCES bookmark(rowid), tag_id INTEGER REFERENCES tag(rowid), PRIMARY KEY (bookmark_id, tag_id))")

    (convert-table old-db "bookmark"
                   (lambda (url dateblob name description)
                     (execute new-db "INSERT INTO bookmark VALUES (?, ?, ?, ?)"
                              url (blob->seconds dateblob) name description)))

    (convert-table old-db "tag"
                   (lambda (name)
                     (execute new-db "INSERT INTO tag VALUES (?)" name)))

    (convert-table old-db "bookmark_tag"
                   (lambda (url tag)
                     (execute new-db "INSERT INTO bookmark_tag VALUES (?, ?)"
                              (first-result new-db "SELECT rowid FROM bookmark WHERE url = ?" url)
                              (first-result new-db "SELECT rowid FROM tag WHERE name = ?" tag)))
                   "SELECT url, name FROM bookmark_tag JOIN tag ON (tag_id = tag.rowid)")

    (format #t "Database converted.~%")))

(define (display-help)
  ;; Display markam's help message.
  (format #t (string-append
              "Usage: markam-convert [options]...~%"
              "~%"
              "Possible options:~%"
              "~%"
              "--help, -h     Display this help and exit~%"
              "--version, -v  Output version information and exit~%")))

(define (main args)
  ;; Convert an old database file unless a conversion database already
  ;; exists.
  (if (null? args)
      (if (file-exists? (data-file "markam.db"))
          (format #t (string-append "Converted database already exists, or something may have gone "
                                    "wrong during the last run~%"))
          (convert))
      (do ((arg (car args) (and (not (null? args))
                                (car args))))
          ((or (null? arg) (not arg)))
        (cond
         ((or (string= arg "-v") (string= arg "--version"))
          (display-version)
          (exit 0))
         ((or (string= arg "-h") (string= arg "--help"))
          (display-help)
          (exit 0))
         (else
          (format #t "Unrecognized option: ~a~%" (car args))))
        (set! args (cdr args)))))

(main (command-line-arguments))