Allow auto-formating battery files.

Author: arik-so

src/config.mts

@@ -30,6 +30,18 @@ export default class Config {
 		return outputDirectoryPath;
 	}
 
+	getBatteryDirectoryPaths(): string[] {
+		const __dirname = url.fileURLToPath(new URL('.', import.meta.url));
+		const ciProjectPath = `${__dirname}/../ci/LDKSwift`;
+		const xcodeProjectPath = `${__dirname}/../xcode/LDKFramework`;
+		return [
+			`${ciProjectPath}/Sources/LDKSwift/batteries/`,
+			`${ciProjectPath}/Tests/LDKSwiftTests/`,
+			`${xcodeProjectPath}/DirectlyLinkedBindingsApp/`,
+			`${xcodeProjectPath}/DirectlyLinkedBindingsAppTests/`,
+		];
+	}
+
 	getSwiftFormatterBinaryPath(): string | null {
 		let swiftformatPath = process.env['SWIFT_FORMAT_PATH'];
 		if (!swiftformatPath) {

src/generation/index.mts

@@ -154,24 +154,32 @@ export default class Generator {
 		}
 	}
 
-	async runFormatter() {
+	async runFormatter(formatBatteryFiles = false) {
 		const configFilePath = url.fileURLToPath(new URL('.', import.meta.url)) + '../../.swift-format';
 		const outputDirectory = this.parser.config.getOutputBaseDirectoryPath();
 		const swiftFormatterBinary = this.parser.config.getSwiftFormatterBinaryPath();
 		if (!swiftFormatterBinary) {
 			return;
 		}
 
+		const paths = [outputDirectory];
+		if (formatBatteryFiles) {
+			paths.push(...this.parser.config.getBatteryDirectoryPaths());
+		}
+
 		try {
 			const command = `${swiftFormatterBinary} --configuration ${configFilePath} --recursive --in-place ./ `;
 			console.log(command);
-			child_process.execSync(command, {
-				cwd: outputDirectory
-			}).toString('utf-8');
+			for (const currentDirectory of paths) {
+				console.log(currentDirectory);
+				child_process.execSync(command, {
+					cwd: currentDirectory
+				}).toString('utf-8');
+			}
 		} catch (e) {
 			const errorOutput = e.stderr.toString('utf-8').trim();
 			console.error('Failed to format Swift output:', errorOutput);
-			throw new Error(errorOutput)
+			throw new Error(errorOutput);
 		}
 
 		// after all is generated and formatted, we generate the version file

src/index.mts

@@ -18,7 +18,7 @@ import Generator from './generation/index.mjs';
 
 	await generator.generateTypes();
 	await generator.generateFunctions();
-	await generator.runFormatter();
+	await generator.runFormatter(true);
 })();
 
 interface NullTest {