f0ckv2/src/inc/lib.mjs

123 lines
3.9 KiB
JavaScript
Raw Normal View History

2021-05-19 14:19:26 +02:00
import crypto from "crypto";
2021-12-04 12:19:47 +01:00
import util from "util";
import sql from "./sql.mjs";
const scrypt = util.promisify(crypto.scrypt);
2021-05-19 14:19:26 +02:00
2021-04-17 10:43:23 +02:00
const epochs = [
["year", 31536000],
["month", 2592000],
["day", 86400],
["hour", 3600],
["minute", 60],
["second", 1]
];
const getDuration = timeAgoInSeconds => {
for(let [name, seconds] of epochs) {
const interval = ~~(timeAgoInSeconds / seconds);
if(interval >= 1) return {
interval: interval,
epoch: name
};
}
};
2020-04-03 13:12:35 +02:00
export default new class {
formatSize(size, i = ~~(Math.log(size) / Math.log(1024))) {
return (size / Math.pow(1024, i)).toFixed(2) * 1 + " " + ["B", "kB", "MB", "GB", "TB"][i];
2021-12-04 12:19:47 +01:00
};
2020-04-09 18:44:09 +02:00
calcSpeed(b, s) {
2021-01-25 20:30:39 +01:00
return (Math.round((b * 8 / s / 1e6) * 1e4) / 1e4);
2021-12-04 12:19:47 +01:00
};
2021-04-17 10:43:23 +02:00
timeAgo(date) {
const { interval, epoch } = getDuration(~~((new Date() - new Date(date)) / 1e3));
return `${interval} ${epoch}${interval === 1 ? "" : "s"} ago`;
2021-12-04 12:19:47 +01:00
};
2021-05-19 14:19:26 +02:00
md5(str) {
return crypto.createHash('md5').update(str).digest("hex");
2021-12-04 12:19:47 +01:00
};
getMode(mode) {
let tmp;
switch(mode) {
case 1: // nsfw
2021-12-18 19:52:38 +01:00
tmp = "items.id in (select item_id from tags_assign where tag_id = 2 group by item_id)";
2021-12-04 12:19:47 +01:00
break;
case 2: // untagged
2021-12-18 19:52:38 +01:00
tmp = "items.id not in (select item_id from tags_assign group by item_id)";
2021-12-04 12:19:47 +01:00
break;
case 3: // all
tmp = "";
break;
default: // sfw
2021-12-18 19:52:38 +01:00
tmp = "items.id in (select item_id from tags_assign where tag_id = 1 group by item_id)";
2021-12-04 12:19:47 +01:00
break;
}
return tmp;
};
createID() {
return crypto.randomBytes(16).toString("hex") + Date.now().toString(24);
};
2021-12-06 10:21:11 +01:00
genLink(env) {
const link = [];
if(env.user) link.push("user", env.user);
2021-12-18 19:52:38 +01:00
if(env.tag) link.push("tag", env.tag);
2021-12-06 10:21:11 +01:00
if(env.mime.length > 2) link.push(env.mime);
if(env.page) link.push("p", env.page);
return link.join("/");
};
2021-12-18 19:52:38 +01:00
parseTag(tag) {
2021-12-20 17:55:47 +01:00
if(!tag)
return null;
2021-12-29 05:48:04 +01:00
return decodeURI(tag);
2021-12-18 19:52:38 +01:00
}
2021-12-04 12:19:47 +01:00
// async funcs
async countf0cks() {
const tagged = (await sql("items").whereRaw("id in (select item_id from tags_assign group by item_id)").count("* as total"))[0].total;
const untagged = (await sql("items").whereRaw("id not in (select item_id from tags_assign group by item_id)").count("* as total"))[0].total;
const sfw = (await sql("items").whereRaw("id in (select item_id from tags_assign where tag_id = 1 group by item_id)").count("* as total"))[0].total;
const nsfw = (await sql("items").whereRaw("id in (select item_id from tags_assign where tag_id = 2 group by item_id)").count("* as total"))[0].total;
return {
tagged,
untagged,
total: tagged + untagged,
sfw,
nsfw
};
};
async hash(str) {
const salt = crypto.randomBytes(16).toString("hex");
const derivedKey = await scrypt(str, salt, 64);
return "$f0ck$" + salt + ":" + derivedKey.toString("hex");
};
async verify(str, hash) {
const [ salt, key ] = hash.substring(6).split(":");
const keyBuffer = Buffer.from(key, "hex");
const derivedKey = await scrypt(str, salt, 64);
return crypto.timingSafeEqual(keyBuffer, derivedKey);
};
2021-12-20 17:55:47 +01:00
async getTags(itemid) {
const tags = await sql("tags_assign")
2021-12-23 10:07:12 +01:00
.select("tags.id", "tags.tag", "user.user")
2021-12-20 17:55:47 +01:00
.leftJoin("tags", "tags.id", "tags_assign.tag_id")
2021-12-23 10:07:12 +01:00
.leftJoin("user", "user.id", "tags_assign.user_id")
2021-12-20 17:55:47 +01:00
.where("tags_assign.item_id", itemid);
2021-12-29 05:48:04 +01:00
for(let t = 0; t < tags.length; t++) {
if(tags[t].tag.startsWith(">"))
tags[t].badge = "badge-greentext badge-light";
else if(/[а-яА-ЯЁё]/.test(tags[t].tag) || tags[t].tag.match(/russia/))
tags[t].badge = "badge-russia badge-light";
else if(tags[t].tag.match(/german/))
tags[t].badge = "badge-german badge-light";
else if(tags[t].tag === "sfw")
tags[t].badge = "badge-success";
else if(tags[t].tag === "nsfw")
tags[t].badge = "badge-danger";
else
tags[t].badge = "badge-light";
}
2021-12-20 17:55:47 +01:00
return tags;
};
2021-12-04 12:19:47 +01:00
2020-04-03 13:12:35 +02:00
};