openreplay/frontend/app/services/DashboardService.ts

import { IDashboard } from "App/mstore/types/dashboard";
import APIClient from 'App/api_client';
import { IWidget } from "App/mstore/types/widget";

export interface IDashboardService {
    initClient(client?: APIClient)
    getWidgets(dashboardId: string): Promise<any>
    
    getDashboards(): Promise<any[]>
    getDashboard(dashboardId: string): Promise<any>

    saveDashboard(dashboard: IDashboard): Promise<any>
    deleteDashboard(dashboardId: string): Promise<any>

    saveMetric(metric: IWidget, dashboardId?: string): Promise<any>

    addWidget(dashboard: IDashboard, metricIds: []): Promise<any>
    saveWidget(dashboardId: string, widget: IWidget): Promise<any>
    deleteWidget(dashboardId: string, widgetId: string): Promise<any>
    
}


export default class DashboardService implements IDashboardService {
    private client: APIClient;

    constructor(client?: APIClient) {
        this.client = client ? client : new APIClient();
    }

    initClient(client?: APIClient) {
        this.client = client || new APIClient();
    }

    /**
     * Get all widgets from a dashboard.
     * @param dashboardId Required
     * @returns 
     */
    getWidgets(dashboardId: string): Promise<any> {
        return this.client.get(`/dashboards/${dashboardId}/widgets`)
            .then(response => response.json())
            .then(response => response.data || []);
    }


    /**
     * Get all dashboards.
     * @returns {Promise<any>}
     */
    getDashboards(): Promise<any[]> {
        return this.client.get('/dashboards')
            .then(response => response.json())
            .then(response => response.data || []);
    }
    
    /**
     * Get a dashboard by dashboardId.
     * @param dashboardId 
     * @returns {Promise<any>}
     */
    getDashboard(dashboardId: string): Promise<any> {
        return this.client.get('/dashboards/' + dashboardId)
            .then(response => response.json())
            .then(response => response.data || {});
    }

    /**
     * Create or update a dashboard.
     * @param dashboard Required
     * @returns {Promise<any>}
     */
    saveDashboard(dashboard: IDashboard): Promise<any> {
        const data = dashboard.toJson();
        if (dashboard.dashboardId) {
            return this.client.put(`/dashboards/${dashboard.dashboardId}`, data)
                .then(response => response.json())
                .then(response => response.data || {});
        } else {
            return this.client.post('/dashboards', data)
                .then(response => response.json())
                .then(response => response.data || {});
        }
    }

    /**
     * Add a widget to a dashboard.
     * @param dashboard 
     * @param metricIds 
     * @returns 
     */
    addWidget(dashboard: IDashboard, metricIds: any): Promise<any> {
        const data = dashboard.toJson()
        data.metrics = metricIds
        return this.client.put(`/dashboards/${dashboard.dashboardId}`, data)
            .then(response => response.json())
            .then(response => response.data || {});
    }

    /**
     * Delete a dashboard.
     * @param dashboardId 
     * @returns {Promise<any>}
     */
    deleteDashboard(dashboardId: string): Promise<any> {
        return this.client.delete(`/dashboards/${dashboardId}`)
    }


    /**
     * Create a new Meitrc, if the dashboardId is not provided, 
     * it will add the metric to the dashboard.
     * @param metric Required
     * @param dashboardId Optional
     * @returns {Promise<any>}
     */
    saveMetric(metric: IWidget, dashboardId?: string): Promise<any> {
        const data = metric.toJson();
        const path = dashboardId ? `/dashboards/${dashboardId}/metrics` : '/metrics';
        if (metric.widgetId) {
            return this.client.put(path + '/' + metric.widgetId, data)
        } else {
            return this.client.post(path, data)
        }
    }

    /**
     * Remove a widget from a dashboard.
     * @param dashboardId Required
     * @param widgetId Required
     * @returns {Promise<any>}
     */
    deleteWidget(dashboardId: string, widgetId: string): Promise<any> {
        return this.client.delete(`/dashboards/${dashboardId}/widgets/${widgetId}`)
    }

    /**
     * Add a widget to a dashboard.
     * @param dashboardId Required
     * @param widget Required
     * @returns {Promise<any>}
     */
    saveWidget(dashboardId: string, widget: IWidget): Promise<any> {
        if (widget.widgetId) {
            return this.client.put(`/dashboards/${dashboardId}/widgets/${widget.widgetId}`, widget.toWidget())
                .then(response => response.json())
                .then(response => response.data || {});
        }
        return this.client.post(`/dashboards/${dashboardId}/widgets`, widget.toWidget())
            .then(response => response.json())
            .then(response => response.data || {});
    }
}