fake-api-token-store.ts

import type { IApiTokenStore } from '../../lib/types/stores/api-token-store';
import type {
    ApiTokenType,
    IApiToken,
    IApiTokenCreate,
} from '../../lib/types/models/api-token';

import EventEmitter from 'events';

export default class FakeApiTokenStore
    extends EventEmitter
    implements IApiTokenStore
{
    countByType(): Promise<Map<ApiTokenType, number>> {
        return Promise.resolve(new Map());
    }
    tokens: IApiToken[] = [];

    async delete(key: string): Promise<void> {
        this.tokens.splice(
            this.tokens.findIndex((t) => t.secret === key),
            1,
        );
    }

    async count(): Promise<number> {
        return this.tokens.length;
    }

    async deleteAll(): Promise<void> {
        this.tokens = [];
    }

    destroy(): void {}

    async exists(key: string): Promise<boolean> {
        return this.tokens.some((token) => token.secret === key);
    }

    async get(key: string): Promise<IApiToken> {
        // get can return undefined. See api-token-store.e2e.test.ts
        return this.tokens.find((t) => t.secret === key);
    }

    async getAll(): Promise<IApiToken[]> {
        return this.tokens;
    }

    async getAllActive(): Promise<IApiToken[]> {
        return this.tokens.filter(
            (token) => token.expiresAt === null || token.expiresAt > new Date(),
        );
    }

    async insert(newToken: IApiTokenCreate): Promise<IApiToken> {
        const apiToken = {
            createdAt: new Date(),
            project: newToken.projects?.join(',') || '*',
            alias: null,
            ...newToken,
        };
        this.tokens.push(apiToken);
        this.emit('insert');
        return apiToken;
    }

    async markSeenAt(secrets: string[]): Promise<void> {
        this.tokens
            .filter((t) => secrets.includes(t.secret))
            .forEach((t) => {
                // eslint-disable-next-line no-param-reassign
                t.seenAt = new Date();
            });
    }

    async setExpiry(secret: string, expiresAt: Date): Promise<IApiToken> {
        const t = await this.get(secret);
        t.expiresAt = expiresAt;
        return t;
    }
}