1 files changed, 93 insertions, 21 deletions
diff --git a/src/aggietime.js b/src/aggietime.js
index 793481c..e039766 100644
--- a/src/aggietime.js
+++ b/src/aggietime.js
@@ -4,12 +4,19 @@ import {
   USER_PATH,
   USER_CACHE_EXP_SEC,
   CLOCKIN_PATH,
+  CLOCKOUT_PATH,
+  OPEN_SHIFT_PATH,
+  OPEN_SHIFT_EXP_SEC,
 } from "./constants.js";
 
 import { client } from "./axios_client.js";
 
 import expireCache from "expire-cache";
 
+const aggietime = client.create({
+  baseURL: AGGIETIME_URI,
+});
+
 const replace_path_args = (path, map) =>
   path.replaceAll(/:([a-zA-Z0-9_]+)/g, (_, key) => map[key]);
 
@@ -27,18 +34,16 @@ const get_user_position_or_specified = async (position) => {
 
 export const get_user_info = async () => {
   if (!expireCache.get("user")) {
-    const user = await client
-      .get(`${AGGIETIME_URI}/${USER_PATH}`)
-      .then(({ data, config }) => {
-        const csrf_token = config.jar
-          .toJSON()
-          .cookies.find(
-            ({ domain, key }) =>
-              domain === AGGIETIME_DOMAIN && key === "XSRF-TOKEN"
-          ).value;
-        expireCache.set("aggietime-csrf", csrf_token);
-        return data;
-      });
+    const user = await aggietime.get(USER_PATH).then(({ data, config }) => {
+      const csrf_token = config.jar
+        .toJSON()
+        .cookies.find(
+          ({ domain, key }) =>
+            domain === AGGIETIME_DOMAIN && key === "XSRF-TOKEN"
+        ).value;
+      expireCache.set("aggietime-csrf", csrf_token);
+      return data;
+    });
 
     expireCache.set("user", user, USER_CACHE_EXP_SEC);
   }
@@ -48,15 +53,82 @@ export const get_user_info = async () => {
 export const clock_in = async ({ position } = {}) => {
   position = await get_user_position_or_specified(position);
 
-  return await client.post(
-    `${AGGIETIME_URI}/${replace_path_args(CLOCKIN_PATH, { position })}`,
-    {
-      comment: "",
-    },
-    {
-      headers: {
-        "X-XSRF-TOKEN": expireCache.get("aggietime-csrf"),
+  return await aggietime
+    .post(
+      replace_path_args(CLOCKIN_PATH, { position }),
+      {
+        comment: "",
       },
+      {
+        headers: {
+          "X-XSRF-TOKEN": expireCache.get("aggietime-csrf"),
+        },
+      }
+    )
+    .then(({ data }) => {
+      expireCache.remove("status_line");
+      return data;
+    });
+};
+
+export const clock_out = async ({ position } = {}) => {
+  position = await get_user_position_or_specified(position);
+
+  return await aggietime
+    .post(
+      replace_path_args(CLOCKOUT_PATH, { position }),
+      {
+        comment: "",
+      },
+      {
+        headers: {
+          "X-XSRF-TOKEN": expireCache.get("aggietime-csrf"),
+        },
+      }
+    )
+    .then(({ data }) => {
+      expireCache.remove("status_line");
+      return data;
+    });
+};
+
+export const current_shift = async () => {
+  const req_path = replace_path_args(OPEN_SHIFT_PATH, await get_user_info());
+  const {
+    request: {
+      res: { responseUrl: response_url },
+    },
+    data,
+  } = await aggietime.get(req_path);
+
+  if (`${AGGIETIME_URI}/${req_path}` != response_url) {
+    // IMO a very weird decision - when there is no open shift the api redirects
+    // home instead of sending back a 404 or something actually *reasonable* :3
+    return null;
+  }
+
+  return data;
+};
+
+export const get_status_line = async () => {
+  if (!expireCache.get("status_line")) {
+    const { anumber } = await get_user_info();
+    const shift = await current_shift();
+    let status_line = "No Shift";
+
+    if (shift && shift?.start) {
+      const start = new Date(shift?.start);
+      status_line =
+        ((new Date().getTime() - start.getTime()) / (1000 * 60 * 60)).toFixed(
+          2
+        ) + " hours";
     }
-  );
+
+    expireCache.set(
+      "status_line",
+      `${anumber} - ${status_line}`,
+      OPEN_SHIFT_EXP_SEC
+    );
+  }
+  return { status: expireCache.get("status_line") };
 };