/*
 * Copyright (C) 2022 Savoir-faire Linux Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation; either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public
 * License along with this program.  If not, see
 * <https://www.gnu.org/licenses/>.
 */
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import { AxiosResponse } from 'axios';
import {
  ConversationInfos,
  IConversationMember,
  IConversationRequest,
  IConversationSummary,
  Message,
} from 'jami-web-common';
import { useCallback } from 'react';

import { useAuthContext } from '../contexts/AuthProvider';
import { ConversationMember } from '../models/conversation-member';
import { useAddToCache, useRemoveFromCache } from '../utils/reactquery';

export const useConversationInfosQuery = (conversationId?: string) => {
  const { axiosInstance } = useAuthContext();
  return useQuery(
    ['conversations', conversationId],
    async () => {
      const { data } = await axiosInstance.get<ConversationInfos>(`/conversations/${conversationId}/infos`);
      return data;
    },
    {
      enabled: !!conversationId,
    }
  );
};

export const useConversationsSummariesQuery = () => {
  const { axiosInstance } = useAuthContext();
  return useQuery(['conversations', 'summaries'], async () => {
    const { data } = await axiosInstance.get<IConversationSummary[]>(`/conversations`);
    return data;
  });
};

const checkConversationSummariesAreEqual = (
  conversationSummary1: IConversationSummary,
  conversationSummary2: IConversationSummary
) => conversationSummary1.id === conversationSummary2.id;
const checkIsConversationSummaryFn = (conversationSummary: IConversationSummary, conversationId: string) =>
  conversationSummary.id === conversationId;

export const useAddConversationSummaryToCache = () =>
  useAddToCache(['conversations', 'summaries'], checkConversationSummariesAreEqual);
export const useRemoveConversationSummaryFromCache = () =>
  useRemoveFromCache(['conversations', 'summaries'], checkIsConversationSummaryFn);

export const useRemoveConversationMutation = () => {
  const { axiosInstance } = useAuthContext();
  const removeConversationSummaryFromCache = useRemoveConversationSummaryFromCache();
  return useMutation(
    ({ conversationId }: { conversationId: string }) => axiosInstance.delete(`/conversations/${conversationId}`),
    {
      onSuccess: (_data, { conversationId }) => {
        removeConversationSummaryFromCache(conversationId);
      },
    }
  );
};

export const useRefreshConversationsSummaries = () => {
  const queryClient = useQueryClient();
  return useCallback(() => {
    queryClient.invalidateQueries(['conversations', 'summaries']);
  }, [queryClient]);
};

export const useMembersQuery = (conversationId?: string) => {
  const { axiosInstance } = useAuthContext();
  return useQuery(
    ['conversations', conversationId, 'members'],
    async () => {
      const { data } = await axiosInstance.get<IConversationMember[]>(`/conversations/${conversationId}/members`);
      return data.map((item) => ConversationMember.fromInterface(item));
    },
    {
      enabled: !!conversationId,
    }
  );
};

export const useMessagesQuery = (conversationId: string) => {
  const { axiosInstance } = useAuthContext();
  return useQuery(
    ['conversations', conversationId, 'messages'],
    async () => {
      const { data } = await axiosInstance.get<Message[]>(`/conversations/${conversationId}/messages`);
      return data;
    },
    {
      enabled: !!conversationId,
    }
  );
};

export const useSendMessageMutation = (conversationId: string) => {
  const { axiosInstance } = useAuthContext();
  const queryClient = useQueryClient();
  return useMutation(
    (message: string) => axiosInstance.post(`/conversations/${conversationId}/messages`, { message }),
    {
      onSuccess: () => queryClient.invalidateQueries(['messages', conversationId]),
    }
  );
};

const CheckConversationRequestsAreEqual = (
  conversationRequest1: IConversationRequest,
  conversationRequest2: IConversationRequest
) => conversationRequest1.conversationId === conversationRequest2.conversationId;
const checkIsConversationRequestFn = (conversationRequest: IConversationRequest, conversationId: string) =>
  conversationRequest.conversationId === conversationId;

export const useAddConversationRequestToCache = () =>
  useAddToCache(['conversationsRequests'], CheckConversationRequestsAreEqual);
export const useRemoveConversationRequestFromCache = () =>
  useRemoveFromCache(['conversationsRequests'], checkIsConversationRequestFn);

export const useConversationRequestsQuery = () => {
  const { axiosInstance } = useAuthContext();
  return useQuery({
    queryKey: ['conversationRequests'],
    queryFn: async () => {
      const { data } = await axiosInstance.get<IConversationRequest[]>('/conversation-requests/');
      return data;
    },
  });
};

export const useAcceptConversationRequestMutation = () => {
  const { axiosInstance } = useAuthContext();
  const addConversationSummaryToCache = useAddConversationSummaryToCache();
  const removeConversationRequestFromCache = useRemoveConversationRequestFromCache();
  return useMutation(
    async (variables: { conversationId: string }) => {
      const { data } = await axiosInstance.post<undefined, AxiosResponse<IConversationSummary>>(
        `/conversation-requests/${variables.conversationId}`
      );
      return data;
    },
    {
      onSuccess: (data, { conversationId }) => {
        addConversationSummaryToCache(data);
        removeConversationRequestFromCache(conversationId);
      },
    }
  );
};

export const useBlockConversationRequestMutation = () => {
  const { axiosInstance } = useAuthContext();
  const removeConversationRequestFromCache = useRemoveConversationRequestFromCache();
  return useMutation(
    ({ conversationId }: { conversationId: string }) =>
      axiosInstance.post(`/conversation-requests/${conversationId}/block`),
    {
      onSuccess: (_data, { conversationId }) => {
        removeConversationRequestFromCache(conversationId);
      },
    }
  );
};

export const useDeclineConversationRequestMutation = () => {
  const { axiosInstance } = useAuthContext();
  const removeConversationRequestFromCache = useRemoveConversationRequestFromCache();
  return useMutation(
    ({ conversationId }: { conversationId: string }) =>
      axiosInstance.delete(`/conversation-requests/${conversationId}`),
    {
      onSuccess: (_data, { conversationId }) => {
        removeConversationRequestFromCache(conversationId);
      },
    }
  );
};