Merge pull request #8 from wickwirew/table-name-pattern

Added support for a table name pattern

Author: wickwirew

Sources/Compiler/Config.swift

@@ -14,6 +14,7 @@ public struct Config: Codable {
     public let output: String?
     public let databaseName: String?
     public let additionalImports: [String]?
+    public let tableNamePattern: String?
 
     struct NotFoundError: Error, CustomStringConvertible {
         var description: String { "Config does not exist" }

Sources/Compiler/Gen/Language.swift

@@ -13,6 +13,8 @@ import Foundation
 public protocol Language {
     init(options: GenerationOptions)
 
+    var options: GenerationOptions { get }
+    
     var boolName: String { get }
 
     /// A list of types that have builtin adapters supplied by the library.
@@ -186,8 +188,12 @@ extension Language {
     }
 
     private func model(for table: Table) -> GeneratedModel {
-        GeneratedModel(
-            name: table.name.name.capitalizedFirst,
+        var name = table.name.name.capitalizedFirst
+        if let pattern = options.tableNamePattern {
+            name = String(format: pattern, name)
+        }
+        return GeneratedModel(
+            name: name,
             fields: table.columns.reduce(into: [:]) { fields, column in
                 let name = column.key.description
                 let type = column.value.type
@@ -365,15 +371,18 @@ public struct GenerationOptions: Sendable {
     public var databaseName: String
     public var imports: [String]
     public var createDirectoryIfNeeded: Bool
+    public var tableNamePattern: String?
 
     public init(
         databaseName: String,
         imports: [String] = [],
-        createDirectoryIfNeeded: Bool = true
+        createDirectoryIfNeeded: Bool = true,
+        tableNamePattern: String? = nil
     ) {
         self.databaseName = databaseName
         self.imports = imports
         self.createDirectoryIfNeeded = createDirectoryIfNeeded
+        self.tableNamePattern = tableNamePattern
     }
 }
 

Sources/Compiler/Gen/SwiftLanguage.swift

@@ -6,7 +6,7 @@
 //
 
 public struct SwiftLanguage: Language {
-    let options: GenerationOptions
+    public let options: GenerationOptions
     private var writer = SourceWriter()
 
     public init(options: GenerationOptions) {

Sources/PureSQLCLI/GenerateCommand.swift

@@ -45,7 +45,8 @@ struct GenerateCommand: AsyncParsableCommand {
         let options = GenerationOptions(
             databaseName: config.databaseName ?? "DB",
             imports: config.additionalImports ?? [],
-            createDirectoryIfNeeded: !skipDirectoryCreate
+            createDirectoryIfNeeded: !skipDirectoryCreate,
+            tableNamePattern: config.tableNamePattern
         )
 
         try await generate(