impress-2020/src/server/util.js

152 lines
4.4 KiB
JavaScript
Raw Normal View History

const beeline = require("honeycomb-beeline");
const fetch = require("node-fetch");
2020-04-25 04:33:05 -07:00
function capitalize(str) {
return str[0].toUpperCase() + str.slice(1);
}
2020-05-23 12:47:06 -07:00
function getPoseFromPetState(petState) {
const { moodId, female, unconverted } = petState;
if (unconverted) {
2020-05-23 11:48:53 -07:00
return "UNCONVERTED";
2020-05-23 12:47:06 -07:00
} else if (moodId == null || female == null) {
2020-05-23 11:48:53 -07:00
return "UNKNOWN";
2020-05-23 12:47:06 -07:00
} else if (String(moodId) === "1" && String(female) === "0") {
2020-05-23 11:48:53 -07:00
return "HAPPY_MASC";
2020-05-23 12:47:06 -07:00
} else if (String(moodId) === "1" && String(female) === "1") {
2020-05-23 11:48:53 -07:00
return "HAPPY_FEM";
2020-05-23 12:47:06 -07:00
} else if (String(moodId) === "2" && String(female) === "0") {
2020-05-23 11:48:53 -07:00
return "SAD_MASC";
2020-05-23 12:47:06 -07:00
} else if (String(moodId) === "2" && String(female) === "1") {
2020-05-23 11:48:53 -07:00
return "SAD_FEM";
2020-05-23 12:47:06 -07:00
} else if (String(moodId) === "4" && String(female) === "0") {
2020-05-23 11:48:53 -07:00
return "SICK_MASC";
2020-05-23 12:47:06 -07:00
} else if (String(moodId) === "4" && String(female) === "1") {
2020-05-23 11:48:53 -07:00
return "SICK_FEM";
} else {
throw new Error(
`could not identify pose: ` +
`moodId=${moodId}, ` +
2020-05-23 12:47:06 -07:00
`female=${female}, ` +
`unconverted=${unconverted}`
2020-05-23 11:48:53 -07:00
);
}
}
function getPetStateFieldsFromPose(pose) {
if (pose === "UNCONVERTED") {
return { moodId: null, female: null, unconverted: true, labeled: true };
} else if (pose === "UNKNOWN") {
return { moodId: null, female: null, unconverted: false, labeled: false };
} else if (pose === "HAPPY_MASC") {
return { moodId: "1", female: false, unconverted: false, labeled: true };
} else if (pose === "HAPPY_FEM") {
return { moodId: "1", female: true, unconverted: false, labeled: true };
} else if (pose === "SAD_MASC") {
return { moodId: "2", female: false, unconverted: false, labeled: true };
} else if (pose === "SAD_FEM") {
return { moodId: "2", female: true, unconverted: false, labeled: true };
} else if (pose === "SICK_MASC") {
return { moodId: "4", female: false, unconverted: false, labeled: true };
} else if (pose === "SICK_FEM") {
return { moodId: "4", female: true, unconverted: false, labeled: true };
} else {
throw new Error(`unexpected pose ${pose}`);
}
}
const POSE_NAMES = {
HAPPY_MASC: "Happy Masc",
SAD_MASC: "Sad Masc",
SICK_MASC: "Sick Masc",
HAPPY_FEM: "Happy Fem",
SAD_FEM: "Sad Fem",
SICK_FEM: "Sick Fem",
UNCONVERTED: "Unconverted",
UNKNOWN: "Unknown",
};
function getPoseName(pose) {
return POSE_NAMES[pose];
}
2020-08-31 19:23:56 -07:00
function getRestrictedZoneIds(zonesRestrict) {
const restrictedZoneIds = [];
for (const [i, bit] of Array.from(zonesRestrict).entries()) {
if (bit === "1") {
restrictedZoneIds.push(i + 1);
}
}
return restrictedZoneIds;
}
async function loadBodyName(bodyId, db) {
if (String(bodyId) === "0") {
return "All bodies";
}
const [rows] = await db.execute(
`SELECT pt.body_id, st.name AS species_name,
ct.name AS color_name, c.standard FROM pet_types pt
INNER JOIN species_translations st
ON pt.species_id = st.species_id AND st.locale = "en"
INNER JOIN color_translations ct
ON pt.color_id = ct.color_id AND ct.locale = "en"
INNER JOIN colors c ON c.id = pt.color_id
WHERE pt.body_id = ?
ORDER BY c.standard DESC, ct.name, st.name LIMIT 1;`,
[bodyId]
);
const row = normalizeRow(rows[0]);
const speciesName = capitalize(row.speciesName);
const colorName = row.standard ? "Standard" : capitalize(row.colorName);
return `${colorName} ${speciesName}`;
}
async function logToDiscord(body) {
const span = beeline.startSpan({ name: "logToDiscord" });
try {
const res = await fetch(process.env["SUPPORT_TOOLS_DISCORD_WEBHOOK_URL"], {
method: "POST",
headers: {
"Content-Type": "application/json",
},
body: JSON.stringify(body),
});
if (!res.ok) {
const resText = await res.text();
throw new Error(
`Discord returned ${res.status} ${res.statusText}: ` + `${resText}`
);
}
} finally {
beeline.finishSpan(span);
}
}
2020-05-23 12:47:06 -07:00
function normalizeRow(row) {
const normalizedRow = {};
for (let [key, value] of Object.entries(row)) {
key = key.replace(/_([a-z])/gi, (m) => m[1].toUpperCase());
if ((key === "id" || key.endsWith("Id")) && typeof value === "number") {
value = String(value);
}
normalizedRow[key] = value;
}
return normalizedRow;
}
module.exports = {
capitalize,
getPoseFromPetState,
getPetStateFieldsFromPose,
getPoseName,
2020-08-31 19:23:56 -07:00
getRestrictedZoneIds,
loadBodyName,
logToDiscord,
2020-05-23 12:47:06 -07:00
normalizeRow,
};