// *****************************************************************************
// Copyright (C) 2025 EclipseSource GmbH 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-only WITH Classpath-exception-2.0
// *****************************************************************************

import { injectable } from 'inversify';
import { Emitter, Event } from '../../common';
import { Widget } from '../widgets';

export interface Badge {
    value: number;
    tooltip: string;
}

@injectable()
export class BadgeService {
    protected readonly badges = new WeakMap<Widget, Badge>();
    protected readonly onDidChangeBadgesEmitter = new Emitter<Widget>();
    get onDidChangeBadges(): Event<Widget> { return this.onDidChangeBadgesEmitter.event; }

    getBadge(widget: Widget): Badge | undefined {
        return this.badges.get(widget);
    }

    showBadge(widget: Widget, badge?: Badge): void {
        if (badge) {
            this.badges.set(widget, badge);
            this.onDidChangeBadgesEmitter.fire(widget);
        } else if (this.badges.delete(widget)) {
            this.onDidChangeBadgesEmitter.fire(widget);
        }
    }
}