cbt.js

// an interval and counter for detecting if the user is on the page that shows the calendar
let findCalendarPageInterval = setInterval(findCalendarPage, 2000);
let findCalendarPageCount = 0;

// a interval that detects whether the user has opened the calendar or not
let listenForCalendarInterval = null;

let calendarOpen = false;

// the courses that the user is registered in, loaded dynamically
let courseData = null;

// detect if the user is on the page that shows the calendar
function findCalendarPage() {
  const bodyText = document.body.innerText;
  const regex = /(View My Courses|View as Course Calendar)/;
  if (bodyText.match(regex) != null) {
    listenForCalendarInterval = setInterval(listenForCalendar, 2000);
    clearInterval(findCalendarPageInterval);
  }
  if (findCalendarPageCount >= 100) {
    clearInterval(findCalendarPageInterval); // stop after a while
  }
  findCalendarPageCount++;
}

// detect if the calendar is opened
function listenForCalendar() {
  const bodyText = document.body.innerHTML;
  const regex = />View as Course Calendar</; // angle brackets because text on its own is in the html when calendar not open
  const textMatched = bodyText.match(regex) != null
  if (textMatched && !calendarOpen) {
    calendarOpen = true;
    injectButtons();
    parseCourseData();
  } else if (!textMatched && calendarOpen) {
    calendarOpen = false;
  }
}

function injectButtons() {
  const calendarToolbar = document.querySelector("[data-automation-id='calendarToolbar']")
  const extensionToolbar = `
    <div class='cbt-toolbar'>
      <a role='button' id='cbt-allButton' class='cbt-button cbt-active'>All</a>
      <a role='button' id='cbt-termOneButton' class='cbt-button'>Term 1</a>
      <a role='button' id='cbt-termTwoButton' class='cbt-button'>Term 2</a>
    </div>
  `;
  
  if (calendarToolbar) {
    calendarToolbar.innerHTML += extensionToolbar;

    const allButton = document.getElementById("cbt-allButton");
    const termOneButton = document.getElementById("cbt-termOneButton");
    const termTwoButton = document.getElementById("cbt-termTwoButton");
    allButton.addEventListener("click", function() {
      resetAll();
      allButton.classList.add("cbt-active");
      termOneButton.classList.remove("cbt-active");
      termTwoButton.classList.remove("cbt-active");
    });
    termOneButton.addEventListener("click", function() {
      chooseTermOne();
      termOneButton.classList.add("cbt-active");
      allButton.classList.remove("cbt-active");
      termTwoButton.classList.remove("cbt-active");
    });
    termTwoButton.addEventListener("click", function() {
      chooseTermTwo();
      termTwoButton.classList.add("cbt-active");
      allButton.classList.remove("cbt-active");
      termOneButton.classList.remove("cbt-active");
    });
  }
}

// reset calendar back to workday default
function resetAll() {
  console.log("resetAll");
  getAllCalendarCourses().forEach((elm) => {
    elm.classList.remove("cbt-hidden");
    elm.parentElement.parentElement.classList.remove("cbt-forceFullWidth");
    elm.parentElement.parentElement.classList.remove("cbt-forcePosition");
  })
}

function chooseTermOne() {
  resetAll();
  getAllCalendarCourses().forEach((elm) => {
    const sectionTitle = elm.innerText.split("\n")[0];
    const sectionData = findSection(sectionTitle);
    if (!(sectionData.startMonth === 9 || sectionData.startMonth === 5)) { // if the course doesnt start on January or May, hide it
      elm.classList.add("cbt-hidden");
    } else {
      elm.parentElement.parentElement.classList.add("cbt-forceFullWidth");
      if (isCourseElementOffset(elm)) {
        elm.parentElement.parentElement.classList.add("cbt-forcePosition");
      }
    }
  })
}

function chooseTermTwo() {
  resetAll();
  getAllCalendarCourses().forEach((elm) => {
    const sectionTitle = elm.innerText.split("\n")[0];
    const sectionData = findSection(sectionTitle);
    if (!(sectionData.endMonth === 4 || sectionData.endMonth === 8)) { // if the course doesnt end on August or April, hide it
      elm.classList.add("cbt-hidden");
    } else {
      elm.parentElement.parentElement.classList.add("cbt-forceFullWidth");
      if (isCourseElementOffset(elm)) {
        elm.parentElement.parentElement.classList.add("cbt-forcePosition");
      }
    }
  })
}

function getAllCalendarCourses() {
  return document.querySelectorAll(".WLSC.WMSC.WMUC.WMVC"); // workday puts these classes for courses in the calendar
}

function findSection(section) {
  const found = courseData.filter((course) => course.section === section);
  return found[0];
}

// some elements are offset if the course times overlap across terms, detect them here and use the forcePosition css class
function isCourseElementOffset(elm) {
  const conditions = ["7.", "21.", "35.", "50", "64.", "78.", "92."];
  return conditions.some((cond) => elm.parentElement.parentElement.style.left.startsWith(cond));
}

// fetch user course data from the html table
function parseCourseData() {
  const courseTables = document.querySelectorAll("[data-automation-id='table']");
  let courses = []
  courseTables.forEach((table) => {
    courses = courses.concat(tableToJson(table));
  })
  courseData = transformCourseJson(courses);
}

// only extract needed course info from the full json
function transformCourseJson(cousesJson) {
  const transformedCourses = cousesJson.map((course) => {
    let newCourseObj = {
      section: course.section.split(" - ")[0], // remove unneccessary info
      startMonth: new Date(Date.parse(course.startDate)).getMonth() + 1, // January is 1 here
      endMonth: new Date(Date.parse(course.endDate)).getMonth() + 1
    };
    return newCourseObj;
  });
  return transformedCourses;
}

// convert the html table to json format for data access
function tableToJson(table) {
  var data = [];
  var headers = ["searchbar", "title", "credits", "grading", "section", "format", "delivery", "pattern", "status", "instructor", "startDate", "endDate"];
  for (var i=2; i<table.rows.length; i++) {
      var tableRow = table.rows[i];
      var rowData = {};
      for (var j=0; j<tableRow.cells.length; j++) {
          rowData[ headers[j] ] = tableRow.cells[j].innerText;
      }
      data.push(rowData);
  }
  return data;
}