<script>
  const courseRelationships = {
    // Year 1
    'ENGL 101': {
      prereqs: [],
      unlocks: ['ENGL 102']
    },
    'ENGL 102': {
      prereqs: ['ENGL 101'],
      unlocks: []
    },
    'MATH 111': {
      prereqs: [],
      unlocks: ['MATH 112', 'PHYS 121']
    },
    'MATH 112': {
      prereqs: ['MATH 111'],
      unlocks: ['MATH 231', 'PHYS 122', 'MATH 211', 'MATH 243']
    },
    'CHEM 115': {
      prereqs: [],
      unlocks: ['MEEN 225']
    },
    'PHYS 121': {
      prereqs: ['MATH 111'],
      unlocks: ['PHYS 122', 'MEEN 240', 'MEEN 200', 'MEEN 225']
    },
    'COSC 114': {
      prereqs: [],
      unlocks: ['COSC 202']
    },
    'GENS 101': {
      prereqs: [],
      unlocks: []
    },
    'GENS 100': {
      prereqs: [],
      unlocks: ['GENS 300']
    },
    'GENS 300': {
      prereqs: ['GENS 100'],
      unlocks: ['GENS 400', 'ENGR 399']
    },
    'HUMA XXX': {
      prereqs: [],
      unlocks: []
    },
  
    // ----------------
    //     Year 2
    // ----------------
    'MEEN 180': {
      prereqs: [],
      unlocks: []
    },
    'PHYS 122': {
      prereqs: ['PHYS 121', 'MATH 112'],
      unlocks: ['MEEN 350']
    },
    'MATH 231': {
      prereqs: ['MATH 112'],
      unlocks: ['MEEN 335', 'MEEN 343']
    },
    'MATH 211': {
      prereqs: ['MATH 112'],
      unlocks: ['MEEN 360', 'MEEN 455', 'MATH 243', 'MEEN 201']
    },
    'MEEN 200': {
      prereqs: ['PHYS 121'],
      unlocks: ['MEEN 201', 'MEEN 325', 'MEEN 455']
    },
    'MEEN 201': {
      prereqs: ['MATH 211', 'MEEN 200'],
      unlocks: ['MEEN 350']
    },
    'MEEN 225': {
      prereqs: ['CHEM 115', 'PHYS 121'],
      unlocks: []
    },
    'MEEN 240': {
      prereqs: ['PHYS 121'],
      unlocks: ['MEEN 335', 'MEEN 343', 'MEEN 441']
    },
    'MATH 243': {
      prereqs: ['MATH 112', 'MATH 211'],
      unlocks: []
    },
    'HUMA XXX': {
      prereqs: [],
      unlocks: []
    },
  
    // Year 3
    'COSC 202': {
      prereqs: ['COSC 114'],
      unlocks: []
    },
    'MEEN 325': {
      prereqs: ['MEEN 200'],
      unlocks: ['MEEN 387', 'MEEN 370']
    },
    'MEEN 335': {
      prereqs: ['MEEN 240', 'MATH 231'],
      unlocks: ['MEEN 343']
    },
    'MEEN 350': {
      prereqs: ['PHYS 122', 'MATH 211', 'MEEN 201'],
      unlocks: ['MEEN 356', 'MEEN 484', 'MEEN 497']
    },
    'MEEN 360': {
      prereqs: ['MATH 211'],
      unlocks: []
    },
    'BUSS 322': {
      prereqs: [],
      unlocks: []
    },
    'MEEN 370': {
      prereqs: ['MEEN 325'],
      unlocks: ['MEEN 497']
    },
    'MEEN 343': {
      prereqs: ['MATH 231', 'MEEN 335', 'MEEN 240'],
      unlocks: []
    },
    'MEEN 356': {
      prereqs: ['MEEN 350'],
      unlocks: []
    },
    'MEEN 387': {
      prereqs: ['MEEN 325'],
      unlocks: ['MEEN 497']
    },
    'ENGR 399': {
      prereqs: ['GENS 300'],
      unlocks: []
    },
  
    // ----------------
    //     Year 4
    // ----------------
    'BUSS XXX': {
      prereqs: [],
      unlocks: []
    },
    'MEEN 455': {
      prereqs: ['MEEN 200', 'MATH 211'],
      unlocks: []
    },
    'MEEN 484': {
      prereqs: ['MEEN 350'],
      unlocks: []
    },
    'MEEN 441': {
      prereqs: ['MEEN 240'],
      unlocks: []
    },
    'MEEN 497': {
      prereqs: ['MEEN 350', 'MEEN 370', 'MEEN 387'],
      unlocks: ['MEEN 498']
    },
    'MEEN 498': {
      prereqs: ['MEEN 497'],
      unlocks: []
    },
    'Major Elective': {
      prereqs: [],
      unlocks: []
    },
    'Technical Elective': {
      prereqs: [],
      unlocks: []
    },
    'GENS 400': {
      prereqs: ['GENS 300'],
      unlocks: []
    }
  };

  const curriculum = {
    Freshman: {
      Fall: [
        { code: 'ENGL 101', name: 'Academic English I', credits: 3 },
        { code: 'MATH 111', name: 'Calculus I', credits: 4 },
        { code: 'CHEM 115', name: 'General Chemistry I', credits: 4 },
        { code: 'GENS 101', name: 'Grand Challenges', credits: 4 },
        { code: 'GENS 100', name: 'Academic Development & Success', credits: 1 }
      ],
      Spring: [
        { code: 'ENGL 102', name: 'Academic English II', credits: 3 },
        { code: 'MATH 112', name: 'Calculus II', credits: 4 },
        { code: 'PHYS 121', name: 'University Physics I', credits: 4 },
        { code: 'COSC 114', name: 'Introduction to Computing Using Python', credits: 4 }
      ],
      Summer: [
        { code: 'HUMA XXX', name: 'Arabic Language / Islamic Studies Elective', credits: 3 }
      ]
    },
  
    Sophomore: {
      Fall: [
        { code: 'MEEN 180', name: 'Computer Aided Design', credits: 3 },
        { code: 'MATH 231', name: 'Calculus III', credits: 3 },
        { code: 'PHYS 122', name: 'University Physics II', credits: 4 },
        { code: 'MATH 211', name: 'Differential Equations and Linear Algebra', credits: 4 },
        { code: 'MEEN 200', name: 'Statics', credits: 3 }
      ],
      Spring: [
        { code: 'HUMA XXX', name: 'UAE Culture (105/106)', credits: 3 },
        { code: 'MATH 243', name: 'Probability and Statistical Inference', credits: 3 },
        { code: 'MEEN 240', name: 'Thermodynamics', credits: 3 },
        { code: 'MEEN 225', name: 'Engineering Materials', credits: 4 },
        { code: 'MEEN 201', name: 'Engineering Dynamics', credits: 3 },
        { code: 'GENS 300', name: 'Career Preparation', credits: 3 }
      ],
      Summer: [
      ]
    },
  
    Junior: {
      Fall: [
        { code: 'COSC 202', name: 'Data Science/AI', credits: 3 },
        { code: 'MEEN 335', name: 'Fluid Mechanics', credits: 4 },
        { code: 'MEEN 350', name: 'Dynamic Systems & Vibration', credits: 3 },
        { code: 'MEEN 360', name: 'Computational Methods for Mechanical Engineers', credits: 3 },
        { code: 'MEEN 325', name: 'Mechanics of Solids', credits: 4 }
      ],
      Spring: [
        { code: 'BUSS 322', name: 'Fundamentals of Innovation & Entrepreneurship', credits: 3 },
        { code: 'MEEN 343', name: 'Heat Transfer', credits: 4 },
        { code: 'MEEN 356', name: 'Computer Controlled Systems', credits: 4 },
        { code: 'MEEN 370', name: 'Introduction to Manufacturing Processes', credits: 4 },
        { code: 'MEEN 387', name: 'Machine Element Design', credits: 3 }
      ],
      Summer: [
        { code: 'ENGR 399', name: 'Engineering Internship', credits: 1 }
      ]
    },
  
    Senior: {
      Fall: [
        { code: 'BUSS XXX', name: 'Any BUSS Elective', credits: 3 },
        { code: 'MEEN 455', name: 'Finite Element Analysis', credits: 3 },
        { code: 'MEEN 484', name: 'Mechatronics', credits: 3 },
        { code: 'MEEN 441', name: 'Applied Thermodynamics', credits: 3 },
        { code: 'MEEN 497', name: 'Senior Design Project I', credits: 3 }
      ],
      Spring: [
        { code: 'Major Elective', name: 'Major Elective', credits: 3 },
        { code: 'Technical Elective', name: 'Technical Elective', credits: 3 },
        { code: 'MEEN 498', name: 'Senior Design Project II', credits: 3 },
        { code: 'GENS 400', name: 'Enhancing Employability and Job Readiness', credits: 1 }
      ],
      Summer: [
        { code: 'ENGR 399', name: 'Engineering Internship', credits: 1 }
      ]
    }
  };



   ;


  let hoveredCode = null;
  let clickedCode = null;

  function renderApp() {
    const app = document.getElementById('app');
    // Clear
    while (app.firstChild) {
      app.removeChild(app.firstChild);
    }

    const yearGrid = document.createElement('div');
    yearGrid.className = "year-grid";
    app.appendChild(yearGrid);

    // MOUSEMOVE => handle hover if not locked
    yearGrid.addEventListener('mousemove', e => {
      if (!clickedCode) {
        const card = e.target.closest('.course-card');
        if (card) {
          const code = card.getAttribute('data-code');
          if (code !== hoveredCode) {
            hoveredCode = code;
            renderApp();
          }
        } else if (hoveredCode) {
          hoveredCode = null;
          renderApp();
        }
      }
    });

    // CLICK to toggle highlight or clear if empty space
    yearGrid.addEventListener('click', e => {
      const card = e.target.closest('.course-card');
      if (card) {
        const code = card.getAttribute('data-code');
        if (clickedCode === code) {
          clickedCode = null;
          hoveredCode = null;
        } else {
          clickedCode = code;
        }
      } else {
        // empty space => clear
        clickedCode = null;
        hoveredCode = null;
      }
      renderApp();
    });

    // Build each year block
    Object.entries(curriculum).forEach(([yearName, terms]) => {
      const yearDiv = document.createElement('div');
      yearDiv.className = "year-block";

      const yearTitle = document.createElement('h2');
      yearTitle.className = "year-title";
      yearTitle.textContent = yearName;
      yearDiv.appendChild(yearTitle);

      // terms
      Object.entries(terms).forEach(([termName, courses]) => {
        const termDiv = document.createElement('div');
        termDiv.className = "term";

        const termTitle = document.createElement('h3');
        termTitle.className = "term-title";
        termTitle.textContent = termName;
        termDiv.appendChild(termTitle);

        const courseGrid = document.createElement('div');
        courseGrid.className = "course-grid";

        courses.forEach(course => {
          const cardDiv = document.createElement('div');
          cardDiv.className = "course-card";
          cardDiv.setAttribute('data-code', course.code);

          const isSelected = getSelectedCode() === course.code;
          const highlightClasses = buildHighlightClasses(course.code, isSelected);
          highlightClasses.forEach(c => cardDiv.classList.add(c));

          // content
          const codeDiv = document.createElement('div');
          codeDiv.className = "course-code";
          codeDiv.textContent = course.code;

          const nameDiv = document.createElement('div');
          nameDiv.className = "course-name";
          nameDiv.textContent = course.name;

          const creditsDiv = document.createElement('div');
          creditsDiv.className = "course-credits";
          creditsDiv.textContent = course.credits + "cr";

          cardDiv.appendChild(codeDiv);
          cardDiv.appendChild(nameDiv);
          cardDiv.appendChild(creditsDiv);

          courseGrid.appendChild(cardDiv);
        });

        termDiv.appendChild(courseGrid);
        yearDiv.appendChild(termDiv);
      });

      yearGrid.appendChild(yearDiv);
    });

    updateLegend();
  }

  function getSelectedCode() {
    return clickedCode || hoveredCode;
  }

  function buildHighlightClasses(courseCode, isSelected) {
    const classes = ["bg-white"];
    const sc = getSelectedCode();
    if (sc) {
      if (isSelected) {
        classes.push("bg-indigo-100", "ring-2-indigo");
      } else if (courseRelationships[sc]) {
        const { prereqs, unlocks } = courseRelationships[sc];
        if (prereqs.includes(courseCode)) {
          classes.push("bg-amber-50", "ring-2-amber");
        } else if (unlocks.includes(courseCode)) {
          classes.push("bg-emerald-50", "ring-2-emerald");
        }
      }
    }
    return classes;
  }

    function updateLegend() {
      const legend = document.getElementById('legend');
      const legendSelected = document.getElementById('legend-selected');
      const prereqCountSpan = document.getElementById('prereq-count');
      const unlockCountSpan = document.getElementById('unlock-count');
      const sc = getSelectedCode();
      
      if (sc && courseRelationships[sc]) { // Ensure the course exists in relationships
        const prereqCount = courseRelationships[sc].prereqs.length;
        const unlockCount = courseRelationships[sc].unlocks.length;
        
        legendSelected.textContent = `Selected: ${sc}`;
        prereqCountSpan.textContent = prereqCount;
        unlockCountSpan.textContent = unlockCount;
      } else {
        legendSelected.textContent = "Hover over any course";
        prereqCountSpan.textContent = '0';
        unlockCountSpan.textContent = '0';
      }
      
      // Show the legend
      legend.classList.remove('hidden');
    }

  // Initial
  renderApp();
</script>