import { IPushRule, PushRuleAction, PushRuleActionName, TweakName } from 'matrix-js-sdk';
import { useCallback, useMemo } from 'react';

export enum NotificationMode {
  OFF = 'OFF',
  Notify = 'Notify',
  NotifyLoud = 'NotifyLoud',
}

export const getNotificationMode = (actions: PushRuleAction[]): NotificationMode => {
  const soundTweak = actions.find(
    (action) => typeof action === 'object' && action.set_tweak === TweakName.Sound
  );
  const notify = actions.find(
    (action) => typeof action === 'string' && action === PushRuleActionName.Notify
  );

  if (notify && soundTweak) return NotificationMode.NotifyLoud;
  if (notify) return NotificationMode.Notify;
  return NotificationMode.OFF;
};

export type NotificationModeOptions = {
  soundValue?: string;
  highlight?: boolean;
};
export const getNotificationModeActions = (
  mode: NotificationMode,
  options?: NotificationModeOptions
): PushRuleAction[] => {
  if (mode === NotificationMode.OFF) return [];

  const actions: PushRuleAction[] = [PushRuleActionName.Notify];

  if (mode === NotificationMode.NotifyLoud) {
    actions.push({
      set_tweak: TweakName.Sound,
      value: options?.soundValue ?? 'default',
    });
  }

  if (options?.highlight) {
    actions.push({
      set_tweak: TweakName.Highlight,
      value: true,
    });
  }

  return actions;
};

export type GetNotificationModeCallback = (mode: NotificationMode) => PushRuleAction[];
export const useNotificationModeActions = (
  options?: NotificationModeOptions
): GetNotificationModeCallback => {
  const getAction: GetNotificationModeCallback = useCallback(
    (mode) => getNotificationModeActions(mode, options),
    [options]
  );

  return getAction;
};

export const useNotificationActionsMode = (actions: PushRuleAction[]): NotificationMode => {
  const mode: NotificationMode = useMemo(() => getNotificationMode(actions), [actions]);

  return mode;
};

// A rule the server has disabled is effectively OFF regardless of its stored
// actions — without this the UI reads loud actions on a disabled rule and
// claims "On" while nothing actually fires.
export const getNotificationModeFromRule = (pushRule: IPushRule): NotificationMode =>
  pushRule.enabled === false ? NotificationMode.OFF : getNotificationMode(pushRule.actions);

// The mode shown for a consolidated group of rules is its loudest member, so a
// partially-muted group doesn't misreport as fully off (mirrors Element).
export const loudestNotificationMode = (modes: NotificationMode[]): NotificationMode => {
  if (modes.includes(NotificationMode.NotifyLoud)) return NotificationMode.NotifyLoud;
  if (modes.includes(NotificationMode.Notify)) return NotificationMode.Notify;
  return NotificationMode.OFF;
};
