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
|
var mastodon = require('mastodon');
var pg = require('pg');
var query = `SELECT id, created_at
FROM public_toots
WHERE favourites_count > (
SELECT avg(favourites_count)
FROM public_toots
WHERE
favourites_count > 1
AND created_at > NOW() - INTERVAL '30 days'
)
AND created_at > NOW() - INTERVAL '5 days';`
var DB_USER = process.env.DB_USER || 'ambassador';
var DB_NAME = process.env.DB_NAME || 'mastodon_production';
var DB_PASSWORD = process.env.DB_PASSWORD || '';
var DB_HOST = process.env.DB_HOST || '/var/run/postgresql';
var AMBASSADOR_TOKEN = process.env.AMBASSADOR_TOKEN;
var INSTANCE_HOST = process.env.INSTANCE_HOST;
var config = {
user: process.env.DB_USER || 'ambassador',
database: process.env.DB_NAME || 'mastodon_production',
password: process.env.DB_PASSWORD || '',
host: process.env.DB_HOST || '/var/run/postgresql',
port: 5432, //env var: PGPORT
max: 2, // max number of clients in the pool
idleTimeoutMillis: 30000 // how long a client is allowed to remain idle before being closed
};
console.dir('STARTING AMBASSADOR');
console.log('\tDB_USER:', DB_USER);
console.log('\tDB_NAME:', DB_NAME);
console.log('\tDB_PASSWORD:', DB_PASSWORD.split('').map(function() { return "*" }).join(''));
console.log('\tDB_HOST:', DB_HOST);
console.log('\tAMBASSADOR_TOKEN:', AMBASSADOR_TOKEN);
console.log('\tINSTANCE_HOST:', INSTANCE_HOST);
var client = new pg.Client(config);
function cycle() {
client.connect(function (err) {
if (err) {
console.error('error connecting to client');
return console.dir(err);
}
client.query(query, [], function (err, result) {
if(err) {
console.error('error running query');
return console.dir(err);
}
client.end(function (err) {
if (err) {
console.error('error disconnecting from client');
console.dir(err);
}
});
boost(result.rows);
});
});
}
var M = new mastodon({
access_token: AMBASSADOR_TOKEN,
api_url: INSTANCE_HOST + '/api/v1'
});
var boosted = (function() {
const bucketSpan = 3600; // 1 hour buckets => up to 121 buckets over 5 days
const buckets = new Map();
function prune() {
// Bucket id for 5 days ago
const threshold = (Date.now() - 5 * 24 * 3600 * 1000) / bucketSpan;
for (var bucket of buckets.keys()) {
if (bucket < threshold) buckets.delete(bucket);
}
}
function bucket(row) {
return row.created_at.getTime() / bucketSpan;
}
function already(row) {
const b = bucket(row);
return buckets.has(b) && buckets.get(b).has(row.id);
}
function set(row) {
const b = bucket(row);
if (!buckets.has(b)) buckets.set(b, new Set());
buckets.get(b).add(row.id);
}
return { already: already, prune: prune, set: set };
})();
function boost(rows) {
rows.filter(function(x) { return !boosted.already(x); })
.forEach(function(row) {
M.post('/statuses/' + row.id + '/reblog', function(err, result) {
if (err) {
if (err.message === 'Validation failed: Reblog of status already exists') {
boosted.set(row);
return console.log('Warning: tried to boost #' + row.id + ' but it had already been boosted by this account. Adding to cache.');
}
return console.log(err);
}
boosted.set(row);
console.log('boosted status #' + row.id);
});
})
}
cycle();
// Prune the set of boosted toots every hour
setInterval(boosted.prune, 1000 * 3600);
setInterval(cycle, 1000 * 60 * 15);
|
