-module(logic_event_recurring_tests).
-include_lib("eunit/include/eunit.hrl").
-include("records.hrl").

-define(TABLES, [user, calendar, event, recurrence_exception]).

setup() ->
  eh_test_support:start_mnesia(),
  eh_test_support:ensure_tables(?TABLES),
  ok.

cleanup(_) ->
  eh_test_support:delete_tables(?TABLES),
  eh_test_support:stop_mnesia(),
  ok.

logic_event_recurring_test_() ->
  {foreach,
    fun setup/0,
    fun cleanup/1,
    [
      {"Create recurring event test", fun test_create_recurring_event/0},
      {"Create recurring event invalid RRULE test", fun test_create_recurring_invalid/0},
      {"Get occurrences test", fun test_get_occurrences/0},
      {"Cancel occurrence test", fun test_cancel_occurrence/0},
      {"Get occurrences with cancelled test", fun test_occurrences_with_cancelled/0},
      {"Materialize for booking test", fun test_materialize_for_booking/0}
    ]}.

create_test_user_and_calendar() ->
  UserId = base64:encode(crypto:strong_rand_bytes(16), #{mode => urlsafe, padding => false}),
  User = #user{
    id = UserId,
    email = <<"test@example.com">>,
    password_hash = <<"hash">>,
    role = user,
    status = active,
    created_at = calendar:universal_time(),
    updated_at = calendar:universal_time()
  },
  mnesia:dirty_write(User),

  {ok, Calendar} = logic_calendar:create_calendar(UserId, <<"Test Calendar">>, <<"">>, manual),
  {UserId, Calendar#calendar.id}.

add_days(DateTime, Days) ->
  Sec = calendar:datetime_to_gregorian_seconds(DateTime) + Days * 86400,
  calendar:gregorian_seconds_to_datetime(Sec).

test_create_recurring_event() ->
  {UserId, CalendarId} = create_test_user_and_calendar(),
  Title = <<"Weekly Meeting">>,
  StartTime = eh_test_support:future_start(),
  Duration = 60,
  RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},

  {ok, Event} = logic_event:create_recurring_event(UserId, CalendarId, Title, StartTime, Duration, RRule),

  ?assertEqual(CalendarId, Event#event.calendar_id),
  ?assertEqual(Title, Event#event.title),
  ?assertEqual(recurring, Event#event.event_type).

test_create_recurring_invalid() ->
  {UserId, CalendarId} = create_test_user_and_calendar(),
  Title = <<"Invalid">>,
  StartTime = eh_test_support:future_start(),
  Duration = 60,
  InvalidRRule = #{<<"freq">> => <<"YEARLY">>, <<"interval">> => 1},

  {error, invalid_rrule} = logic_event:create_recurring_event(
    UserId, CalendarId, Title, StartTime, Duration, InvalidRRule
  ).

test_get_occurrences() ->
  {UserId, CalendarId} = create_test_user_and_calendar(),
  StartTime = eh_test_support:future_start(),
  RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},

  {ok, Event} = logic_event:create_recurring_event(
    UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
  ),

  RangeEnd = add_days(StartTime, 28),
  {ok, Occurrences} = logic_event:get_occurrences(UserId, Event#event.id, RangeEnd),

  ?assertEqual(5, length(Occurrences)).

test_cancel_occurrence() ->
  {UserId, CalendarId} = create_test_user_and_calendar(),
  StartTime = eh_test_support:future_start(),
  RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},

  {ok, Event} = logic_event:create_recurring_event(
    UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
  ),

  CancelTime = add_days(StartTime, 7),
  {ok, cancelled} = logic_event:cancel_occurrence(UserId, Event#event.id, CancelTime).

test_occurrences_with_cancelled() ->
  {UserId, CalendarId} = create_test_user_and_calendar(),
  StartTime = eh_test_support:future_start(),
  RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},

  {ok, Event} = logic_event:create_recurring_event(
    UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
  ),

  CancelTime = add_days(StartTime, 7),
  {ok, cancelled} = logic_event:cancel_occurrence(UserId, Event#event.id, CancelTime),

  RangeEnd = add_days(StartTime, 28),
  {ok, Occurrences} = logic_event:get_occurrences(UserId, Event#event.id, RangeEnd),

  ?assertEqual(4, length(Occurrences)),

  Starts = [O || {virtual, O} <- Occurrences],
  ?assertNot(lists:member(CancelTime, Starts)).

test_materialize_for_booking() ->
  {UserId, CalendarId} = create_test_user_and_calendar(),
  StartTime = eh_test_support:future_start(),
  RRule = #{<<"freq">> => <<"WEEKLY">>, <<"interval">> => 1},

  {ok, Event} = logic_event:create_recurring_event(
    UserId, CalendarId, <<"Weekly">>, StartTime, 60, RRule
  ),

  OccurrenceStart = add_days(StartTime, 7),
  SpecialistId = <<"specialist123">>,

  {ok, Instance} = logic_event:materialize_for_booking(
    Event#event.id, OccurrenceStart, SpecialistId
  ),

  ?assertEqual(Event#event.id, Instance#event.master_id),
  ?assertEqual(OccurrenceStart, Instance#event.start_time),
  ?assertEqual(SpecialistId, Instance#event.specialist_id),
  ?assertEqual(true, Instance#event.is_instance).