mirror of
				https://github.com/Unleash/unleash.git
				synced 2025-10-27 11:02:16 +01:00 
			
		
		
		
	
							parent
							
								
									44bcc49062
								
							
						
					
					
						commit
						3c01813826
					
				@ -85,7 +85,6 @@ exports[`should create default config 1`] = `
 | 
			
		||||
  },
 | 
			
		||||
  "inlineSegmentConstraints": true,
 | 
			
		||||
  "isEnterprise": false,
 | 
			
		||||
  "isOss": false,
 | 
			
		||||
  "listen": {
 | 
			
		||||
    "host": undefined,
 | 
			
		||||
    "port": 4242,
 | 
			
		||||
 | 
			
		||||
@ -620,10 +620,6 @@ export function createConfig(options: IUnleashOptions): IUnleashConfig {
 | 
			
		||||
        Boolean(options.enterpriseVersion) &&
 | 
			
		||||
        ui.environment?.toLowerCase() !== 'pro';
 | 
			
		||||
 | 
			
		||||
    const isTest = process.env.NODE_ENV === 'test';
 | 
			
		||||
    const isOss = isTest
 | 
			
		||||
        ? options.isOss || false
 | 
			
		||||
        : !isEnterprise && ui.environment !== 'pro';
 | 
			
		||||
    const metricsRateLimiting = loadMetricsRateLimitingConfig(options);
 | 
			
		||||
 | 
			
		||||
    const rateLimiting = loadRateLimitingConfig(options);
 | 
			
		||||
@ -764,7 +760,6 @@ export function createConfig(options: IUnleashOptions): IUnleashConfig {
 | 
			
		||||
        publicFolder: options.publicFolder,
 | 
			
		||||
        disableScheduler: options.disableScheduler,
 | 
			
		||||
        isEnterprise: isEnterprise,
 | 
			
		||||
        isOss: isOss,
 | 
			
		||||
        metricsRateLimiting,
 | 
			
		||||
        rateLimiting,
 | 
			
		||||
        feedbackUriPath,
 | 
			
		||||
 | 
			
		||||
@ -3,14 +3,13 @@ import type {
 | 
			
		||||
    FeatureEnvironmentKey,
 | 
			
		||||
    IFeatureEnvironmentStore,
 | 
			
		||||
} from '../types/stores/feature-environment-store';
 | 
			
		||||
import type { Logger } from '../logger';
 | 
			
		||||
import type { Logger, LogProvider } from '../logger';
 | 
			
		||||
import metricsHelper from '../util/metrics-helper';
 | 
			
		||||
import { DB_TIME } from '../metric-events';
 | 
			
		||||
import type { IFeatureEnvironment, IVariant } from '../types/model';
 | 
			
		||||
import NotFoundError from '../error/notfound-error';
 | 
			
		||||
import { v4 as uuidv4 } from 'uuid';
 | 
			
		||||
import type { Db } from './db';
 | 
			
		||||
import type { IUnleashConfig } from '../types';
 | 
			
		||||
 | 
			
		||||
const T = {
 | 
			
		||||
    featureEnvs: 'feature_environments',
 | 
			
		||||
@ -37,12 +36,7 @@ export class FeatureEnvironmentStore implements IFeatureEnvironmentStore {
 | 
			
		||||
 | 
			
		||||
    private readonly timer: Function;
 | 
			
		||||
 | 
			
		||||
    private readonly isOss: boolean;
 | 
			
		||||
    constructor(
 | 
			
		||||
        db: Db,
 | 
			
		||||
        eventBus: EventEmitter,
 | 
			
		||||
        { getLogger, isOss }: Pick<IUnleashConfig, 'getLogger' | 'isOss'>,
 | 
			
		||||
    ) {
 | 
			
		||||
    constructor(db: Db, eventBus: EventEmitter, getLogger: LogProvider) {
 | 
			
		||||
        this.db = db;
 | 
			
		||||
        this.logger = getLogger('feature-environment-store.ts');
 | 
			
		||||
        this.timer = (action) =>
 | 
			
		||||
@ -50,7 +44,6 @@ export class FeatureEnvironmentStore implements IFeatureEnvironmentStore {
 | 
			
		||||
                store: 'feature-environments',
 | 
			
		||||
                action,
 | 
			
		||||
            });
 | 
			
		||||
        this.isOss = isOss;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    async delete({
 | 
			
		||||
@ -103,30 +96,11 @@ export class FeatureEnvironmentStore implements IFeatureEnvironmentStore {
 | 
			
		||||
        );
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    addOssFilterIfNeeded(queryBuilder) {
 | 
			
		||||
        if (this.isOss) {
 | 
			
		||||
            return queryBuilder
 | 
			
		||||
                .join(
 | 
			
		||||
                    'environments',
 | 
			
		||||
                    'environments.name',
 | 
			
		||||
                    '=',
 | 
			
		||||
                    `${T.featureEnvs}.environment`,
 | 
			
		||||
                )
 | 
			
		||||
                .whereIn('environments.name', [
 | 
			
		||||
                    'default',
 | 
			
		||||
                    'development',
 | 
			
		||||
                    'production',
 | 
			
		||||
                ]);
 | 
			
		||||
        }
 | 
			
		||||
        return queryBuilder;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    async getAll(query?: Object): Promise<IFeatureEnvironment[]> {
 | 
			
		||||
        let rows = this.db(T.featureEnvs);
 | 
			
		||||
        if (query) {
 | 
			
		||||
            rows = rows.where(query);
 | 
			
		||||
        }
 | 
			
		||||
        this.addOssFilterIfNeeded(rows);
 | 
			
		||||
        return (await rows).map((r) => ({
 | 
			
		||||
            enabled: r.enabled,
 | 
			
		||||
            featureName: r.feature_name,
 | 
			
		||||
@ -145,7 +119,6 @@ export class FeatureEnvironmentStore implements IFeatureEnvironmentStore {
 | 
			
		||||
        if (environment) {
 | 
			
		||||
            rows = rows.where({ environment });
 | 
			
		||||
        }
 | 
			
		||||
        this.addOssFilterIfNeeded(rows);
 | 
			
		||||
        return (await rows).map((r) => ({
 | 
			
		||||
            enabled: r.enabled,
 | 
			
		||||
            featureName: r.feature_name,
 | 
			
		||||
 | 
			
		||||
@ -94,7 +94,12 @@ export const createStores = (
 | 
			
		||||
        settingStore: new SettingStore(db, getLogger),
 | 
			
		||||
        userStore: new UserStore(db, getLogger, config.flagResolver),
 | 
			
		||||
        accountStore: new AccountStore(db, getLogger),
 | 
			
		||||
        projectStore: new ProjectStore(db, eventBus, config),
 | 
			
		||||
        projectStore: new ProjectStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            getLogger,
 | 
			
		||||
            config.flagResolver,
 | 
			
		||||
        ),
 | 
			
		||||
        tagStore: new TagStore(db, eventBus, getLogger),
 | 
			
		||||
        tagTypeStore: new TagTypeStore(db, eventBus, getLogger),
 | 
			
		||||
        addonStore: new AddonStore(db, eventBus, getLogger),
 | 
			
		||||
@ -117,14 +122,15 @@ export const createStores = (
 | 
			
		||||
        clientFeatureToggleStore: new FeatureToggleClientStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            config,
 | 
			
		||||
            getLogger,
 | 
			
		||||
            config.flagResolver,
 | 
			
		||||
        ),
 | 
			
		||||
        environmentStore: new EnvironmentStore(db, eventBus, config),
 | 
			
		||||
        environmentStore: new EnvironmentStore(db, eventBus, getLogger),
 | 
			
		||||
        featureTagStore: new FeatureTagStore(db, eventBus, getLogger),
 | 
			
		||||
        featureEnvironmentStore: new FeatureEnvironmentStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            config,
 | 
			
		||||
            getLogger,
 | 
			
		||||
        ),
 | 
			
		||||
        userSplashStore: new UserSplashStore(db, eventBus, getLogger),
 | 
			
		||||
        roleStore: new RoleStore(db, eventBus, getLogger),
 | 
			
		||||
 | 
			
		||||
@ -25,7 +25,7 @@ export const createAccessService = (
 | 
			
		||||
    const groupStore = new GroupStore(db);
 | 
			
		||||
    const accountStore = new AccountStore(db, getLogger);
 | 
			
		||||
    const roleStore = new RoleStore(db, eventBus, getLogger);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, config);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, getLogger);
 | 
			
		||||
    const accessStore = new AccessStore(db, eventBus, getLogger);
 | 
			
		||||
    const eventService = createEventsService(db, config);
 | 
			
		||||
    const groupService = new GroupService(
 | 
			
		||||
 | 
			
		||||
@ -21,7 +21,7 @@ export const createApiTokenService = (
 | 
			
		||||
        getLogger,
 | 
			
		||||
        config.flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, config);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, getLogger);
 | 
			
		||||
    const eventService = createEventsService(db, config);
 | 
			
		||||
 | 
			
		||||
    return new ApiTokenService(
 | 
			
		||||
 | 
			
		||||
@ -1,7 +1,7 @@
 | 
			
		||||
import { Knex } from 'knex';
 | 
			
		||||
import metricsHelper from '../../util/metrics-helper';
 | 
			
		||||
import { DB_TIME } from '../../metric-events';
 | 
			
		||||
import type { Logger } from '../../logger';
 | 
			
		||||
import type { Logger, LogProvider } from '../../logger';
 | 
			
		||||
import type {
 | 
			
		||||
    IFeatureToggleClient,
 | 
			
		||||
    IFeatureToggleClientStore,
 | 
			
		||||
@ -9,7 +9,6 @@ import type {
 | 
			
		||||
    IFlagResolver,
 | 
			
		||||
    IStrategyConfig,
 | 
			
		||||
    ITag,
 | 
			
		||||
    IUnleashConfig,
 | 
			
		||||
    PartialDeep,
 | 
			
		||||
} from '../../types';
 | 
			
		||||
import {
 | 
			
		||||
@ -47,16 +46,11 @@ export default class FeatureToggleClientStore
 | 
			
		||||
 | 
			
		||||
    private flagResolver: IFlagResolver;
 | 
			
		||||
 | 
			
		||||
    private readonly isOss: boolean;
 | 
			
		||||
 | 
			
		||||
    constructor(
 | 
			
		||||
        db: Db,
 | 
			
		||||
        eventBus: EventEmitter,
 | 
			
		||||
        {
 | 
			
		||||
            getLogger,
 | 
			
		||||
            flagResolver,
 | 
			
		||||
            isOss,
 | 
			
		||||
        }: Pick<IUnleashConfig, 'getLogger' | 'flagResolver' | 'isOss'>,
 | 
			
		||||
        getLogger: LogProvider,
 | 
			
		||||
        flagResolver: IFlagResolver,
 | 
			
		||||
    ) {
 | 
			
		||||
        this.db = db;
 | 
			
		||||
        this.logger = getLogger('feature-toggle-client-store.ts');
 | 
			
		||||
@ -66,7 +60,6 @@ export default class FeatureToggleClientStore
 | 
			
		||||
                action,
 | 
			
		||||
            });
 | 
			
		||||
        this.flagResolver = flagResolver;
 | 
			
		||||
        this.isOss = isOss;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private async getAll({
 | 
			
		||||
@ -79,6 +72,7 @@ export default class FeatureToggleClientStore
 | 
			
		||||
        const isPlayground = requestType === 'playground';
 | 
			
		||||
        const environment = featureQuery?.environment || DEFAULT_ENV;
 | 
			
		||||
        const stopTimer = this.timer(`getAllBy${requestType}`);
 | 
			
		||||
 | 
			
		||||
        let selectColumns = [
 | 
			
		||||
            'features.name as name',
 | 
			
		||||
            'features.description as description',
 | 
			
		||||
@ -109,10 +103,6 @@ export default class FeatureToggleClientStore
 | 
			
		||||
 | 
			
		||||
        let query = this.db('features')
 | 
			
		||||
            .modify(FeatureToggleStore.filterByArchived, archived)
 | 
			
		||||
            .modify(
 | 
			
		||||
                FeatureToggleStore.filterByProjectsAccessibleByOss,
 | 
			
		||||
                this.isOss,
 | 
			
		||||
            )
 | 
			
		||||
            .leftJoin(
 | 
			
		||||
                this.db('feature_strategies')
 | 
			
		||||
                    .select('*')
 | 
			
		||||
 | 
			
		||||
@ -10,10 +10,13 @@ export const createClientFeatureToggleService = (
 | 
			
		||||
    db: Db,
 | 
			
		||||
    config: IUnleashConfig,
 | 
			
		||||
): ClientFeatureToggleService => {
 | 
			
		||||
    const { getLogger, eventBus, flagResolver } = config;
 | 
			
		||||
 | 
			
		||||
    const featureToggleClientStore = new FeatureToggleClientStore(
 | 
			
		||||
        db,
 | 
			
		||||
        config.eventBus,
 | 
			
		||||
        config,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        getLogger,
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    const segmentReadModel = new SegmentReadModel(db);
 | 
			
		||||
@ -23,7 +26,7 @@ export const createClientFeatureToggleService = (
 | 
			
		||||
            clientFeatureToggleStore: featureToggleClientStore,
 | 
			
		||||
        },
 | 
			
		||||
        segmentReadModel,
 | 
			
		||||
        config,
 | 
			
		||||
        { getLogger, flagResolver },
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    return clientFeatureToggleService;
 | 
			
		||||
@ -32,6 +35,8 @@ export const createClientFeatureToggleService = (
 | 
			
		||||
export const createFakeClientFeatureToggleService = (
 | 
			
		||||
    config: IUnleashConfig,
 | 
			
		||||
): ClientFeatureToggleService => {
 | 
			
		||||
    const { getLogger, flagResolver } = config;
 | 
			
		||||
 | 
			
		||||
    const fakeClientFeatureToggleStore = new FakeClientFeatureToggleStore();
 | 
			
		||||
 | 
			
		||||
    const fakeSegmentReadModel = new FakeSegmentReadModel();
 | 
			
		||||
@ -41,7 +46,7 @@ export const createFakeClientFeatureToggleService = (
 | 
			
		||||
            clientFeatureToggleStore: fakeClientFeatureToggleStore,
 | 
			
		||||
        },
 | 
			
		||||
        fakeSegmentReadModel,
 | 
			
		||||
        config,
 | 
			
		||||
        { getLogger, flagResolver },
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    return clientFeatureToggleService;
 | 
			
		||||
 | 
			
		||||
@ -150,7 +150,12 @@ export const deferredExportImportTogglesService = (
 | 
			
		||||
        );
 | 
			
		||||
        const tagStore = new TagStore(db, eventBus, getLogger);
 | 
			
		||||
        const tagTypeStore = new TagTypeStore(db, eventBus, getLogger);
 | 
			
		||||
        const projectStore = new ProjectStore(db, eventBus, config);
 | 
			
		||||
        const projectStore = new ProjectStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            getLogger,
 | 
			
		||||
            flagResolver,
 | 
			
		||||
        );
 | 
			
		||||
        const featureTagStore = new FeatureTagStore(db, eventBus, getLogger);
 | 
			
		||||
        const strategyStore = new StrategyStore(db, getLogger);
 | 
			
		||||
        const contextFieldStore = new ContextFieldStore(
 | 
			
		||||
@ -167,7 +172,7 @@ export const deferredExportImportTogglesService = (
 | 
			
		||||
        const featureEnvironmentStore = new FeatureEnvironmentStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            config,
 | 
			
		||||
            getLogger,
 | 
			
		||||
        );
 | 
			
		||||
        const eventStore = new EventStore(db, getLogger);
 | 
			
		||||
        const accessService = createAccessService(db, config);
 | 
			
		||||
 | 
			
		||||
@ -19,11 +19,11 @@ export const createFeatureLifecycleService =
 | 
			
		||||
        const { eventBus, getLogger } = config;
 | 
			
		||||
        const eventStore = new EventStore(db, getLogger);
 | 
			
		||||
        const featureLifecycleStore = new FeatureLifecycleStore(db);
 | 
			
		||||
        const environmentStore = new EnvironmentStore(db, eventBus, config);
 | 
			
		||||
        const environmentStore = new EnvironmentStore(db, eventBus, getLogger);
 | 
			
		||||
        const featureEnvironmentStore = new FeatureEnvironmentStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            config,
 | 
			
		||||
            getLogger,
 | 
			
		||||
        );
 | 
			
		||||
        const eventService = createEventsService(db, config);
 | 
			
		||||
        const featureLifecycleService = new FeatureLifecycleService(
 | 
			
		||||
 | 
			
		||||
@ -80,13 +80,19 @@ export const createFeatureToggleService = (
 | 
			
		||||
    const featureToggleClientStore = new FeatureToggleClientStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        config,
 | 
			
		||||
        getLogger,
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const projectStore = new ProjectStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        getLogger,
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const projectStore = new ProjectStore(db, eventBus, config);
 | 
			
		||||
    const featureEnvironmentStore = new FeatureEnvironmentStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        config,
 | 
			
		||||
        getLogger,
 | 
			
		||||
    );
 | 
			
		||||
    const contextFieldStore = new ContextFieldStore(
 | 
			
		||||
        db,
 | 
			
		||||
@ -99,7 +105,7 @@ export const createFeatureToggleService = (
 | 
			
		||||
    const accessStore = new AccessStore(db, eventBus, getLogger);
 | 
			
		||||
    const featureTagStore = new FeatureTagStore(db, eventBus, getLogger);
 | 
			
		||||
    const roleStore = new RoleStore(db, eventBus, getLogger);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, config);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, getLogger);
 | 
			
		||||
    const eventService = createEventsService(db, config);
 | 
			
		||||
    const groupService = new GroupService(
 | 
			
		||||
        { groupStore, accountStore },
 | 
			
		||||
 | 
			
		||||
@ -439,16 +439,6 @@ export default class FeatureToggleStore implements IFeatureToggleStore {
 | 
			
		||||
            : queryBuilder.whereNull('archived_at');
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    static filterByProjectsAccessibleByOss: Knex.QueryCallbackWithArgs = (
 | 
			
		||||
        queryBuilder: Knex.QueryBuilder,
 | 
			
		||||
        isOss: boolean,
 | 
			
		||||
    ) => {
 | 
			
		||||
        if (isOss) {
 | 
			
		||||
            return queryBuilder.andWhere('project', '=', 'default');
 | 
			
		||||
        }
 | 
			
		||||
        return queryBuilder;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    rowToFeature(row: FeaturesTable): FeatureToggle {
 | 
			
		||||
        if (!row) {
 | 
			
		||||
            throw new NotFoundError('No feature flag found');
 | 
			
		||||
 | 
			
		||||
@ -58,8 +58,13 @@ export const createInstanceStatsService = (db: Db, config: IUnleashConfig) => {
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const userStore = new UserStore(db, getLogger, flagResolver);
 | 
			
		||||
    const projectStore = new ProjectStore(db, eventBus, config);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, config);
 | 
			
		||||
    const projectStore = new ProjectStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        getLogger,
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, getLogger);
 | 
			
		||||
    const strategyStore = new StrategyStore(db, getLogger);
 | 
			
		||||
    const contextFieldStore = new ContextFieldStore(
 | 
			
		||||
        db,
 | 
			
		||||
 | 
			
		||||
@ -21,16 +21,21 @@ export const createEnvironmentService =
 | 
			
		||||
        const featureEnvironmentStore = new FeatureEnvironmentStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            config,
 | 
			
		||||
            getLogger,
 | 
			
		||||
        );
 | 
			
		||||
        const projectStore = new ProjectStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            getLogger,
 | 
			
		||||
            flagResolver,
 | 
			
		||||
        );
 | 
			
		||||
        const projectStore = new ProjectStore(db, eventBus, config);
 | 
			
		||||
        const featureStrategiesStore = new FeatureStrategiesStore(
 | 
			
		||||
            db,
 | 
			
		||||
            eventBus,
 | 
			
		||||
            getLogger,
 | 
			
		||||
            flagResolver,
 | 
			
		||||
        );
 | 
			
		||||
        const environmentStore = new EnvironmentStore(db, eventBus, config);
 | 
			
		||||
        const environmentStore = new EnvironmentStore(db, eventBus, getLogger);
 | 
			
		||||
        const eventService = createEventsService(db, config);
 | 
			
		||||
        return new EnvironmentService(
 | 
			
		||||
            {
 | 
			
		||||
 | 
			
		||||
@ -1,6 +1,6 @@
 | 
			
		||||
import type EventEmitter from 'events';
 | 
			
		||||
import type { Db } from '../../db/db';
 | 
			
		||||
import type { Logger } from '../../logger';
 | 
			
		||||
import type { Logger, LogProvider } from '../../logger';
 | 
			
		||||
import metricsHelper from '../../util/metrics-helper';
 | 
			
		||||
import { DB_TIME } from '../../metric-events';
 | 
			
		||||
import type {
 | 
			
		||||
@ -12,7 +12,6 @@ import NotFoundError from '../../error/notfound-error';
 | 
			
		||||
import type { IEnvironmentStore } from './environment-store-type';
 | 
			
		||||
import { snakeCaseKeys } from '../../util/snakeCase';
 | 
			
		||||
import type { CreateFeatureStrategySchema } from '../../openapi';
 | 
			
		||||
import type { IUnleashConfig } from '../../types';
 | 
			
		||||
 | 
			
		||||
interface IEnvironmentsTable {
 | 
			
		||||
    name: string;
 | 
			
		||||
@ -105,18 +104,11 @@ export default class EnvironmentStore implements IEnvironmentStore {
 | 
			
		||||
 | 
			
		||||
    private db: Db;
 | 
			
		||||
 | 
			
		||||
    private isOss: boolean;
 | 
			
		||||
 | 
			
		||||
    private timer: (string) => any;
 | 
			
		||||
 | 
			
		||||
    constructor(
 | 
			
		||||
        db: Db,
 | 
			
		||||
        eventBus: EventEmitter,
 | 
			
		||||
        { getLogger, isOss }: Pick<IUnleashConfig, 'getLogger' | 'isOss'>,
 | 
			
		||||
    ) {
 | 
			
		||||
    constructor(db: Db, eventBus: EventEmitter, getLogger: LogProvider) {
 | 
			
		||||
        this.db = db;
 | 
			
		||||
        this.logger = getLogger('db/environment-store.ts');
 | 
			
		||||
        this.isOss = isOss;
 | 
			
		||||
        this.timer = (action) =>
 | 
			
		||||
            metricsHelper.wrapTimer(eventBus, DB_TIME, {
 | 
			
		||||
                store: 'environment',
 | 
			
		||||
@ -156,15 +148,9 @@ export default class EnvironmentStore implements IEnvironmentStore {
 | 
			
		||||
 | 
			
		||||
    async get(key: string): Promise<IEnvironment> {
 | 
			
		||||
        const stopTimer = this.timer('get');
 | 
			
		||||
        let keyQuery = this.db<IEnvironmentsTable>(TABLE).where({ name: key });
 | 
			
		||||
        if (this.isOss) {
 | 
			
		||||
            keyQuery = keyQuery.whereIn('name', [
 | 
			
		||||
                'default',
 | 
			
		||||
                'development',
 | 
			
		||||
                'production',
 | 
			
		||||
            ]);
 | 
			
		||||
        }
 | 
			
		||||
        const row = await keyQuery.first();
 | 
			
		||||
        const row = await this.db<IEnvironmentsTable>(TABLE)
 | 
			
		||||
            .where({ name: key })
 | 
			
		||||
            .first();
 | 
			
		||||
        stopTimer();
 | 
			
		||||
        if (row) {
 | 
			
		||||
            return mapRow(row);
 | 
			
		||||
@ -183,9 +169,6 @@ export default class EnvironmentStore implements IEnvironmentStore {
 | 
			
		||||
        if (query) {
 | 
			
		||||
            qB = qB.where(query);
 | 
			
		||||
        }
 | 
			
		||||
        if (this.isOss) {
 | 
			
		||||
            qB = qB.whereIn('name', ['default', 'development', 'production']);
 | 
			
		||||
        }
 | 
			
		||||
        const rows = await qB;
 | 
			
		||||
        stopTimer();
 | 
			
		||||
        return rows.map(mapRow);
 | 
			
		||||
@ -213,9 +196,6 @@ export default class EnvironmentStore implements IEnvironmentStore {
 | 
			
		||||
        if (query) {
 | 
			
		||||
            qB = qB.where(query);
 | 
			
		||||
        }
 | 
			
		||||
        if (this.isOss) {
 | 
			
		||||
            qB = qB.whereIn('name', ['default', 'development', 'production']);
 | 
			
		||||
        }
 | 
			
		||||
        const rows = await qB;
 | 
			
		||||
        stopTimer();
 | 
			
		||||
        return rows.map(mapRowWithCounts);
 | 
			
		||||
@ -250,13 +230,6 @@ export default class EnvironmentStore implements IEnvironmentStore {
 | 
			
		||||
        if (query) {
 | 
			
		||||
            qB = qB.where(query);
 | 
			
		||||
        }
 | 
			
		||||
        if (this.isOss) {
 | 
			
		||||
            qB = qB.whereIn('environments.name', [
 | 
			
		||||
                'default',
 | 
			
		||||
                'production',
 | 
			
		||||
                'development',
 | 
			
		||||
            ]);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const rows = await qB;
 | 
			
		||||
        stopTimer();
 | 
			
		||||
 | 
			
		||||
@ -16,7 +16,12 @@ export const createProjectInsightsService = (
 | 
			
		||||
    config: IUnleashConfig,
 | 
			
		||||
): ProjectInsightsService => {
 | 
			
		||||
    const { eventBus, getLogger, flagResolver } = config;
 | 
			
		||||
    const projectStore = new ProjectStore(db, eventBus, config);
 | 
			
		||||
    const projectStore = new ProjectStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        getLogger,
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const featureToggleStore = new FeatureToggleStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
 | 
			
		||||
@ -24,7 +24,12 @@ export const createProjectStatusService = (
 | 
			
		||||
    config: IUnleashConfig,
 | 
			
		||||
): ProjectStatusService => {
 | 
			
		||||
    const eventStore = new EventStore(db, config.getLogger);
 | 
			
		||||
    const projectStore = new ProjectStore(db, config.eventBus, config);
 | 
			
		||||
    const projectStore = new ProjectStore(
 | 
			
		||||
        db,
 | 
			
		||||
        config.eventBus,
 | 
			
		||||
        config.getLogger,
 | 
			
		||||
        config.flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const apiTokenStore = new ApiTokenStore(
 | 
			
		||||
        db,
 | 
			
		||||
        config.eventBus,
 | 
			
		||||
 | 
			
		||||
@ -63,7 +63,12 @@ export const createProjectService = (
 | 
			
		||||
): ProjectService => {
 | 
			
		||||
    const { eventBus, getLogger, flagResolver } = config;
 | 
			
		||||
    const eventStore = new EventStore(db, getLogger);
 | 
			
		||||
    const projectStore = new ProjectStore(db, eventBus, config);
 | 
			
		||||
    const projectStore = new ProjectStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        getLogger,
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const projectOwnersReadModel = new ProjectOwnersReadModel(db);
 | 
			
		||||
    const projectFlagCreatorsReadModel = new ProjectFlagCreatorsReadModel(db);
 | 
			
		||||
    const groupStore = new GroupStore(db);
 | 
			
		||||
@ -74,11 +79,11 @@ export const createProjectService = (
 | 
			
		||||
        flagResolver,
 | 
			
		||||
    );
 | 
			
		||||
    const accountStore = new AccountStore(db, getLogger);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, config);
 | 
			
		||||
    const environmentStore = new EnvironmentStore(db, eventBus, getLogger);
 | 
			
		||||
    const featureEnvironmentStore = new FeatureEnvironmentStore(
 | 
			
		||||
        db,
 | 
			
		||||
        eventBus,
 | 
			
		||||
        config,
 | 
			
		||||
        getLogger,
 | 
			
		||||
    );
 | 
			
		||||
    const projectStatsStore = new ProjectStatsStore(db, eventBus, getLogger);
 | 
			
		||||
    const accessService: AccessService = createAccessService(db, config);
 | 
			
		||||
 | 
			
		||||
@ -1,4 +1,4 @@
 | 
			
		||||
import type { Logger } from '../../logger';
 | 
			
		||||
import type { Logger, LogProvider } from '../../logger';
 | 
			
		||||
 | 
			
		||||
import NotFoundError from '../../error/notfound-error';
 | 
			
		||||
import type {
 | 
			
		||||
@ -8,7 +8,6 @@ import type {
 | 
			
		||||
    IProjectApplication,
 | 
			
		||||
    IProjectApplications,
 | 
			
		||||
    IProjectUpdate,
 | 
			
		||||
    IUnleashConfig,
 | 
			
		||||
    ProjectMode,
 | 
			
		||||
} from '../../types';
 | 
			
		||||
import type {
 | 
			
		||||
@ -73,16 +72,11 @@ class ProjectStore implements IProjectStore {
 | 
			
		||||
 | 
			
		||||
    private timer: Function;
 | 
			
		||||
 | 
			
		||||
    private isOss: boolean;
 | 
			
		||||
 | 
			
		||||
    constructor(
 | 
			
		||||
        db: Db,
 | 
			
		||||
        eventBus: EventEmitter,
 | 
			
		||||
        {
 | 
			
		||||
            getLogger,
 | 
			
		||||
            flagResolver,
 | 
			
		||||
            isOss,
 | 
			
		||||
        }: Pick<IUnleashConfig, 'getLogger' | 'flagResolver' | 'isOss'>,
 | 
			
		||||
        getLogger: LogProvider,
 | 
			
		||||
        flagResolver: IFlagResolver,
 | 
			
		||||
    ) {
 | 
			
		||||
        this.db = db;
 | 
			
		||||
        this.logger = getLogger('project-store.ts');
 | 
			
		||||
@ -92,7 +86,6 @@ class ProjectStore implements IProjectStore {
 | 
			
		||||
                action,
 | 
			
		||||
            });
 | 
			
		||||
        this.flagResolver = flagResolver;
 | 
			
		||||
        this.isOss = isOss;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
 | 
			
		||||
@ -129,9 +122,6 @@ class ProjectStore implements IProjectStore {
 | 
			
		||||
            .orderBy('name', 'asc');
 | 
			
		||||
 | 
			
		||||
        projects = projects.where(`${TABLE}.archived_at`, null);
 | 
			
		||||
        if (this.isOss) {
 | 
			
		||||
            projects = projects.where('id', 'default');
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        const rows = await projects;
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -143,7 +143,6 @@ export interface IUnleashOptions {
 | 
			
		||||
    metricsRateLimiting?: Partial<IMetricsRateLimiting>;
 | 
			
		||||
    dailyMetricsStorageDays?: number;
 | 
			
		||||
    rateLimiting?: Partial<IRateLimiting>;
 | 
			
		||||
    isOss?: boolean;
 | 
			
		||||
    resourceLimits?: Partial<
 | 
			
		||||
        Pick<
 | 
			
		||||
            ResourceLimitsSchema,
 | 
			
		||||
@ -274,7 +273,6 @@ export interface IUnleashConfig {
 | 
			
		||||
    publicFolder?: string;
 | 
			
		||||
    disableScheduler?: boolean;
 | 
			
		||||
    isEnterprise: boolean;
 | 
			
		||||
    isOss: boolean;
 | 
			
		||||
    rateLimiting: IRateLimiting;
 | 
			
		||||
    feedbackUriPath?: string;
 | 
			
		||||
    openAIAPIKey?: string;
 | 
			
		||||
 | 
			
		||||
@ -1,68 +0,0 @@
 | 
			
		||||
import {
 | 
			
		||||
    type IUnleashTest,
 | 
			
		||||
    setupAppWithCustomConfig,
 | 
			
		||||
} from '../../helpers/test-helper';
 | 
			
		||||
import dbInit, { type ITestDb } from '../../helpers/database-init';
 | 
			
		||||
import getLogger from '../../../fixtures/no-logger';
 | 
			
		||||
 | 
			
		||||
let app: IUnleashTest;
 | 
			
		||||
let db: ITestDb;
 | 
			
		||||
 | 
			
		||||
beforeAll(async () => {
 | 
			
		||||
    db = await dbInit('environment_api_is_oss_serial', getLogger, {
 | 
			
		||||
        isOss: true,
 | 
			
		||||
    });
 | 
			
		||||
    app = await setupAppWithCustomConfig(
 | 
			
		||||
        db.stores,
 | 
			
		||||
        {
 | 
			
		||||
            experimental: {
 | 
			
		||||
                flags: {
 | 
			
		||||
                    strictSchemaValidation: true,
 | 
			
		||||
                },
 | 
			
		||||
            },
 | 
			
		||||
            isOss: true,
 | 
			
		||||
        },
 | 
			
		||||
        db.rawDatabase,
 | 
			
		||||
    );
 | 
			
		||||
    await db.stores.environmentStore.create({
 | 
			
		||||
        name: 'development',
 | 
			
		||||
        type: 'development',
 | 
			
		||||
        enabled: true,
 | 
			
		||||
    });
 | 
			
		||||
    await db.stores.environmentStore.create({
 | 
			
		||||
        name: 'production',
 | 
			
		||||
        type: 'production',
 | 
			
		||||
        enabled: true,
 | 
			
		||||
    });
 | 
			
		||||
    await db.stores.environmentStore.create({
 | 
			
		||||
        name: 'customenvironment',
 | 
			
		||||
        type: 'production',
 | 
			
		||||
        enabled: true,
 | 
			
		||||
    });
 | 
			
		||||
    await db.stores.environmentStore.create({
 | 
			
		||||
        name: 'customenvironment2',
 | 
			
		||||
        type: 'production',
 | 
			
		||||
        enabled: true,
 | 
			
		||||
    });
 | 
			
		||||
    await db.stores.environmentStore.create({
 | 
			
		||||
        name: 'customenvironment3',
 | 
			
		||||
        type: 'production',
 | 
			
		||||
        enabled: true,
 | 
			
		||||
    });
 | 
			
		||||
});
 | 
			
		||||
 | 
			
		||||
afterAll(async () => {
 | 
			
		||||
    await app.destroy();
 | 
			
		||||
    await db.destroy();
 | 
			
		||||
});
 | 
			
		||||
 | 
			
		||||
test('querying environments in OSS only returns environments that are included in oss', async () => {
 | 
			
		||||
    await app.request
 | 
			
		||||
        .get('/api/admin/environments')
 | 
			
		||||
        .expect(200)
 | 
			
		||||
        .expect((res) => {
 | 
			
		||||
            expect(res.body.environments).toHaveLength(3);
 | 
			
		||||
            const names = res.body.environments.map((env) => env.name);
 | 
			
		||||
            expect(names).toEqual(['default', 'development', 'production']);
 | 
			
		||||
        });
 | 
			
		||||
});
 | 
			
		||||
@ -1,82 +0,0 @@
 | 
			
		||||
import getLogger from '../../../fixtures/no-logger';
 | 
			
		||||
import {
 | 
			
		||||
    type IUnleashTest,
 | 
			
		||||
    setupAppWithCustomConfig,
 | 
			
		||||
} from '../../helpers/test-helper';
 | 
			
		||||
import dbInit from '../../helpers/database-init';
 | 
			
		||||
import type { ITestDb } from '../../helpers/database-init';
 | 
			
		||||
import type { IAuditUser, IUser } from '../../../../lib/types';
 | 
			
		||||
 | 
			
		||||
let app: IUnleashTest;
 | 
			
		||||
let db: ITestDb;
 | 
			
		||||
let testUser: IUser;
 | 
			
		||||
const auditUser = {
 | 
			
		||||
    username: 'audituser',
 | 
			
		||||
    id: -42,
 | 
			
		||||
    ip: 'localhost',
 | 
			
		||||
} as IAuditUser;
 | 
			
		||||
 | 
			
		||||
let userIndex = 0;
 | 
			
		||||
const createUser = async (role?: number) => {
 | 
			
		||||
    const name = `User ${userIndex}`;
 | 
			
		||||
    const email = `user-${userIndex}@getunleash.io`;
 | 
			
		||||
    userIndex++;
 | 
			
		||||
 | 
			
		||||
    const { userStore } = db.stores;
 | 
			
		||||
    return userStore.insert({ name, email });
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
beforeAll(async () => {
 | 
			
		||||
    db = await dbInit('feature_api_client_is_oss', getLogger, { isOss: true });
 | 
			
		||||
    app = await setupAppWithCustomConfig(
 | 
			
		||||
        db.stores,
 | 
			
		||||
        {
 | 
			
		||||
            experimental: {
 | 
			
		||||
                flags: {
 | 
			
		||||
                    strictSchemaValidation: true,
 | 
			
		||||
                },
 | 
			
		||||
            },
 | 
			
		||||
            isOss: true,
 | 
			
		||||
        },
 | 
			
		||||
        db.rawDatabase,
 | 
			
		||||
    );
 | 
			
		||||
    testUser = await createUser();
 | 
			
		||||
    await app.services.projectService.createProject(
 | 
			
		||||
        { id: 'secondproject', name: 'Second project not returned when oss' },
 | 
			
		||||
        testUser,
 | 
			
		||||
        auditUser,
 | 
			
		||||
    );
 | 
			
		||||
    await app.services.featureToggleService.createFeatureToggle(
 | 
			
		||||
        'secondproject',
 | 
			
		||||
        {
 | 
			
		||||
            name: 'my.feature.toggle',
 | 
			
		||||
        },
 | 
			
		||||
        auditUser,
 | 
			
		||||
        true,
 | 
			
		||||
    );
 | 
			
		||||
    await app.services.featureToggleService.createFeatureToggle(
 | 
			
		||||
        'default',
 | 
			
		||||
        {
 | 
			
		||||
            name: 'my.default.toggle',
 | 
			
		||||
        },
 | 
			
		||||
        auditUser,
 | 
			
		||||
        true,
 | 
			
		||||
    );
 | 
			
		||||
});
 | 
			
		||||
afterAll(async () => {
 | 
			
		||||
    await app.destroy();
 | 
			
		||||
    await db.destroy();
 | 
			
		||||
});
 | 
			
		||||
 | 
			
		||||
describe('OSS downgrade', () => {
 | 
			
		||||
    test('features created in projects other than default is not visible in client endpoint', async () => {
 | 
			
		||||
        return app.request
 | 
			
		||||
            .get('/api/client/features')
 | 
			
		||||
            .expect('Content-Type', /json/)
 | 
			
		||||
            .expect(200)
 | 
			
		||||
            .expect((res) => {
 | 
			
		||||
                expect(res.body.features).toHaveLength(1);
 | 
			
		||||
                expect(res.body.features[0].name).toBe('my.default.toggle');
 | 
			
		||||
            });
 | 
			
		||||
    });
 | 
			
		||||
});
 | 
			
		||||
		Loading…
	
		Reference in New Issue
	
	Block a user