vscode-gitlens/src/annotations/hoverBlameAnnotationProvider.ts

'use strict';
import { DecorationOptions, Range } from 'vscode';
import { FileAnnotationType } from './annotationController';
import { Annotations } from './annotations';
import { BlameAnnotationProviderBase } from './blameAnnotationProvider';
import { GitBlameCommit } from '../gitService';
import { Logger } from '../logger';

export class HoverBlameAnnotationProvider extends BlameAnnotationProviderBase {

    async provideAnnotation(shaOrLine?: string | number): Promise<boolean> {
        this.annotationType = FileAnnotationType.Hover;

        const cfg = this._config.annotations.file.hover;

        const blame = await this.getBlame(cfg.heatmap.enabled);
        if (blame === undefined) return false;

        if (cfg.heatmap.enabled) {
            const start = process.hrtime();

            const now = Date.now();
            const offset = this.uri.offset;
            const renderOptions = Annotations.hoverRenderOptions(this._config.theme, cfg.heatmap);

            const decorations: DecorationOptions[] = [];
            const decorationsMap: { [sha: string]: DecorationOptions } = Object.create(null);

            let commit: GitBlameCommit | undefined;
            let hover: DecorationOptions | undefined;

            for (const l of blame.lines) {
                const line = l.line + offset;

                hover = decorationsMap[l.sha];

                if (hover !== undefined) {
                    hover = {
                        ...hover,
                        range: new Range(line, 0, line, 0)
                    } as DecorationOptions;

                    decorations.push(hover);

                    continue;
                }

                commit = blame.commits.get(l.sha);
                if (commit === undefined) continue;

                hover = Annotations.hover(commit, renderOptions, now);
                hover.range = new Range(line, 0, line, 0);

                decorations.push(hover);
                decorationsMap[l.sha] = hover;

            }

            if (decorations.length) {
                this.editor.setDecorations(this.decoration!, decorations);
            }

            const duration = process.hrtime(start);
            Logger.log(`${(duration[0] * 1000) + Math.floor(duration[1] / 1000000)} ms to compute hover blame annotations`);
        }

        this.registerHoverProvider();
        this.selection(shaOrLine, blame);
        return true;
    }
}