feat: namespace files support for export patterns and imports

Author: EagleoutIce

src/project/plugins/file-plugins/files/flowr-namespace-file.ts

@@ -1,10 +1,13 @@
 import type { FileRole, FlowrFileProvider } from '../../../context/flowr-file';
 import { FlowrFile } from '../../../context/flowr-file';
+import { unquoteArgument } from '../../../../abstract-interpretation/data-frame/resolve-args';
 
 export interface NamespaceInfo {
 	exportedSymbols:      string[];
 	exportedFunctions:    string[];
 	exportS3Generics:     Map<string, string[]>;
+	exportedPatterns?:    string[];
+	importedPackages?:    Map<string, string[] | 'all'>;
 	loadsWithSideEffects: boolean;
 }
 
@@ -59,6 +62,8 @@ function parseNamespace(file: FlowrFileProvider): NamespaceFormat {
 			exportedSymbols:      [] as string[],
 			exportedFunctions:    [] as string[],
 			exportS3Generics:     new Map<string, string[]>(),
+			exportedPatterns:     [] as string[],
+			importedPackages:     new Map<string, string[] | 'all'>(),
 			loadsWithSideEffects: false,
 		},
 	} as NamespaceFormat;
@@ -107,12 +112,37 @@ function parseNamespace(file: FlowrFileProvider): NamespaceFormat {
 						exportedSymbols:      [],
 						exportedFunctions:    [],
 						exportS3Generics:     new Map<string, string[]>(),
+						exportedPatterns:     [],
+						importedPackages:     new Map<string, string[] | 'all'>(),
 						loadsWithSideEffects: false,
 					};
 				}
 				result[pkg].loadsWithSideEffects = true;
 				break;
 			}
+			case 'import': {
+				const pkg = args.trim();
+				result.current.importedPackages?.set(pkg, 'all');
+				break;
+			}
+			case 'importFrom': {
+				const parts = args.split(',').map(s => s.trim());
+				if(parts.length < 2) {
+					continue;
+				}
+				const [pkg, ...symbols] = parts;
+				let arr = result.current.importedPackages?.get(pkg);
+				if(!arr || arr === 'all') {
+					arr = [];
+					result.current.importedPackages?.set(pkg, arr);
+				}
+				arr.push(...symbols);
+				break;
+			}
+			case 'exportPattern': {
+				result.current.exportedPatterns?.push(unquoteArgument(args.trim()));
+				break;
+			}
 		}
 	}
 

test/functionality/project/plugin/namespace-file.test.ts

@@ -43,6 +43,7 @@ export(coord_cartesian)
 export(ggsave)
 export(fortify)
 export(scale_type)
+exportPattern("^[^\\.].*")
 import(grid)
 import(rlang)
 importFrom(scales,alpha)
@@ -52,7 +53,7 @@ importFrom(stats,setNames)`));
 	ctx.addRequests([{ request: 'file', content: 'test.R' }]);
 	ctx.resolvePreAnalysis();
 
-	describe.sequential('Basic exports', function() {
+	describe('Basic exports', function() {
 		test('Functions are exported', () => {
 			const deps = ctx.deps.getDependency('current');
 			assert.isDefined(deps);
@@ -74,9 +75,16 @@ importFrom(stats,setNames)`));
 			assert.isTrue(deps.namespaceInfo?.exportedSymbols.includes('coord_cartesian'));
 			assert.isTrue(deps.namespaceInfo?.exportedSymbols.includes('ggsave'));
 		});
+
+		test('Export pattern registered', () => {
+			const deps = ctx.deps.getDependency('current');
+			assert.isDefined(deps);
+			assert.strictEqual(deps.namespaceInfo?.exportedPatterns?.length, 1);
+			assert.strictEqual(deps.namespaceInfo?.exportedPatterns?.[0], '^[^\\.].*');
+		});
 	});
 
-	describe.sequential('S3 methods - fortify', function() {
+	describe('S3 methods - fortify', function() {
 		test('All fortify methods registered in namespaceInfo', () => {
 			const deps = ctx.deps.getDependency('current');
 			assert.isDefined(deps);
@@ -100,7 +108,24 @@ importFrom(stats,setNames)`));
 		});
 	});
 
-	describe.sequential('Mixed exports and S3', function() {
+	describe('Correct Imports', function() {
+		test('Imports are registered correctly', () => {
+			const deps = ctx.deps.getDependency('current');
+			assert.isDefined(deps);
+
+			assert.isTrue(deps.namespaceInfo?.importedPackages?.has('grid'));
+			assert.isTrue(deps.namespaceInfo?.importedPackages?.has('rlang'));
+			assert.isTrue(deps.namespaceInfo?.importedPackages?.has('scales'));
+			assert.isTrue(deps.namespaceInfo?.importedPackages?.has('stats'));
+
+			assert.strictEqual(deps.namespaceInfo?.importedPackages?.get('grid'), 'all');
+			assert.strictEqual(deps.namespaceInfo?.importedPackages?.get('rlang'), 'all');
+			assert.deepEqual(deps.namespaceInfo?.importedPackages?.get('scales'), ['alpha']);
+			assert.deepEqual(deps.namespaceInfo?.importedPackages?.get('stats'), ['setNames']);
+		});
+	});
+
+	describe('Mixed exports and S3', function() {
 		test('ggplot has export and S3 method', () => {
 			const deps = ctx.deps.getDependency('current');
 			assert.isDefined(deps);