// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

import {ActionType} from '@constants';
import DatabaseManager from '@database/manager';
import NetworkManager from '@managers/network_manager';
import websocketManager from '@managers/websocket_manager';
import {getIsScheduledPostEnabled} from '@queries/servers/scheduled_post';
import {getCurrentUserId} from '@queries/servers/system';
import ScheduledPostModel from '@typings/database/models/servers/scheduled_post';
import {getFullErrorMessage} from '@utils/errors';
import {logError} from '@utils/log';
import {isScheduledPostModel} from '@utils/scheduled_post';

import {forceLogoutIfNecessary} from './session';

import type {CreateResponse} from '@hooks/handle_send_message';

export async function createScheduledPost(serverUrl: string, schedulePost: ScheduledPost): Promise<CreateResponse> {
    try {
        const operator = DatabaseManager.getServerDatabaseAndOperator(serverUrl).operator;

        const connectionId = websocketManager.getClient(serverUrl)?.getConnectionId();
        const client = NetworkManager.getClient(serverUrl);
        const response = await client.createScheduledPost(schedulePost, connectionId);

        if (response) {
            await operator.handleScheduledPosts({
                actionType: ActionType.SCHEDULED_POSTS.CREATE_OR_UPDATED_SCHEDULED_POST,
                scheduledPosts: [response],
                prepareRecordsOnly: false,
            });
        }

        return {data: true, response};
    } catch (error) {
        logError('error on createScheduledPost', getFullErrorMessage(error));
        forceLogoutIfNecessary(serverUrl, error);
        return {error: getFullErrorMessage(error)};
    }
}

export async function updateScheduledPost(serverUrl: string, scheduledPost: ScheduledPost | ScheduledPostModel, updateTime: number, fetchOnly = false) {
    try {
        const {operator, database} = DatabaseManager.getServerDatabaseAndOperator(serverUrl);
        const client = NetworkManager.getClient(serverUrl);
        const currentUserId = await getCurrentUserId(database);
        const normalizedScheduledPost = isScheduledPostModel(scheduledPost) ? scheduledPost.toApi(currentUserId) : scheduledPost;
        normalizedScheduledPost.scheduled_at = updateTime;
        const connectionId = websocketManager.getClient(serverUrl)?.getConnectionId();
        const response = await client.updateScheduledPost(normalizedScheduledPost, connectionId);

        if (response && !fetchOnly) {
            await operator.handleScheduledPosts({
                actionType: ActionType.SCHEDULED_POSTS.CREATE_OR_UPDATED_SCHEDULED_POST,
                scheduledPosts: [response],
                prepareRecordsOnly: false,
            });
        }

        return {scheduledPost: response};
    } catch (error) {
        logError('error on updateScheduledPost', getFullErrorMessage(error));
        forceLogoutIfNecessary(serverUrl, error);
        return {error: getFullErrorMessage(error)};
    }
}

export async function fetchScheduledPosts(serverUrl: string, teamId: string, includeDirectChannels = false, groupLabel?: RequestGroupLabel) {
    try {
        const {operator, database} = DatabaseManager.getServerDatabaseAndOperator(serverUrl);

        const client = NetworkManager.getClient(serverUrl);

        const scheduledPostEnabled = await getIsScheduledPostEnabled(database);
        if (!scheduledPostEnabled) {
            return {scheduledPosts: []};
        }

        const scheduledPostsResponse = await client.getScheduledPostsForTeam(teamId, includeDirectChannels, groupLabel);
        const {directChannels = [], ...scheduledPostsByTeam} = scheduledPostsResponse;
        const scheduledPosts = [...Object.values(scheduledPostsByTeam).flat(), ...directChannels];
        await operator.handleScheduledPosts({
            actionType: ActionType.SCHEDULED_POSTS.RECEIVED_ALL_SCHEDULED_POSTS,
            scheduledPosts,
            prepareRecordsOnly: false,
            includeDirectChannelPosts: includeDirectChannels,
        });

        return {scheduledPosts};
    } catch (error) {
        logError('error on fetchScheduledPosts', getFullErrorMessage(error));
        forceLogoutIfNecessary(serverUrl, error);
        return {error};
    }
}

export async function deleteScheduledPost(serverUrl: string, scheduledPostId: string) {
    try {
        const {operator} = DatabaseManager.getServerDatabaseAndOperator(serverUrl);

        const client = NetworkManager.getClient(serverUrl);
        const connectionId = websocketManager.getClient(serverUrl)?.getConnectionId();

        const result = await client.deleteScheduledPost(scheduledPostId, connectionId);

        if (result) {
            await operator.handleScheduledPosts({
                actionType: ActionType.SCHEDULED_POSTS.DELETE_SCHEDULED_POST,
                scheduledPosts: [result],
                prepareRecordsOnly: false,
            });
        }

        return {scheduledPost: result};
    } catch (error) {
        logError('error on deleteScheduledPost', getFullErrorMessage(error));
        forceLogoutIfNecessary(serverUrl, error);
        return {error};
    }
}