background.js

chrome.runtime.onInstalled.addListener(() => {
  console.log("UsageLens Installed");
});

function getToday() {
  return new Date().toISOString().split("T")[0];
}

function getMonth() {
  const d = new Date();
  return `${d.getFullYear()}-${d.getMonth() + 1}`;
}

function resetUsageIfNeeded() {
  chrome.storage.local.get(["usageData"], (res) => {
    const usageData = res.usageData || {};

    Object.keys(usageData).forEach((site) => {
      const item = usageData[site];

      if (item.lastDay !== getToday()) {
        item.daily = 0;
        item.lastDay = getToday();
      }

      if (item.lastMonth !== getMonth()) {
        item.monthly = 0;
        item.lastMonth = getMonth();
      }
    });

    chrome.storage.local.set({ usageData });
  });
}

/* every minute check reset */
chrome.alarms.create("usageResetCheck", {
  periodInMinutes: 1
});

chrome.alarms.onAlarm.addListener((alarm) => {
  if (alarm.name === "usageResetCheck") {
    resetUsageIfNeeded();
  }
});

/* optional badge count */
chrome.tabs.onActivated.addListener(() => {
  chrome.storage.local.get(["usageData"], (res) => {
    const data = res.usageData || {};
    let total = 0;

    Object.values(data).forEach((item) => {
      total += item.daily || 0;
    });

    chrome.action.setBadgeText({
      text: total ? String(total) : ""
    });

    chrome.action.setBadgeBackgroundColor({
      color: "#8b5cf6"
    });
  });
});