<script>
    const courseRelationships = {
    '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 242', 'MATH 211', 'CIVE 370']
    },
    'CHEM 115': {
        prereqs: [],
        unlocks: ['CIVE 370', 'CIVE 336']
    },
    'GENS 101': {
        prereqs: [],
        unlocks: ['CIVE 332']
    },
    'GENS 100': {
        prereqs: [],
        unlocks: ['GENS 300']
    },
    'PHYS 121': {
        prereqs: ['MATH 111'],
        unlocks: ['CIVE 200', 'CIVE 335']
    },
    'COSC 114': {
        prereqs: [],
        unlocks: ['COSC 202', 'CIVE 180']
    },
    'HUMA XXX': {
        prereqs: [],
        unlocks: []
    },
    'MATH 211': {
        prereqs: ['MATH 112'],
        unlocks: ['MATH 251']
    },
    'MATH 251': {
        prereqs: ['MATH 211'],
        unlocks: []
    },
    'COSC 202': {
        prereqs: ['COSC 114'],
        unlocks: []
    },
    'MATH 242': {
        prereqs: ['MATH 112'],
        unlocks: []
    },
    'BUSS XXX': {
        prereqs: [],
        unlocks: []
    },
    'Technical Elective': {
        prereqs: [],
        unlocks: []
    },
    'GENS 300': {
        prereqs: ['GENS 100'],
        unlocks: ['ENGR 399']
    },
    'GENS 400': {
        prereqs: ['GENS 300'],
        unlocks: []
    },
    'BUSS 322': {
        prereqs: [],
        unlocks: []
    },
    'ENGR 399': {
        prereqs: ['GENS 300'],
        unlocks: []
    },
    'CIVE 180': {
        prereqs: ['COSC 114'],
        unlocks: ['CIVE 310']
    },
    'CIVE 200': {
        prereqs: ['PHYS 121'],
        unlocks: ['CIVE 225']
    },
    'CIVE 225': {
        prereqs: ['CIVE 200'],
        unlocks: ['CIVE 336', 'CIVE 338', 'CIVE 340']
    },
    'CIVE 310': {
        prereqs: ['CIVE 180'],
        unlocks: ['CIVE 380']
    },
    'CIVE 332': {
        prereqs: ['GENS 101'],
        unlocks: ['CIVE 497']
    },
    'CIVE 336': {
        prereqs: ['CHEM 115', 'CIVE 225'],
        unlocks: ['CIVE 341']
    },
    'CIVE 338': {
        prereqs: ['CIVE 225'],
        unlocks: ['CIVE 470', 'CIVE 497']
    },
    'CIVE 340': {
        prereqs: ['CIVE 225'],
        unlocks: ['CIVE 341']
    },
    'CIVE 370': {
        prereqs: ['CHEM 115', 'MATH 112'],
        unlocks: ['CIVE 497']
    },
    'CIVE 335': {
        prereqs: ['PHYS 121', 'MATH 231'],
        unlocks: ['CIVE 497']
    },
    'CIVE 341': {
        prereqs: ['CIVE 336', 'CIVE 340'],
        unlocks: ['CIVE 497']
    },
    'CIVE 380': {
        prereqs: ['CIVE 310'],
        unlocks: ['CIVE 497']
    },
    'CIVE 442': {
        prereqs: ['CIVE 336'],
        unlocks: []
    },
    'CIVE 497': {
        prereqs: ['CIVE 332', 'CIVE 335', 'CIVE 338', 'CIVE 341', 'CIVE 370', 'CIVE 380'],
        unlocks: ['CIVE 498']
    },
    'CIVE 470': {
        prereqs: ['CIVE 338'],
        unlocks: []
    },
    'CIVE 498': {
        prereqs: ['CIVE 497'],
        unlocks: []
    },
    'Science Elective': {
        prereqs: [],
        unlocks: []
    }
    };

  
  // courses
  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: 'Computing with Python', credits: 4 }
      ],
      Summer: [
        { code: 'HUMA XXX', name: 'Arabic Language Elective', credits: 3 },
        { code: 'HUMA XXX', name: 'UAE Culture Elective', credits: 3 }
      ]
    },
  
    Sophomore: {
      Fall: [
        { code: 'MATH 211', name: 'Differential Equation & Linear Algebra', credits: 4 },
        { code: 'COSC 202', name: 'Data Science & AI', credits: 3 },
        { code: 'CIVE 200', name: 'Statics', credits: 3 },
        { code: 'CIVE 180', name: 'Engineering Graphics & Visualization', credits: 3 },
        { code: 'Science Elective', name: 'Science Elective', credits: 3 }
      ],
      Spring: [
        { code: 'MATH 251', name: 'Operations Research I', credits: 4 },
        { code: 'MATH 242', name: 'Intro to Probability & Statistics', credits: 3 },
        { code: 'CIVE 225', name: 'Mechanics of Solids', credits: 4 },
        { code: 'CIVE 310', name: 'Geomatics', credits: 3 }
      ],
      Summer: [
        { code: 'BUSS XXX', name: 'Business Studies Elective', credits: 3 }
      ]
    },
  
    Junior: {
      Fall: [
        { code: 'CIVE 332', name: 'Construction Engnrg & Management', credits: 3 },
        { code: 'CIVE 336', name: 'Civil Engineering Materials', credits: 4 },
        { code: 'CIVE 340', name: 'Behavior & Analysis of Structures', credits: 3 },
        { code: 'CIVE 370', name: 'Environmental Engineering', credits: 4 },
        { code: 'GENS 300', name: 'Career Preparation', credits: 1 }
      ],
      Spring: [
        { code: 'CIVE 335', name: 'Fluid Mechanics', credits: 4 },
        { code: 'CIVE 338', name: 'Geotechnical Engineering', credits: 4 },
        { code: 'CIVE 341', name: 'Design of Steel Structures', credits: 3 },
        { code: 'CIVE 380', name: 'Transportation Engineering', credits: 3 },
        { code: 'BUSS 322', name: 'Fundamentals of Innovation and Entrepreneurship', credits: 3 }
      ],
      Summer: [
        { code: 'ENGR 399', name: 'Engineering Internship', credits: 1 }
      ]
    },
  
    Senior: {
      Fall: [
        { code: 'CIVE 442', name: 'Design of Concrete Structures', credits: 3 },
        { code: 'CIVE 497', name: 'Senior Design Project I', credits: 3 },
        { code: 'Technical Elective', name: 'Technical Elective', credits: 3 },
        { code: 'Technical Elective', name: 'Technical Elective', credits: 3 },
        { code: 'GENS 400', name: 'Enhancing Employability and Job Readiness', credits: 1 }
      ],
      Spring: [
        { code: 'CIVE 470', name: 'Foundation Engineering', credits: 4 },
        { code: 'CIVE 498', name: 'Senior Design Project II', credits: 3 },
        { code: 'Technical Elective', name: 'Technical Elective', credits: 3 },
        { code: 'Technical Elective', name: 'Technical Elective', credits: 3 }
      ],
      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>
