From 04a851a1381a0bd6d9a702819c0f68577ef93369 Mon Sep 17 00:00:00 2001
From: Matt Dunn-Rankin <mattdunnrankin@khanacademy.org>
Date: Sun, 3 May 2020 01:04:34 -0700
Subject: [PATCH] new endpoint for blob of valid pet poses

---
 api/validPetPoses.js                          |   6 ++
 .../getValidPetPoses.test.js.snap             | Bin 0 -> 6335 bytes
 src/server/getValidPetPoses.js                |  73 ++++++++++++++++++
 src/server/getValidPetPoses.test.js           |   8 ++
 src/server/util.js                            |  14 ++--
 5 files changed, 95 insertions(+), 6 deletions(-)
 create mode 100644 api/validPetPoses.js
 create mode 100644 src/server/__snapshots__/getValidPetPoses.test.js.snap
 create mode 100644 src/server/getValidPetPoses.js
 create mode 100644 src/server/getValidPetPoses.test.js

diff --git a/api/validPetPoses.js b/api/validPetPoses.js
new file mode 100644
index 0000000..027caea
--- /dev/null
+++ b/api/validPetPoses.js
@@ -0,0 +1,6 @@
+import getValidPetPoses from "../src/server/getValidPetPoses";
+
+export default async (req, res) => {
+  const buffer = await getValidPetPoses();
+  res.status(200).send(buffer);
+};
diff --git a/src/server/__snapshots__/getValidPetPoses.test.js.snap b/src/server/__snapshots__/getValidPetPoses.test.js.snap
new file mode 100644
index 0000000000000000000000000000000000000000..67045aeec450192acc8e5173101e1dab542bfeb6
GIT binary patch
literal 6335
zcmeIuy9&ZU6og^xJjIY&T+%8Qwqg+xY(&@%IlBqOY{)rY9^YW?Bg}O3m!ESuiW)9s
zn6%4=SG7iGHtE*6l=CJHE;fhrsV)lnOnElF`6PA?L*Le7byiU<L1CRd5XKhoxwj?$
zzGVb7&lqJ!<?q<Tmn)6}6rcbFC_n)UP=Epypa2CZKmiI+fC3bt00k&O0Sf#VDE+SZ
E0{WgPhX4Qo

literal 0
HcmV?d00001

diff --git a/src/server/getValidPetPoses.js b/src/server/getValidPetPoses.js
new file mode 100644
index 0000000..e17e09c
--- /dev/null
+++ b/src/server/getValidPetPoses.js
@@ -0,0 +1,73 @@
+import connectToDb from "./db";
+
+import { getEmotion, getGenderPresentation } from "./util";
+
+export default async function getValidPetPoses() {
+  const db = await connectToDb();
+
+  const numSpeciesPromise = getNumSpecies(db);
+  const numColorsPromise = getNumColors(db);
+  const poseTuplesPromise = getPoseTuples(db);
+
+  const [numSpecies, numColors, poseTuples] = await Promise.all([
+    numSpeciesPromise,
+    numColorsPromise,
+    poseTuplesPromise,
+  ]);
+
+  const poseStrs = new Set();
+  for (const poseTuple of poseTuples) {
+    const { species_id, color_id, mood_id, female } = poseTuple;
+    const emotion = getEmotion(mood_id);
+    const genderPresentation = getGenderPresentation(female);
+    const poseStr = `${species_id}-${color_id}-${emotion}-${genderPresentation}`;
+    poseStrs.add(poseStr);
+  }
+
+  function hasPose(speciesId, colorId, emotion, genderPresentation) {
+    const poseStr = `${speciesId}-${colorId}-${emotion}-${genderPresentation}`;
+    return poseStrs.has(poseStr);
+  }
+
+  const numPairs = numSpecies * numColors;
+  const buffer = Buffer.alloc(numPairs);
+
+  for (let speciesId = 1; speciesId <= numSpecies; speciesId++) {
+    for (let colorId = 1; colorId <= numColors; colorId++) {
+      let byte = 0;
+      byte += hasPose(speciesId, colorId, "HAPPY", "MASCULINE") ? 1 : 0;
+      byte <<= 1;
+      byte += hasPose(speciesId, colorId, "SAD", "MASCULINE") ? 1 : 0;
+      byte <<= 1;
+      byte += hasPose(speciesId, colorId, "SICK", "MASCULINE") ? 1 : 0;
+      byte <<= 1;
+      byte += hasPose(speciesId, colorId, "HAPPY", "FEMININE") ? 1 : 0;
+      byte <<= 1;
+      byte += hasPose(speciesId, colorId, "SAD", "FEMININE") ? 1 : 0;
+      byte <<= 1;
+      byte += hasPose(speciesId, colorId, "SICK", "FEMININE") ? 1 : 0;
+
+      buffer.writeUInt8(byte);
+    }
+  }
+
+  return buffer;
+}
+
+async function getNumSpecies(db) {
+  const [rows, _] = await db.query(`SELECT count(*) FROM species`);
+  return rows[0]["count(*)"];
+}
+
+async function getNumColors(db) {
+  const [rows, _] = await db.query(`SELECT count(*) FROM colors`);
+  return rows[0]["count(*)"];
+}
+
+async function getPoseTuples(db) {
+  const [rows, _] = await db.query(`
+    SELECT DISTINCT species_id, color_id, mood_id, female FROM pet_states
+    INNER JOIN pet_types ON pet_types.id = pet_states.pet_type_id
+    WHERE mood_id IS NOT NULL AND female IS NOT NULL AND color_id >= 1`);
+  return rows;
+}
diff --git a/src/server/getValidPetPoses.test.js b/src/server/getValidPetPoses.test.js
new file mode 100644
index 0000000..18ee109
--- /dev/null
+++ b/src/server/getValidPetPoses.test.js
@@ -0,0 +1,8 @@
+import getValidPetPoses from "./getValidPetPoses";
+
+describe("getValidPetPoses", () => {
+  it("gets them and writes them to a buffer", async () => {
+    const buffer = await getValidPetPoses();
+    expect(buffer.toString()).toMatchSnapshot();
+  });
+});
diff --git a/src/server/util.js b/src/server/util.js
index ef692ea..88f95d1 100644
--- a/src/server/util.js
+++ b/src/server/util.js
@@ -3,13 +3,14 @@ function capitalize(str) {
 }
 
 function getEmotion(moodId) {
-  if (moodId === "1") {
+  const moodIdStr = String(moodId);
+  if (moodIdStr === "1") {
     return "HAPPY";
-  } else if (moodId === "2") {
+  } else if (moodIdStr === "2") {
     return "SAD";
-  } else if (moodId === "4") {
+  } else if (moodIdStr === "4") {
     return "SICK";
-  } else if (moodId === null) {
+  } else if (moodIdStr === null) {
     return null;
   } else {
     throw new Error(`unrecognized moodId ${JSON.stringify(moodId)}`);
@@ -17,9 +18,10 @@ function getEmotion(moodId) {
 }
 
 function getGenderPresentation(modelPetWasFemale) {
-  if (modelPetWasFemale === 1) {
+  const modelPetWasFemaleStr = String(modelPetWasFemale);
+  if (modelPetWasFemaleStr === "1") {
     return "FEMININE";
-  } else if (modelPetWasFemale === 0) {
+  } else if (modelPetWasFemaleStr === "0") {
     return "MASCULINE";
   } else {
     return null;