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
| | # Copyright (C) 2018-2020 all contributors <meta@public-inbox.org>
# License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
package PublicInbox::IMAPTracker;
use strict;
use parent qw(PublicInbox::Lock);
use DBI;
use DBD::SQLite;
use PublicInbox::Config;
sub create_tables ($) {
my ($dbh) = @_;
$dbh->do(<<'');
CREATE TABLE IF NOT EXISTS imap_last (
url VARCHAR PRIMARY KEY NOT NULL,
uid_validity INTEGER NOT NULL,
uid INTEGER NOT NULL,
UNIQUE (url)
)
}
sub dbh_new ($) {
my ($dbname) = @_;
my $dbh = DBI->connect("dbi:SQLite:dbname=$dbname", '', '', {
AutoCommit => 1,
RaiseError => 1,
PrintError => 0,
sqlite_use_immediate_transaction => 1,
});
$dbh->{sqlite_unicode} = 1;
$dbh->do('PRAGMA journal_mode = TRUNCATE');
create_tables($dbh);
$dbh;
}
sub get_last ($) {
my ($self) = @_;
my $sth = $self->{dbh}->prepare_cached(<<'', undef, 1);
SELECT uid_validity, uid FROM imap_last WHERE url = ?
$sth->execute($self->{url});
$sth->fetchrow_array;
}
sub update_last ($$$) {
my ($self, $validity, $last) = @_;
my $sth = $self->{dbh}->prepare_cached(<<'');
INSERT OR REPLACE INTO imap_last (url, uid_validity, uid)
VALUES (?, ?, ?)
$self->lock_acquire;
my $rv = $sth->execute($self->{url}, $validity, $last);
$self->lock_release;
$rv;
}
sub new {
my ($class, $url) = @_;
# original name for compatibility with old setups:
my $dbname = PublicInbox::Config->config_dir() . "/imap.sqlite3";
# use the new XDG-compliant name for new setups:
if (!-f $dbname) {
$dbname = ($ENV{XDG_DATA_HOME} //
(($ENV{HOME} // '/nonexistent').'/.local/share')) .
'/public-inbox/imap.sqlite3';
}
if (!-f $dbname) {
require File::Path;
require File::Basename;
File::Path::mkpath(File::Basename::dirname($dbname));
}
my $self = bless { lock_path => "$dbname.lock", url => $url }, $class;
$self->lock_acquire;
$self->{dbh} = dbh_new($dbname);
$self->lock_release;
$self;
}
1;
|