mirror of
https://github.com/elisspace/FxLifeSheet.git
synced 2026-08-29 15:44:00 +00:00
471 lines
19 KiB
JavaScript
471 lines
19 KiB
JavaScript
"use strict";
|
||
exports.__esModule = true;
|
||
var moment = require("moment");
|
||
var needle = require("needle");
|
||
var _a = require("telegraf"), Router = _a.Router, Markup = _a.Markup, Extra = _a.Extra;
|
||
var config = require("./classes/config.js");
|
||
var postgres = require("./classes/postgres.js");
|
||
var telegram = require("./classes/telegram.js");
|
||
var bot = telegram.bot;
|
||
var currentlyAskedQuestionObject = null;
|
||
var currentlyAskedQuestionMessageId = null;
|
||
var currentlyAskedQuestionQueue = [];
|
||
initBot();
|
||
function roundNumberExactly(number, decimals) {
|
||
return (Math.round(number * Math.pow(10, decimals)) / Math.pow(10, decimals)).toFixed(decimals);
|
||
}
|
||
function getButtonText(number) {
|
||
var emojiNumber = {
|
||
"0": "0️⃣",
|
||
"1": "1️⃣",
|
||
"2": "2️⃣",
|
||
"3": "3️⃣",
|
||
"4": "4️⃣",
|
||
"5": "5️⃣"
|
||
}[number];
|
||
if (currentlyAskedQuestionObject.buttons == null) {
|
||
currentlyAskedQuestionObject.buttons = {
|
||
"0": "Terrible",
|
||
"1": "Bad",
|
||
"2": "Okay",
|
||
"3": "Good",
|
||
"4": "Great",
|
||
"5": "Excellent"
|
||
};
|
||
}
|
||
return emojiNumber + " " + currentlyAskedQuestionObject.buttons[number];
|
||
}
|
||
function printGraph(key, ctx, numberOfRecentValuesToPrint, additionalValue, skipImage) {
|
||
postgres.client.query({
|
||
text: "SELECT * FROM raw_data WHERE key = $1 ORDER BY timestamp DESC LIMIT 300",
|
||
values: [key]
|
||
}, function (err, res) {
|
||
console.log(res);
|
||
if (err) {
|
||
console.error(err);
|
||
ctx.reply(err);
|
||
return;
|
||
}
|
||
var rows = res.rows;
|
||
console.log("Rows: " + rows.length);
|
||
var allValues = [];
|
||
var allTimes = [];
|
||
var rawText = [];
|
||
var minimum = 10000;
|
||
var maximum = 0;
|
||
for (var i = 0; i < rows.length; i++) {
|
||
var time = moment(Number(rows[i].timestamp));
|
||
var value = Number(rows[i].value);
|
||
allValues.unshift(value);
|
||
allTimes.unshift(time.format("MM-DD"));
|
||
if (i < numberOfRecentValuesToPrint - 1) {
|
||
rawText.unshift(time.format("YYYY-MM-DD") + ": " + value.toFixed(2));
|
||
}
|
||
if (value < minimum) {
|
||
minimum = value;
|
||
}
|
||
if (value > maximum) {
|
||
maximum = value;
|
||
}
|
||
}
|
||
if (additionalValue) {
|
||
allValues.push(additionalValue);
|
||
allTimes.push(moment());
|
||
rawText.push(moment().format("YYYY-MM-DD") +
|
||
": " +
|
||
Number(additionalValue).toFixed(2));
|
||
}
|
||
if (numberOfRecentValuesToPrint > 0) {
|
||
ctx.reply(rawText.join("\n") + "\nMinimum: " + minimum + "\nMaximum: " + maximum);
|
||
}
|
||
if (!skipImage) {
|
||
minimum -= 2;
|
||
maximum += 2;
|
||
var url = "https://chart.googleapis.com/chart?cht=lc&chd=t:" +
|
||
allValues.join(",") +
|
||
"&chs=800x350&chl=" +
|
||
allTimes.join("%7C") +
|
||
"&chtt=" +
|
||
key +
|
||
"&chf=bg,s,e0e0e0&chco=000000,0000FF&chma=30,30,30,30&chds=" +
|
||
minimum +
|
||
"," +
|
||
maximum;
|
||
console.log(url);
|
||
ctx.replyWithPhoto({
|
||
url: url
|
||
});
|
||
}
|
||
var queryToUse = "SELECT";
|
||
var weekTimestamp = moment()
|
||
.subtract(7, "days")
|
||
.unix() * 1000;
|
||
var monthTimestamp = moment()
|
||
.subtract(30, "days")
|
||
.unix() * 1000;
|
||
var quarterTimestamp = moment()
|
||
.subtract(90, "days")
|
||
.unix() * 1000;
|
||
var yearTimestamp = moment()
|
||
.subtract(365, "days")
|
||
.unix() * 1000;
|
||
var athTimestamp = moment("2019-04-12").unix() * 1000;
|
||
if (key == "mood") {
|
||
athTimestamp = moment("2018-02-01").unix() * 1000;
|
||
}
|
||
queryToUse += "(SELECT ROUND(AVG(value::numeric), 4) FROM raw_data WHERE timestamp > " + weekTimestamp + " AND key='" + key + "') as " + key + "Week,";
|
||
queryToUse += "(SELECT ROUND(AVG(value::numeric), 4) FROM raw_data WHERE timestamp > " + monthTimestamp + " AND key='" + key + "') as " + key + "Month,";
|
||
queryToUse += "(SELECT ROUND(AVG(value::numeric), 4) FROM raw_data WHERE timestamp > " + quarterTimestamp + " AND key='" + key + "') as " + key + "Quarter,";
|
||
queryToUse += "(SELECT ROUND(AVG(value::numeric), 4) FROM raw_data WHERE timestamp > " + yearTimestamp + " AND key='" + key + "') as " + key + "Year,";
|
||
queryToUse += "(SELECT ROUND(AVG(value::numeric), 4) FROM raw_data WHERE timestamp > " + athTimestamp + " AND key='" + key + "') as " + key + "AllTime,";
|
||
queryToUse += "(SELECT COUNT(value::numeric) FROM raw_data WHERE timestamp > " + weekTimestamp + " AND key='" + key + "') as " + key + "WeekCount,";
|
||
queryToUse += "(SELECT COUNT(value::numeric) FROM raw_data WHERE timestamp > " + monthTimestamp + " AND key='" + key + "') as " + key + "MonthCount,";
|
||
queryToUse += "(SELECT COUNT(value::numeric) FROM raw_data WHERE timestamp > " + quarterTimestamp + " AND key='" + key + "') as " + key + "QuarterCount,";
|
||
queryToUse += "(SELECT COUNT(value::numeric) FROM raw_data WHERE timestamp > " + yearTimestamp + " AND key='" + key + "') as " + key + "YearCount,";
|
||
queryToUse += "(SELECT COUNT(value::numeric) FROM raw_data WHERE timestamp > " + athTimestamp + " AND key='" + key + "') as " + key + "AllTimeCount";
|
||
console.log(queryToUse);
|
||
postgres.client.query({
|
||
text: queryToUse
|
||
}, function (err, res) {
|
||
var rows = ["week", "month", "quarter", "year", "alltime"];
|
||
var c = res.rows[0];
|
||
console.log(c);
|
||
var finalText = ["Moving averages for " + key];
|
||
for (var i = 0; i < rows.length; i++) {
|
||
finalText.push(roundNumberExactly(c[key + rows[i]], 2) + " - " + rows[i]);
|
||
}
|
||
ctx.reply(finalText.join("\n"));
|
||
});
|
||
});
|
||
}
|
||
function triggerNextQuestionFromQueue(ctx) {
|
||
var keyboard = Extra.markup(function (m) { return m.removeKeyboard(); });
|
||
var questionAppendix = "";
|
||
currentlyAskedQuestionObject = currentlyAskedQuestionQueue.shift();
|
||
if (currentlyAskedQuestionObject == null) {
|
||
ctx.reply("All done for now, let's do this 💪", keyboard);
|
||
return;
|
||
}
|
||
if (currentlyAskedQuestionObject.question == null) {
|
||
console.error("No text defined for");
|
||
console.error(currentlyAskedQuestionObject);
|
||
}
|
||
if (currentlyAskedQuestionObject.type == "header") {
|
||
ctx
|
||
.reply(currentlyAskedQuestionObject.question, keyboard)
|
||
.then(function (_a) {
|
||
var message_id = _a.message_id;
|
||
triggerNextQuestionFromQueue(ctx);
|
||
});
|
||
return;
|
||
}
|
||
if (currentlyAskedQuestionObject.type == "range") {
|
||
var allButtons = [
|
||
[getButtonText("5")],
|
||
[getButtonText("4")],
|
||
[getButtonText("3")],
|
||
[getButtonText("2")],
|
||
[getButtonText("1")],
|
||
[getButtonText("0")]
|
||
];
|
||
shuffleArray(allButtons);
|
||
keyboard = Markup.keyboard(allButtons)
|
||
.oneTime()
|
||
.extra();
|
||
}
|
||
else if (currentlyAskedQuestionObject.type == "boolean") {
|
||
keyboard = Markup.keyboard([["1: Yes"], ["0: No"]])
|
||
.oneTime()
|
||
.extra();
|
||
}
|
||
else if (currentlyAskedQuestionObject.type == "text") {
|
||
questionAppendix +=
|
||
"You can use a Bear note, and then paste the deep link to the note here";
|
||
}
|
||
else if (currentlyAskedQuestionObject.type == "location") {
|
||
keyboard = Extra.markup(function (markup) {
|
||
return markup.keyboard([
|
||
markup.locationRequestButton("📡 Send location")
|
||
]);
|
||
});
|
||
}
|
||
questionAppendix = currentlyAskedQuestionQueue.length + " more question";
|
||
if (currentlyAskedQuestionQueue.length != 1) {
|
||
questionAppendix += "s";
|
||
}
|
||
if (currentlyAskedQuestionQueue.length == 0) {
|
||
questionAppendix = "last question";
|
||
}
|
||
var question = currentlyAskedQuestionObject.question + " (" + questionAppendix + ")";
|
||
ctx.reply(question, keyboard).then(function (_a) {
|
||
var message_id = _a.message_id;
|
||
currentlyAskedQuestionMessageId = message_id;
|
||
});
|
||
if (currentlyAskedQuestionObject.type == "number" ||
|
||
currentlyAskedQuestionObject.type == "range" ||
|
||
currentlyAskedQuestionObject.type == "boolean") {
|
||
printGraph(currentlyAskedQuestionObject.key, ctx, 0, null, false);
|
||
}
|
||
}
|
||
function shuffleArray(array) {
|
||
var _a;
|
||
for (var i = array.length - 1; i > 0; i--) {
|
||
var j = Math.floor(Math.random() * (i + 1));
|
||
_a = [array[j], array[i]], array[i] = _a[0], array[j] = _a[1];
|
||
}
|
||
}
|
||
function insertNewValue(parsedUserValue, ctx, key, type, fakeDate) {
|
||
if (fakeDate === void 0) { fakeDate = null; }
|
||
console.log("Inserting value '" + parsedUserValue + "' for key " + key);
|
||
var dateToAdd;
|
||
if (fakeDate) {
|
||
dateToAdd = fakeDate;
|
||
}
|
||
else {
|
||
dateToAdd = moment(ctx.update.message.date * 1000);
|
||
}
|
||
var questionText = null;
|
||
if (currentlyAskedQuestionObject) {
|
||
questionText = currentlyAskedQuestionObject.question;
|
||
}
|
||
var row = {
|
||
Timestamp: dateToAdd.valueOf(),
|
||
YearMonth: dateToAdd.format("YYYYMM"),
|
||
YearWeek: dateToAdd.format("YYYYWW"),
|
||
Year: dateToAdd.year(),
|
||
Quarter: dateToAdd.quarter(),
|
||
Month: dateToAdd.format("MM"),
|
||
Day: dateToAdd.date(),
|
||
Hour: dateToAdd.hours(),
|
||
Minute: dateToAdd.minutes(),
|
||
Week: dateToAdd.week(),
|
||
Key: key,
|
||
Question: questionText,
|
||
Type: type,
|
||
Value: parsedUserValue,
|
||
Source: "telegram",
|
||
Importedat: moment(ctx.update.message.date * 1000),
|
||
Importid: null
|
||
};
|
||
postgres.client.query({
|
||
text: "INSERT INTO raw_data (" +
|
||
Object.keys(row).join(",") +
|
||
") VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17)",
|
||
values: Object.values(row)
|
||
}, function (err, res) {
|
||
console.log(res);
|
||
if (err) {
|
||
ctx.reply("Error saving value: " + err);
|
||
console.log(err.stack);
|
||
}
|
||
else {
|
||
}
|
||
});
|
||
if (ctx) {
|
||
}
|
||
}
|
||
function parseUserInput(ctx, text) {
|
||
if (text === void 0) { text = null; }
|
||
if (ctx.update.message.from.username != process.env.TELEGRAM_USER_ID) {
|
||
console.error("Invalid user " + ctx.update.message.from.username);
|
||
return;
|
||
}
|
||
if (currentlyAskedQuestionMessageId == null) {
|
||
ctx
|
||
.reply("Sorry, I forgot the question I asked, this usually means it took too long for you to respond, please trigger the question again by running the `/` command")
|
||
.then(function (_a) {
|
||
var message_id = _a.message_id;
|
||
sendAvailableCommands(ctx);
|
||
});
|
||
return;
|
||
}
|
||
var userValue;
|
||
if (text != null) {
|
||
userValue = text;
|
||
}
|
||
else {
|
||
userValue = ctx.match[1];
|
||
}
|
||
var parsedUserValue = null;
|
||
if (currentlyAskedQuestionObject.type != "text") {
|
||
if (currentlyAskedQuestionObject.type == "range" ||
|
||
currentlyAskedQuestionObject.type == "boolean") {
|
||
var tryToParseNumber = parseInt(userValue[0]);
|
||
if (!isNaN(tryToParseNumber)) {
|
||
parsedUserValue = tryToParseNumber;
|
||
}
|
||
else {
|
||
ctx.reply("Sorry, looks like your input is invalid, please enter a valid number from the selection", Extra.inReplyTo(ctx.update.message.message_id));
|
||
}
|
||
}
|
||
if (parsedUserValue == null) {
|
||
userValue = userValue.match(/^(\d+(\.\d+)?)$/);
|
||
if (userValue == null) {
|
||
ctx.reply("Sorry, looks like you entered an invalid number, please try again", Extra.inReplyTo(ctx.update.message.message_id));
|
||
return;
|
||
}
|
||
parsedUserValue = userValue[1];
|
||
}
|
||
}
|
||
else {
|
||
parsedUserValue = userValue;
|
||
}
|
||
if (currentlyAskedQuestionObject.type == "range") {
|
||
if (parsedUserValue < 0 || parsedUserValue > 6) {
|
||
ctx.reply("Please enter a value from 0 to 6", Extra.inReplyTo(ctx.update.message.message_id));
|
||
return;
|
||
}
|
||
}
|
||
if (currentlyAskedQuestionObject.type == "number" ||
|
||
currentlyAskedQuestionObject.type == "range" ||
|
||
currentlyAskedQuestionObject.type == "boolean") {
|
||
printGraph(currentlyAskedQuestionObject.key, ctx, 5, parsedUserValue, true);
|
||
}
|
||
console.log("Got a new value: " +
|
||
parsedUserValue +
|
||
" for question " +
|
||
currentlyAskedQuestionObject.key);
|
||
if (currentlyAskedQuestionObject.replies &&
|
||
currentlyAskedQuestionObject.replies[parsedUserValue]) {
|
||
ctx.reply(currentlyAskedQuestionObject.replies[parsedUserValue], Extra.inReplyTo(ctx.update.message.message_id));
|
||
}
|
||
insertNewValue(parsedUserValue, ctx, currentlyAskedQuestionObject.key, currentlyAskedQuestionObject.type);
|
||
setTimeout(function () {
|
||
triggerNextQuestionFromQueue(ctx);
|
||
}, 50);
|
||
}
|
||
function sendAvailableCommands(ctx) {
|
||
ctx.reply("Available commands:").then(function (_a) {
|
||
var message_id = _a.message_id;
|
||
ctx.reply("\n\n/skip\n/report\n\n/" + Object.keys(config.userConfig).join("\n/"));
|
||
});
|
||
}
|
||
function saveLastRun(command) {
|
||
postgres.client.query({
|
||
text: "insert into last_run (command, last_run) VALUES ($1, $2) on conflict (command) do update set last_run = $2",
|
||
values: [command, moment().valueOf()]
|
||
}, function (err, res) {
|
||
console.log(res);
|
||
if (err) {
|
||
console.log(err.stack);
|
||
}
|
||
else {
|
||
console.log("Stored timestamp of last run for " + command);
|
||
}
|
||
});
|
||
}
|
||
function initBot() {
|
||
console.log("Launching up Telegram bot...");
|
||
bot.hears(/^([^\/].*)$/, function (ctx) {
|
||
parseUserInput(ctx);
|
||
});
|
||
bot.hears("/skip", function (ctx) {
|
||
if (ctx.update.message.from.username != process.env.TELEGRAM_USER_ID) {
|
||
console.error("Invalid user " + ctx.update.message.from.username);
|
||
return;
|
||
}
|
||
console.log("user is skipping this question");
|
||
ctx.reply("Okay, skipping question. If you see yourself skipping a question too often, maybe it's time to rephrase or remove it");
|
||
triggerNextQuestionFromQueue(ctx);
|
||
});
|
||
bot.hears("/skip_all", function (ctx) {
|
||
if (ctx.update.message.from.username != process.env.TELEGRAM_USER_ID) {
|
||
return;
|
||
}
|
||
currentlyAskedQuestionQueue = [];
|
||
triggerNextQuestionFromQueue(ctx);
|
||
ctx.reply("Okay, removing all questions that are currently in the queue");
|
||
});
|
||
bot.hears(/\/track (\w+)/, function (ctx) {
|
||
if (ctx.update.message.from.username != process.env.TELEGRAM_USER_ID) {
|
||
console.error("Invalid user " + ctx.update.message.from.username);
|
||
return;
|
||
}
|
||
var toTrack = ctx.match[1];
|
||
console.log("User wants to track a specific value, without the whole survey: " +
|
||
toTrack);
|
||
var questionToAsk = null;
|
||
Object.keys(config.userConfig).forEach(function (key) {
|
||
var survey = config.userConfig[key];
|
||
for (var i = 0; i < survey.questions.length; i++) {
|
||
var currentQuestion = survey.questions[i];
|
||
if (currentQuestion.key == toTrack) {
|
||
questionToAsk = currentQuestion;
|
||
return;
|
||
}
|
||
}
|
||
});
|
||
if (questionToAsk) {
|
||
currentlyAskedQuestionQueue = currentlyAskedQuestionQueue.concat(questionToAsk);
|
||
triggerNextQuestionFromQueue(ctx);
|
||
}
|
||
else {
|
||
ctx.reply("Sorry, I couldn't find the key `" +
|
||
toTrack +
|
||
"`, please make sure it's not mispelled");
|
||
}
|
||
});
|
||
bot.hears(/\/graph (\w+)/, function (ctx) {
|
||
if (ctx.update.message.from.username != process.env.TELEGRAM_USER_ID) {
|
||
return;
|
||
}
|
||
var key = ctx.match[1];
|
||
console.log("User wants to graph a specific value " + key);
|
||
printGraph(key, ctx, 100, null, false);
|
||
});
|
||
bot.on("location", function (ctx) {
|
||
if (ctx.update.message.from.username != process.env.TELEGRAM_USER_ID) {
|
||
return;
|
||
}
|
||
if (currentlyAskedQuestionMessageId == null) {
|
||
ctx
|
||
.reply("Sorry, I forgot the question I asked, this usually means it took too long for you to respond, please trigger the question again by running the `/` command")
|
||
.then(function (_a) {
|
||
var message_id = _a.message_id;
|
||
sendAvailableCommands(ctx);
|
||
});
|
||
return;
|
||
}
|
||
var location = ctx.update.message.location;
|
||
var lat = location.latitude;
|
||
var lng = location.longitude;
|
||
insertNewValue(lat, ctx, "locationLat", "number");
|
||
insertNewValue(lng, ctx, "locationLng", "number");
|
||
triggerNextQuestionFromQueue(ctx);
|
||
});
|
||
bot.hears(/\/(\w+)/, function (ctx) {
|
||
if (ctx.update.message.from.username != process.env.TELEGRAM_USER_ID) {
|
||
return;
|
||
}
|
||
var command = ctx.match[1];
|
||
var matchingCommandObject = config.userConfig[command];
|
||
if (matchingCommandObject && matchingCommandObject.questions) {
|
||
console.log("User wants to run: " + command);
|
||
saveLastRun(command);
|
||
if (currentlyAskedQuestionQueue.length > 0 &&
|
||
currentlyAskedQuestionMessageId) {
|
||
ctx.reply("^ Okay, but please answer my previous question also, thanks ^", Extra.inReplyTo(currentlyAskedQuestionMessageId));
|
||
}
|
||
currentlyAskedQuestionQueue = currentlyAskedQuestionQueue.concat(matchingCommandObject.questions.slice(0));
|
||
if (currentlyAskedQuestionObject == null) {
|
||
triggerNextQuestionFromQueue(ctx);
|
||
}
|
||
}
|
||
else {
|
||
ctx
|
||
.reply("Sorry, I don't know how to run `/" + command)
|
||
.then(function (_a) {
|
||
var message_id = _a.message_id;
|
||
sendAvailableCommands(ctx);
|
||
});
|
||
}
|
||
});
|
||
bot.start(function (ctx) { return ctx.reply("Welcome to FxLifeSheet"); });
|
||
bot.help(function (ctx) {
|
||
return ctx.reply("No in-bot help right now, for now please visit https://github.com/KrauseFx/FxLifeSheet");
|
||
});
|
||
bot.on("sticker", function (ctx) { return ctx.reply("Sorry, I don't support stickers"); });
|
||
bot.hears("hi", function (ctx) { return ctx.reply("Hey there"); });
|
||
bot.launch();
|
||
}
|
||
//# sourceMappingURL=worker.js.map
|