@azure-net/cli 2.0.3 → 2.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@azure-net/cli",
-  "version": "2.0.3",
+  "version": "2.0.5",
   "type": "module",
   "bin": {
     "azure-net": "bin/azure-net.js"

package/src/commands/generate/generateCrud.js

@@ -3,9 +3,9 @@ import fs from 'fs/promises';
 import path from 'path';
 import { selectContext, getContextPath, toPascalCase, toCamelCase, getAvailableFiles } from '../../utils/contextUtils.js';
 import { writeIfNotExists, updateIndexTs, ensureIndexExports } from '../../utils/fileUtils.js';
-import { ensureProvider, addToProvider } from '../../utils/providerUtils.js';
+import { ensureProvider, addToProvider, ensureDatasourceProvider } from '../../utils/providerUtils.js';
 
-// Templates for CRUD generation
+// Templates remain the same as before...
 const entityTemplate = `export interface I{{name}} {
 \tid: number;
 \t// Add entity fields here
@@ -13,7 +13,7 @@ const entityTemplate = `export interface I{{name}} {
 \tupdated_at: string;
 }`;
 
-const portsIndexTemplate = `import type { I{{name}} } from '../../Entities/{{name}}';
+const portsIndexTemplate = `import type { I{{name}} } from '\${{context}}/Domain/Entities/{{name}}';
 
 export interface I{{name}}Collection {
 \tdata: I{{name}}[];
@@ -28,10 +28,12 @@ export interface I{{name}}CollectionQuery {
 
 export interface I{{name}}CreateRequest {
 \t// Add request fields here
+\t[key: string]: unknown;
 }
 
 export interface I{{name}}UpdateRequest {
-\t// Add request fields here  
+\t// Add request fields here
+\t[key: string]: unknown;
 }`;
 
 const repositoryTemplate = `import { {{datasource}} } from '{{datasourceImport}}';
@@ -80,7 +82,7 @@ export class {{name}}Repository {
 }`;
 
 const serviceTemplate = `import { ClassMirror } from '@azure-net/kit';
-import { {{name}}Repository } from '../../Infrastructure/Http/Repositories';
+import { {{name}}Repository } from '\${{context}}/Infrastructure/Http/Repositories';
 
 export class {{name}}Service extends ClassMirror<{{name}}Repository> {
 \tconstructor(private {{camelName}}Repository: {{name}}Repository) {
@@ -112,7 +114,7 @@ export const Update{{name}}Schema = {{schemaFactory}}<I{{name}}UpdateRequest>()
 \t}))
 \t.create();`;
 
-const presenterTemplate = `import { {{presenterFactory}} } from '{{presenterImport}}';
+const presenterWithCoreTemplate = `import { {{presenterFactory}} } from '\$core/Presenter';
 import { ApplicationProvider } from '\${{context}}/Application';
 import { Create{{name}}Schema, Update{{name}}Schema } from './Schema';
 import type { 
@@ -142,6 +144,36 @@ export const {{name}}Presenter = {{presenterFactory}}('{{name}}Presenter', ({ cr
 \treturn { collection, resource, create, update, remove };
 });`;
 
+const presenterWithoutCoreTemplate = `import { createPresenter } from '@azure-net/kit';
+import { ApplicationProvider } from '\${{context}}/Application';
+import { Create{{name}}Schema, Update{{name}}Schema } from './Schema';
+import type { 
+\tI{{name}}CollectionQuery,
+\tI{{name}}CreateRequest,
+\tI{{name}}UpdateRequest
+} from '\${{context}}/Domain/Ports/{{name}}';
+
+export const {{name}}Presenter = createPresenter('{{name}}Presenter', () => {
+\tconst { {{name}}Service } = ApplicationProvider();
+\t
+\tconst collection = async (query?: I{{name}}CollectionQuery) => 
+\t\tawait {{name}}Service.collection(query);
+\t
+\tconst resource = async (id: number) => 
+\t\tawait {{name}}Service.resource(id);
+\t
+\tconst create = async (request: I{{name}}CreateRequest) =>
+\t\tawait {{name}}Service.create(Create{{name}}Schema.from(request).json());
+\t
+\tconst update = async (id: number, request: I{{name}}UpdateRequest) =>
+\t\tawait {{name}}Service.update(id, Update{{name}}Schema.from(request).json());
+\t
+\tconst remove = async (id: number) => 
+\t\tawait {{name}}Service.remove(id);
+\t
+\treturn { collection, resource, create, update, remove };
+});`;
+
 export default async function generateCrud() {
     const context = await selectContext('Select context for CRUD module:');
 
@@ -219,14 +251,14 @@ export default async function generateCrud() {
     console.log('\n🚀 Generating CRUD module...\n');
 
     // 1. Create Entity
-    const entityPath = path.join(contextPath, 'Domain', 'Entities', pascalName);
+    const entityPath = path.join(contextPath, 'Domain/Entities', pascalName);
     await writeIfNotExists(
         path.join(entityPath, 'index.ts'),
         entityTemplate.replace(/{{name}}/g, pascalName)
     );
 
     // 2. Create Ports
-    const portsPath = path.join(contextPath, 'Domain', 'Ports', pascalName);
+    const portsPath = path.join(contextPath, 'Domain/Ports', pascalName);
     await writeIfNotExists(
         path.join(portsPath, 'index.ts'),
         portsIndexTemplate
@@ -235,7 +267,7 @@ export default async function generateCrud() {
     );
 
     // 3. Create Repository
-    const repoPath = path.join(contextPath, 'Infrastructure', 'Http', 'Repositories');
+    const repoPath = path.join(contextPath, 'Infrastructure/Http/Repositories');
     const datasourceImport = datasource.from === 'core'
         ? '$core/Datasource'
         : `\$${context}/Infrastructure/Http/Datasource`;
@@ -253,12 +285,13 @@ export default async function generateCrud() {
     await updateIndexTs(repoPath);
 
     // 4. Create Service
-    const servicePath = path.join(contextPath, 'Application', 'Services');
+    const servicePath = path.join(contextPath, 'Application/Services');
     await writeIfNotExists(
         path.join(servicePath, `${pascalName}Service.ts`),
         serviceTemplate
             .replace(/{{name}}/g, pascalName)
             .replace(/{{camelName}}/g, camelName)
+            .replace(/{{context}}/g, context)
     );
     await updateIndexTs(servicePath);
 
@@ -273,7 +306,6 @@ export default async function generateCrud() {
         : '$core/Schema';
 
     if (schemaType === 'package') {
-        // Add the import for createSchemaFactory and rules
         const schemaImportFull = `import { createSchemaFactory } from '@azure-net/kit';
 import { createRules, validationMessagesI18n } from '@azure-net/kit/schema';
 
@@ -318,10 +350,11 @@ const Schema = createSchemaFactory(createRules(validationMessagesI18n));`;
     await updateIndexTs(schemaPath);
 
     // 6. Create Presenter
+    const presenterTemplate = presenterType === 'package'
+        ? presenterWithoutCoreTemplate
+        : presenterWithCoreTemplate;
+
     const presenterFactory = presenterType === 'package' ? 'createPresenter' : presenterType;
-    const presenterImport = presenterType === 'package'
-        ? '@azure-net/kit'
-        : '$core/Presenter';
 
     await writeIfNotExists(
         path.join(deliveryModulePath, `${pascalName}Presenter.ts`),
@@ -329,7 +362,6 @@ const Schema = createSchemaFactory(createRules(validationMessagesI18n));`;
             .replace(/{{name}}/g, pascalName)
             .replace(/{{context}}/g, context)
             .replace(/{{presenterFactory}}/g, presenterFactory)
-            .replace(/{{presenterImport}}/g, presenterImport)
     );
 
     // Create module index
@@ -340,24 +372,14 @@ const Schema = createSchemaFactory(createRules(validationMessagesI18n));`;
 
     // 7. Setup Providers
     // Ensure DatasourceProvider
-    const datasourceProviderPath = path.join(contextPath, 'Infrastructure', 'Providers', 'DatasourceProvider.ts');
-    if (!(await fs.access(datasourceProviderPath).then(() => true).catch(() => false))) {
-        const datasourceProviderContent = `import { createBoundaryProvider } from '@azure-net/kit';
-import { ${datasource.name} } from '${datasource.from === 'core' ? '$core/Datasource' : '../Http/Datasource'}';
-
-export const DatasourceProvider = createBoundaryProvider('${contextName}DatasourceProvider', () => ({
-\t${datasource.name}: () => new ${datasource.name}()
-}));`;
-        await writeIfNotExists(datasourceProviderPath, datasourceProviderContent);
-        await updateIndexTs(path.dirname(datasourceProviderPath));
-    }
+    await ensureDatasourceProvider(context, datasource.name, datasource.from === 'core');
 
     // Ensure InfrastructureProvider
-    const infraProvider = await ensureProvider(context, 'Infrastructure', {});
+    const infraProvider = await ensureProvider(context, 'Infrastructure', { hasDatasource: true });
     await addToProvider(
         infraProvider.path,
         `${pascalName}Repository`,
-        '../Http/Repositories',
+        `\$${context}/Infrastructure/Http/Repositories`,
         `DatasourceProvider.${datasource.name}`
     );
 
@@ -366,7 +388,7 @@ export const DatasourceProvider = createBoundaryProvider('${contextName}Datasour
     await addToProvider(
         appProvider.path,
         `${pascalName}Service`,
-        '../Services',
+        `\$${context}/Application/Services`,
         `InfrastructureProvider.${pascalName}Repository`
     );
 

package/src/commands/generate/generateDatasource.js

@@ -1,19 +1,26 @@
 import prompts from 'prompts';
-import { selectContext, getContextPath, toPascalCase } from '../../utils/contextUtils.js';
-import { writeIfNotExists, updateIndexTs } from '../../utils/fileUtils.js';
 import path from 'path';
+import fs from 'fs/promises';
+import { selectContext, getContextPath, toPascalCase, getAvailableFiles } from '../../utils/contextUtils.js';
+import { writeIfNotExists, updateIndexTs } from '../../utils/fileUtils.js';
 
-const datasourceTemplate = `import { BaseHttpDatasource } from '@azure-net/kit/infra';
+const datasourceWithResponseTemplate = `import { BaseHttpDatasource, type CreateRequestCallbackType } from '@azure-net/kit/infra';
+import { {{responseClass}}, type {{responseInterface}} } from '{{responseImport}}';
 import { HttpService } from '@azure-net/kit/infra';
 
 export class {{name}}Datasource extends BaseHttpDatasource {
-\tconstructor() {
+\tconstructor(config?: { baseUrl?: string; [key: string]: unknown }) {
 \t\tsuper({ 
 \t\t\thttp: new HttpService({ 
-\t\t\t\tbaseUrl: 'https://api.example.com'
+\t\t\t\tbaseUrl: config?.baseUrl || 'https://api.example.com',
+\t\t\t\t...config
 \t\t\t}) 
 \t\t});
 \t}
+
+\tasync createRequest<T>(callback: CreateRequestCallbackType<{{responseInterface}}<T>>) {
+\t\treturn new {{responseClass}}<T>(await this.createRawRequest<{{responseInterface}}<T>>(callback));
+\t}
 }`;
 
 export default async function generateDatasource() {
@@ -27,15 +34,62 @@ export default async function generateDatasource() {
 
     const pascalName = toPascalCase(name);
     const contextPath = getContextPath(context);
+
+    // Get available responses
+    const coreResponses = await getAvailableFiles(
+        path.join(process.cwd(), 'src/app/core/Response')
+    );
+
+    const contextResponses = context !== 'core'
+        ? await getAvailableFiles(path.join(contextPath, 'Infrastructure/Response'))
+        : [];
+
+    const allResponses = [
+        ...coreResponses.map(r => ({
+            title: `${r} (core)`,
+            value: { name: r, from: 'core' }
+        })),
+        ...contextResponses.map(r => ({
+            title: `${r} (${context})`,
+            value: { name: r, from: context }
+        }))
+    ];
+
+    if (allResponses.length === 0) {
+        console.error('❌ No responses available. Create a response first with make:response');
+        return;
+    }
+
+    const { response } = await prompts({
+        type: 'select',
+        name: 'response',
+        message: 'Select response for datasource:',
+        choices: allResponses
+    });
+
+    // Determine interface name
+    const responseInterface = `I${response.name.replace('Response', '')}Response`;
+
     const datasourcePath = context === 'core'
         ? path.join(contextPath, 'Datasource')
-        : path.join(contextPath, 'Infrastructure', 'Http', 'Datasource');
+        : path.join(contextPath, 'Infrastructure/Http/Datasource');
 
-    const content = datasourceTemplate.replace(/{{name}}/g, pascalName);
-    const filePath = path.join(datasourcePath, `${pascalName}Datasource.ts`);
+    const responseImport = response.from === 'core'
+        ? context === 'core' ? '../Response' : '$core/Response'
+        : context === 'core'
+            ? `../contexts/${context}/Infrastructure/Response`
+            : '../../Response';
+
+    const content = datasourceWithResponseTemplate
+        .replace(/{{name}}/g, pascalName)
+        .replace(/{{responseClass}}/g, response.name)
+        .replace(/{{responseInterface}}/g, responseInterface)
+        .replace(/{{responseImport}}/g, responseImport);
 
+    const filePath = path.join(datasourcePath, `${pascalName}Datasource.ts`);
     await writeIfNotExists(filePath, content);
     await updateIndexTs(datasourcePath);
 
     console.log(`✅ Datasource created at ${filePath}`);
+    console.log(`\n💡 Remember to configure the datasource in DatasourceProvider when using it in repositories.`);
 }

package/src/commands/generate/generatePresenter.js

@@ -10,7 +10,7 @@ export const {{name}}Presenter = {{corePresenter}}('{{name}}Presenter', ({ creat
 \t{{serviceInit}}
 \t
 \tconst example = async () => {
-\t\t// Implement presenter logic
+\t\t// Using async helpers from core presenter factory
 \t\treturn await createAsyncResource(Promise.resolve({}));
 \t};
 \t
@@ -24,8 +24,8 @@ export const {{name}}Presenter = createPresenter('{{name}}Presenter', () => {
 \t{{serviceInit}}
 \t
 \tconst example = async () => {
-\t\t// Implement presenter logic
-\t\treturn {};
+\t\t// Direct method without async helpers
+\t\treturn await Promise.resolve({});
 \t};
 \t
 \treturn { example };
@@ -65,12 +65,12 @@ export default async function generatePresenter() {
     // Get available services
     const contextPath = getContextPath(context);
     const services = await getAvailableFiles(
-        path.join(contextPath, 'Application', 'Services')
+        path.join(contextPath, 'Application/Services')
     );
 
     const serviceChoices = [
         { title: 'Without service', value: null },
-        ...services.map(s => ({ title: `${s}`, value: s }))
+        ...services.map(s => ({ title: s, value: s }))
     ];
 
     const { service } = await prompts({

package/src/commands/generate/generateRepository.js

@@ -3,7 +3,7 @@ import fs from 'fs/promises';
 import path from 'path';
 import { selectContext, getContextPath, toPascalCase, getAvailableFiles, toCamelCase } from '../../utils/contextUtils.js';
 import { writeIfNotExists, updateIndexTs } from '../../utils/fileUtils.js';
-import { ensureProvider, addToProvider } from '../../utils/providerUtils.js';
+import { ensureProvider, addToProvider, ensureDatasourceProvider } from '../../utils/providerUtils.js';
 
 const repositoryTemplate = `import { {{datasource}} } from '{{datasourceImport}}';
 
@@ -37,7 +37,7 @@ export default async function generateRepository() {
         path.join(process.cwd(), 'src/app/core/Datasource')
     );
     const contextDatasources = await getAvailableFiles(
-        path.join(contextPath, 'Infrastructure', 'Http', 'Datasource')
+        path.join(contextPath, 'Infrastructure/Http/Datasource')
     );
 
     const allDatasources = [
@@ -59,7 +59,7 @@ export default async function generateRepository() {
 
     const pascalName = toPascalCase(name);
     const contextName = toPascalCase(context);
-    const repoPath = path.join(contextPath, 'Infrastructure', 'Http', 'Repositories');
+    const repoPath = path.join(contextPath, 'Infrastructure/Http/Repositories');
     const filePath = path.join(repoPath, `${pascalName}Repository.ts`);
 
     // Generate repository
@@ -77,24 +77,14 @@ export default async function generateRepository() {
     await updateIndexTs(repoPath);
 
     // Ensure DatasourceProvider exists
-    const datasourceProviderPath = path.join(contextPath, 'Infrastructure', 'Providers', 'DatasourceProvider.ts');
-    if (!(await fs.access(datasourceProviderPath).then(() => true).catch(() => false))) {
-        const datasourceProviderContent = `import { createBoundaryProvider } from '@azure-net/kit';
-import { ${datasource.name} } from '${datasource.from === 'core' ? '$core/Datasource' : '../Http/Datasource'}';
-
-export const DatasourceProvider = createBoundaryProvider('${contextName}DatasourceProvider', () => ({
-\t${datasource.name}: () => new ${datasource.name}()
-}));`;
-        await writeIfNotExists(datasourceProviderPath, datasourceProviderContent);
-        await updateIndexTs(path.dirname(datasourceProviderPath));
-    }
+    await ensureDatasourceProvider(context, datasource.name, datasource.from === 'core');
 
     // Ensure InfrastructureProvider and add repository
     const infraProvider = await ensureProvider(context, 'Infrastructure', { hasDatasource: true });
     await addToProvider(
         infraProvider.path,
         `${pascalName}Repository`,
-        '../Http/Repositories',
+        `\$${context}/Infrastructure/Http/Repositories`,
         `DatasourceProvider.${datasource.name}`
     );
 

package/src/commands/generate/generateResponse.js

@@ -1,9 +1,9 @@
 import prompts from 'prompts';
+import path from 'path';
 import { selectContext, getContextPath, toPascalCase } from '../../utils/contextUtils.js';
 import { writeIfNotExists, updateIndexTs } from '../../utils/fileUtils.js';
-import path from 'path';
 
-const responseTemplate = `import { ResponseBuilder } from '@azure-net/kit';
+const responseTemplate = `import { ResponseBuilder } from '@azure-net/kit/infra';
 
 export interface I{{name}}Response<T = unknown> {
 \tdata: T;
@@ -23,14 +23,14 @@ export default async function generateResponse() {
     const { name } = await prompts({
         type: 'text',
         name: 'name',
-        message: 'Response name (PascalCase):'
+        message: 'Response name (without "Response" suffix):'
     });
 
     const pascalName = toPascalCase(name);
     const contextPath = getContextPath(context);
     const responsePath = context === 'core'
-        ? path.join(contextPath, 'Responses')
-        : path.join(contextPath, 'Infrastructure', 'Responses');
+        ? path.join(contextPath, 'Response')
+        : path.join(contextPath, 'Infrastructure/Response');
 
     const content = responseTemplate.replace(/{{name}}/g, pascalName);
     const filePath = path.join(responsePath, `${pascalName}Response.ts`);

package/src/utils/contextUtils.js

@@ -40,13 +40,28 @@ export async function getAvailableFiles(dir, pattern = '.ts') {
     }
 }
 
+// Fixed to preserve PascalCase properly
 export function toPascalCase(str) {
-    return str
-        .split(/[-_\s]+/)
-        .map(word => word.charAt(0).toUpperCase() + word.slice(1).toLowerCase())
+    // If already in PascalCase with multiple words, preserve it
+    if (/^[A-Z][a-z]+(?:[A-Z][a-z]+)*$/.test(str)) {
+        return str;
+    }
+
+    // Split by common delimiters but preserve existing camelCase/PascalCase
+    const words = str.split(/[-_\s]+/);
+    return words
+        .map((word, index) => {
+            // If word is already in camelCase or PascalCase, preserve it
+            if (/[a-z][A-Z]/.test(word)) {
+                return word.charAt(0).toUpperCase() + word.slice(1);
+            }
+            // Otherwise capitalize first letter
+            return word.charAt(0).toUpperCase() + word.slice(1).toLowerCase();
+        })
         .join('');
 }
 
+// Fixed to preserve camelCase properly
 export function toCamelCase(str) {
     const pascal = toPascalCase(str);
     return pascal.charAt(0).toLowerCase() + pascal.slice(1);

package/src/utils/providerUtils.js

@@ -35,7 +35,6 @@ function generateProviderContent(context, layer, dependencies) {
 
     let imports = [`import { createBoundaryProvider } from '@azure-net/kit';`];
     let deps = [];
-    let depParams = '';
 
     if (dependencies.hasDatasource && layer === 'Infrastructure') {
         imports.push(`import { DatasourceProvider } from './DatasourceProvider';`);
@@ -47,10 +46,7 @@ function generateProviderContent(context, layer, dependencies) {
         deps.push('InfrastructureProvider');
     }
 
-    if (deps.length > 0) {
-        depParams = `{ ${deps.join(', ')} }`;
-    }
-
+    const depParams = deps.length > 0 ? `{ ${deps.join(', ')} }` : '';
     const dependsOnStr = deps.length > 0
         ? `,\n\t{ dependsOn: { ${deps.join(', ')} } }`
         : '';
@@ -59,7 +55,7 @@ function generateProviderContent(context, layer, dependencies) {
 
 export const ${varName} = createBoundaryProvider(
 \t'${boundaryName}',
-\t(${depParams ? depParams : ''}) => ({
+\t(${depParams}) => ({
 \t\t// Services will be added here
 \t})${dependsOnStr}
 );`;
@@ -68,12 +64,69 @@ export const ${varName} = createBoundaryProvider(
 export async function addToProvider(providerPath, serviceName, importPath, dependency = null) {
     let content = await fs.readFile(providerPath, 'utf-8');
 
-    // Add import if not exists
-    const importStatement = `import { ${serviceName} } from '${importPath}';`;
-    if (!content.includes(serviceName)) {
-        const lastImportIndex = content.lastIndexOf('import ');
-        const endOfImportLine = content.indexOf('\n', lastImportIndex);
-        content = content.slice(0, endOfImportLine + 1) + importStatement + '\n' + content.slice(endOfImportLine + 1);
+    // Parse existing imports to group them
+    const importRegex = /^import\s+{([^}]+)}\s+from\s+['"]([^'"]+)['"]/gm;
+    const imports = new Map();
+    let match;
+
+    while ((match = importRegex.exec(content)) !== null) {
+        const [fullMatch, importedItems, fromPath] = match;
+        if (!imports.has(fromPath)) {
+            imports.set(fromPath, new Set());
+        }
+        importedItems.split(',').forEach(item => {
+            imports.get(fromPath).add(item.trim());
+        });
+    }
+
+    // Add new import to the map
+    if (!Array.from(imports.values()).some(set => set.has(serviceName))) {
+        if (!imports.has(importPath)) {
+            imports.set(importPath, new Set());
+        }
+        imports.get(importPath).add(serviceName);
+    }
+
+    // Rebuild import section
+    const newImports = [`import { createBoundaryProvider } from '@azure-net/kit';`];
+
+    // Add DatasourceProvider import if exists
+    if (content.includes('DatasourceProvider') && !content.includes("import { DatasourceProvider }")) {
+        newImports.push(`import { DatasourceProvider } from './DatasourceProvider';`);
+    }
+
+    // Add other imports
+    for (const [fromPath, items] of imports) {
+        if (fromPath !== '@azure-net/kit' && !fromPath.includes('DatasourceProvider')) {
+            const itemsStr = Array.from(items).join(', ');
+            newImports.push(`import { ${itemsStr} } from '${fromPath}';`);
+        }
+    }
+
+    // Find where imports end and provider starts
+    const providerStartIndex = content.indexOf('export const');
+    const beforeProvider = content.substring(0, providerStartIndex);
+    const afterProvider = content.substring(providerStartIndex);
+
+    // Check if DatasourceProvider dependency needs to be added
+    let updatedAfterProvider = afterProvider;
+    if (dependency && dependency.includes('DatasourceProvider')) {
+        // Check if provider already has dependencies
+        if (!afterProvider.includes('({ DatasourceProvider })')) {
+            // Add DatasourceProvider to function params
+            updatedAfterProvider = updatedAfterProvider.replace(
+                /createBoundaryProvider\(\s*'[^']+',\s*\(\)/,
+                `createBoundaryProvider(\n\t'$1',\n\t({ DatasourceProvider })`
+            );
+
+            // Add dependsOn if not exists
+            if (!updatedAfterProvider.includes('dependsOn')) {
+                updatedAfterProvider = updatedAfterProvider.replace(
+                    /\)\s*\);$/,
+                    `),\n\t{ dependsOn: { DatasourceProvider } }\n);`
+                );
+            }
+        }
     }
 
     // Add factory method with dependency
@@ -81,27 +134,121 @@ export async function addToProvider(providerPath, serviceName, importPath, depen
         ? `\t\t${serviceName}: () => new ${serviceName}(${dependency})`
         : `\t\t${serviceName}: () => new ${serviceName}()`;
 
-    if (!content.includes(`${serviceName}:`)) {
+    if (!updatedAfterProvider.includes(`${serviceName}:`)) {
         // Find the position to insert
-        const returnIndex = content.indexOf('=> ({');
-        const endOfReturn = content.indexOf('\t})', returnIndex);
+        const returnIndex = updatedAfterProvider.indexOf('=> ({');
+        const endOfReturn = updatedAfterProvider.indexOf('\t})', returnIndex);
 
         // Check if there are existing services
-        const servicesSection = content.substring(returnIndex, endOfReturn);
-        const hasServices = servicesSection.includes(':');
+        const servicesSection = updatedAfterProvider.substring(returnIndex, endOfReturn);
+        const hasServices = servicesSection.includes(':') && !servicesSection.includes('// Services will be added here');
 
         if (hasServices) {
             // Add comma before new service
-            const beforeEnd = content.lastIndexOf('\n', endOfReturn);
-            content = content.slice(0, beforeEnd) + ',\n' + factoryLine + content.slice(beforeEnd);
+            const beforeEnd = updatedAfterProvider.lastIndexOf('\n', endOfReturn);
+            updatedAfterProvider = updatedAfterProvider.slice(0, beforeEnd) + ',\n' + factoryLine + updatedAfterProvider.slice(beforeEnd);
         } else {
             // First service, replace comment
-            content = content.replace(
+            updatedAfterProvider = updatedAfterProvider.replace(
                 '// Services will be added here',
                 factoryLine
             );
         }
     }
 
-    await fs.writeFile(providerPath, content, 'utf-8');
+    // Combine everything
+    const finalContent = newImports.join('\n') + '\n\n' + updatedAfterProvider;
+    await fs.writeFile(providerPath, finalContent, 'utf-8');
+}
+
+export async function ensureDatasourceProvider(context, datasourceName, isFromCore = true) {
+    const contextName = toPascalCase(context);
+    const contextPath = path.join(process.cwd(), 'src/app/contexts', context);
+    const providerPath = path.join(contextPath, 'Infrastructure/Providers/DatasourceProvider.ts');
+
+    try {
+        await fs.access(providerPath);
+        // Provider exists, add datasource if needed
+        await addDatasourceToProvider(providerPath, datasourceName, isFromCore);
+    } catch {
+        // Create new DatasourceProvider
+        const content = `import { createBoundaryProvider } from '@azure-net/kit';
+import { ${datasourceName} } from '${isFromCore ? '$core' : '../Http/Datasource'}';
+
+export const DatasourceProvider = createBoundaryProvider('${contextName}DatasourceProvider', () => ({
+\t${datasourceName}: () => new ${datasourceName}({})
+}));`;
+
+        await writeIfNotExists(providerPath, content);
+        await updateIndexTs(path.dirname(providerPath));
+    }
+}
+
+async function addDatasourceToProvider(providerPath, datasourceName, isFromCore = true) {
+    let content = await fs.readFile(providerPath, 'utf-8');
+
+    // Check if datasource already exists
+    if (content.includes(`${datasourceName}:`)) {
+        return;
+    }
+
+    // Parse existing imports to group them
+    const importRegex = /^import\s+{([^}]+)}\s+from\s+['"]([^'"]+)['"]/gm;
+    const imports = new Map();
+    let match;
+
+    while ((match = importRegex.exec(content)) !== null) {
+        const [fullMatch, importedItems, fromPath] = match;
+        if (!imports.has(fromPath)) {
+            imports.set(fromPath, new Set());
+        }
+        importedItems.split(',').forEach(item => {
+            imports.get(fromPath).add(item.trim());
+        });
+    }
+
+    // Add datasource to imports
+    const importPath = isFromCore ? '$core' : '../Http/Datasource';
+    if (!imports.has(importPath)) {
+        imports.set(importPath, new Set());
+    }
+    imports.get(importPath).add(datasourceName);
+
+    // Rebuild import section
+    const newImports = [`import { createBoundaryProvider } from '@azure-net/kit';`];
+    for (const [fromPath, items] of imports) {
+        if (fromPath !== '@azure-net/kit') {
+            const itemsStr = Array.from(items).join(', ');
+            newImports.push(`import { ${itemsStr} } from '${fromPath}';`);
+        }
+    }
+
+    // Find provider content
+    const providerStartIndex = content.indexOf('export const');
+    const afterProvider = content.substring(providerStartIndex);
+
+    // Add factory method
+    const factoryLine = `\t${datasourceName}: () => new ${datasourceName}({})`;
+
+    const returnIndex = afterProvider.indexOf('=> ({');
+    const endOfReturn = afterProvider.indexOf('})', returnIndex);
+
+    // Check if there are existing services
+    const servicesSection = afterProvider.substring(returnIndex, endOfReturn);
+    const hasServices = servicesSection.includes(':');
+
+    let updatedAfterProvider = afterProvider;
+    if (hasServices) {
+        // Add comma before new service
+        const beforeEnd = updatedAfterProvider.lastIndexOf('\n', endOfReturn);
+        updatedAfterProvider = updatedAfterProvider.slice(0, beforeEnd) + ',\n' + factoryLine + updatedAfterProvider.slice(beforeEnd);
+    } else {
+        // First service
+        const insertPoint = returnIndex + '=> ({\n'.length;
+        updatedAfterProvider = updatedAfterProvider.slice(0, insertPoint) + factoryLine + '\n' + updatedAfterProvider.slice(insertPoint);
+    }
+
+    // Combine everything
+    const finalContent = newImports.join('\n') + '\n\n' + updatedAfterProvider;
+    await fs.writeFile(providerPath, finalContent, 'utf-8');
 }