diff --git a/src/lib/db/client-applications-store.ts b/src/lib/db/client-applications-store.ts index b3502c8ed5..efb3bc6d66 100644 --- a/src/lib/db/client-applications-store.ts +++ b/src/lib/db/client-applications-store.ts @@ -10,6 +10,7 @@ import type { Logger, LogProvider } from '../logger'; import type { Db } from './db'; import type { IApplicationOverview } from '../features/metrics/instance/models'; import { applySearchFilters } from '../features/feature-search/search-utils'; +import type { IFlagResolver } from '../types'; const COLUMNS = [ 'app_name', @@ -110,14 +111,6 @@ const remapRow = (input) => { return temp; }; -const remapUsageRow = (input) => { - return { - app_name: input.appName, - project: input.project || '*', - environment: input.environment || '*', - }; -}; - export default class ClientApplicationsStore implements IClientApplicationsStore { @@ -125,24 +118,32 @@ export default class ClientApplicationsStore private logger: Logger; - constructor(db: Db, eventBus: EventEmitter, getLogger: LogProvider) { + private flagResolver: IFlagResolver; + + constructor( + db: Db, + eventBus: EventEmitter, + getLogger: LogProvider, + flagResolver: IFlagResolver, + ) { this.db = db; + this.flagResolver = flagResolver; this.logger = getLogger('client-applications-store.ts'); } async upsert(details: Partial): Promise { const row = remapRow(details); await this.db(TABLE).insert(row).onConflict('app_name').merge(); - const usageRow = remapUsageRow(details); + const usageRows = this.remapUsageRow(details); await this.db(TABLE_USAGE) - .insert(usageRow) + .insert(usageRows) .onConflict(['app_name', 'project', 'environment']) .merge(); } async bulkUpsert(apps: Partial[]): Promise { const rows = apps.map(remapRow); - const usageRows = apps.map(remapUsageRow); + const usageRows = apps.flatMap(this.remapUsageRow); await this.db(TABLE).insert(rows).onConflict('app_name').merge(); await this.db(TABLE_USAGE) .insert(usageRows) @@ -420,4 +421,31 @@ export default class ClientApplicationsStore }, }; } + + private remapUsageRow = (input) => { + console.log(input); + if (this.flagResolver.isEnabled('parseProjectFromSession')) { + if (!input.projects || input.projects.length === 0) { + return [ + { + app_name: input.appName, + project: '*', + environment: input.environment || '*', + }, + ]; + } else { + return input.projects.map((project) => ({ + app_name: input.appName, + project: project, + environment: input.environment || '*', + })); + } + } else { + return { + app_name: input.appName, + project: input.project || '*', + environment: input.environment || '*', + }; + } + }; } diff --git a/src/lib/db/index.ts b/src/lib/db/index.ts index 7a97999e30..3fc41b3a99 100644 --- a/src/lib/db/index.ts +++ b/src/lib/db/index.ts @@ -65,6 +65,7 @@ export const createStores = ( db, eventBus, getLogger, + config.flagResolver, ), clientInstanceStore: new ClientInstanceStore(db, eventBus, getLogger), clientMetricsStoreV2: new ClientMetricsStoreV2( diff --git a/src/lib/features/metrics/instance/register.ts b/src/lib/features/metrics/instance/register.ts index 00dda75d1f..0397bb39e9 100644 --- a/src/lib/features/metrics/instance/register.ts +++ b/src/lib/features/metrics/instance/register.ts @@ -1,6 +1,6 @@ import type { Response } from 'express'; import Controller from '../../../routes/controller'; -import type { IUnleashServices } from '../../../types'; +import type { IFlagResolver, IUnleashServices } from '../../../types'; import type { IUnleashConfig } from '../../../types/option'; import type { Logger } from '../../../logger'; import type ClientInstanceService from './instance-service'; @@ -24,6 +24,8 @@ export default class RegisterController extends Controller { openApiService: OpenApiService; + flagResolver: IFlagResolver; + constructor( { clientInstanceService, @@ -35,6 +37,7 @@ export default class RegisterController extends Controller { this.logger = config.getLogger('/api/client/register'); this.clientInstanceService = clientInstanceService; this.openApiService = openApiService; + this.flagResolver = config.flagResolver; this.route({ method: 'post', @@ -62,7 +65,7 @@ export default class RegisterController extends Controller { }); } - private static resolveEnvironment( + private resolveEnvironment( user: IUser | IApiUser, data: Partial, ) { @@ -76,7 +79,14 @@ export default class RegisterController extends Controller { return 'default'; } - private static extractProjectFromRequest( + private resolveProject(user: IUser | IApiUser) { + if (user instanceof ApiUser) { + return user.projects; + } + return ['default']; + } + + private extractProjectFromRequest( req: IAuthRequest, ) { const token = req.get('Authorisation') || req.headers.authorization; @@ -91,8 +101,15 @@ export default class RegisterController extends Controller { res: Response, ): Promise { const { body: data, ip: clientIp, user } = req; - data.environment = RegisterController.resolveEnvironment(user, data); - data.project = RegisterController.extractProjectFromRequest(req); + data.environment = this.resolveEnvironment(user, data); + if (this.flagResolver.isEnabled('parseProjectFromSession')) { + data.projects = this.resolveProject(user); + console.log('Project from session: ', data.projects); + } else { + data.project = this.extractProjectFromRequest(req); + console.log('Project from request: ', data.project); + } + await this.clientInstanceService.registerClient(data, clientIp); res.header('X-Unleash-Version', version).status(202).end(); } diff --git a/src/lib/features/metrics/shared/schema.ts b/src/lib/features/metrics/shared/schema.ts index 54a8098ff3..5896dee9ee 100644 --- a/src/lib/features/metrics/shared/schema.ts +++ b/src/lib/features/metrics/shared/schema.ts @@ -92,4 +92,5 @@ export const clientRegisterSchema = joi interval: joi.number().required(), environment: joi.string().optional(), project: joi.string().optional(), + projects: joi.array().optional().items(joi.string()), }); diff --git a/src/lib/types/experimental.ts b/src/lib/types/experimental.ts index 5da6f14d53..bf58248e5b 100644 --- a/src/lib/types/experimental.ts +++ b/src/lib/types/experimental.ts @@ -59,7 +59,8 @@ export type IFlagKey = | 'bearerTokenMiddleware' | 'projectOverviewRefactorFeedback' | 'featureLifecycle' - | 'projectListFilterMyProjects'; + | 'projectListFilterMyProjects' + | 'parseProjectFromSession'; export type IFlags = Partial<{ [key in IFlagKey]: boolean | Variant }>; @@ -293,6 +294,10 @@ const flags: IFlags = { process.env.UNLEASH_EXPERIMENTAL_PROJECTS_LIST_MY_PROJECTS, false, ), + parseProjectFromSession: parseEnvVarBoolean( + process.env.UNLEASH_EXPERIMENTAL_PARSE_PROJECT_FROM_SESSION, + false, + ), }; export const defaultExperimentalOptions: IExperimentalOptions = { diff --git a/src/server-dev.ts b/src/server-dev.ts index 39e9971de6..9cfaac89f2 100644 --- a/src/server-dev.ts +++ b/src/server-dev.ts @@ -56,6 +56,7 @@ process.nextTick(async () => { projectOverviewRefactorFeedback: true, featureLifecycle: true, projectListFilterMyProjects: true, + parseProjectFromSession: false, }, }, authentication: {