// Local storage keys
export const LOCAL_STORAGE_KEYS = {
  LOGIN_METHOD: "openhands_login_method",
  ENTERPRISE_FORM_SAAS: "openhands_enterprise_form_saas",
  ENTERPRISE_FORM_SELF_HOSTED: "openhands_enterprise_form_self_hosted",
  SELECTED_ORG: "openhands_selected_org",
};

// Login methods
export enum LoginMethod {
  GITHUB = "github",
  GITLAB = "gitlab",
  BITBUCKET = "bitbucket",
  BITBUCKET_DATA_CENTER = "bitbucket_data_center",
  AZURE_DEVOPS = "azure_devops",
  ENTERPRISE_SSO = "enterprise_sso",
}

/**
 * Set the login method in local storage
 * @param method The login method (github, gitlab, bitbucket, or azure_devops)
 */
export const setLoginMethod = (method: LoginMethod): void => {
  localStorage.setItem(LOCAL_STORAGE_KEYS.LOGIN_METHOD, method);
};

/**
 * Get the login method from local storage
 * @returns The login method or null if not set
 */
export const getLoginMethod = (): LoginMethod | null => {
  const method = localStorage.getItem(LOCAL_STORAGE_KEYS.LOGIN_METHOD);
  return method as LoginMethod | null;
};

/**
 * Clear login method and last page from local storage
 */
export const clearLoginData = (): void => {
  localStorage.removeItem(LOCAL_STORAGE_KEYS.LOGIN_METHOD);
};

export const setSelectedOrg = (orgId: string): void => {
  localStorage.setItem(LOCAL_STORAGE_KEYS.SELECTED_ORG, orgId);
};

// CTA locations that can be dismissed
export type CTALocation = "homepage";

// Generate storage key for a CTA location
const getCTAKey = (location: CTALocation): string =>
  `${location}-cta-dismissed`;

/**
 * Set a CTA as dismissed in local storage (persists across tabs)
 * @param location The CTA location to dismiss
 */
export const setCTADismissed = (location: CTALocation): void => {
  localStorage.setItem(getCTAKey(location), "true");
};

/**
 * Check if a CTA has been dismissed
 * @param location The CTA location to check
 * @returns true if dismissed, false otherwise
 */
export const isCTADismissed = (location: CTALocation): boolean =>
  localStorage.getItem(getCTAKey(location)) === "true";

// Enterprise form data types
export type EnterpriseFormType = "saas" | "self-hosted";

export interface EnterpriseFormData {
  name: string;
  company: string;
  email: string;
  message: string;
}

const getEnterpriseFormKey = (formType: EnterpriseFormType): string =>
  formType === "saas"
    ? LOCAL_STORAGE_KEYS.ENTERPRISE_FORM_SAAS
    : LOCAL_STORAGE_KEYS.ENTERPRISE_FORM_SELF_HOSTED;

/**
 * Save enterprise form data to localStorage
 * @param formType The type of form (saas or self-hosted)
 * @param data The form data to save
 */
export const saveEnterpriseFormData = (
  formType: EnterpriseFormType,
  data: EnterpriseFormData,
): void => {
  localStorage.setItem(getEnterpriseFormKey(formType), JSON.stringify(data));
};

/**
 * Get enterprise form data from localStorage
 * @param formType The type of form (saas or self-hosted)
 * @returns The saved form data or null if not found
 */
export const getEnterpriseFormData = (
  formType: EnterpriseFormType,
): EnterpriseFormData | null => {
  const data = localStorage.getItem(getEnterpriseFormKey(formType));
  if (!data) return null;
  try {
    return JSON.parse(data) as EnterpriseFormData;
  } catch {
    return null;
  }
};

/**
 * Clear enterprise form data from localStorage
 * @param formType The type of form (saas or self-hosted)
 */
export const clearEnterpriseFormData = (formType: EnterpriseFormType): void => {
  localStorage.removeItem(getEnterpriseFormKey(formType));
};