/********************************************************************************
 * Copyright (c) 2020-2023 EclipseSource and others.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the Eclipse
 * Public License v. 2.0 are satisfied: GNU General Public License, version 2
 * with the GNU Classpath Exception which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 ********************************************************************************/
import { BaseJsonrpcGLSPClient, ClientState, JsonrpcGLSPClient } from '@eclipse-glsp/client';
import { MessageService } from '@theia/core/';
import { Message } from 'vscode-jsonrpc';

export class TheiaJsonrpcGLSPClient extends BaseJsonrpcGLSPClient {
    protected messageService: MessageService;

    constructor(options: TheiaJsonrpcGLSPClient.Options) {
        super(options);
        this.messageService = options.messageService;
    }

    protected override handleConnectionError(error: Error, message: Message, count: number): void {
        super.handleConnectionError(error, message, count);
        this.messageService.error(`Connection the ${this.id} glsp server is erroring. Shutting down server.`);
    }

    protected override handleConnectionClosed(): void {
        if (this.state !== ClientState.Stopping && this.state !== ClientState.Stopped) {
            this.messageService.error(`Connection to the ${this.id} glsp server got closed. Server will not be restarted.`);
        }
        super.handleConnectionClosed();
    }
}

// eslint-disable-next-line no-redeclare
export namespace TheiaJsonrpcGLSPClient {
    export interface Options extends JsonrpcGLSPClient.Options {
        messageService: MessageService;
    }

    export function isOptions(object: any): object is Options {
        return JsonrpcGLSPClient.isOptions(object) && 'messageService' in object;
    }
}