Finished refactoring

Author: czechboy0

Sources/_OpenAPIGeneratorCore/Translator/CommonTranslations/translateAllAnyOneOf.swift

@@ -209,7 +209,7 @@ extension TypesFileTranslator {
         let decoder: Declaration
         if let discriminator {
             let originalName = discriminator.propertyName
-            let swiftName = context.asSwiftSafeName(originalName, .noncapitalized)
+            let swiftName = context.safeNameGenerator.swiftMemberName(for: originalName)
             codingKeysDecls = [
                 .enum(
                     accessModifier: config.access,

Sources/_OpenAPIGeneratorCore/Translator/CommonTranslations/translateRawEnum.swift

@@ -101,12 +101,12 @@ extension FileTranslator {
                 // This is unlikely to be fixed, so handling that case here.
                 // https://github.com/apple/swift-openapi-generator/issues/118
                 if isNullable && anyValue is Void {
-                    try addIfUnique(id: .string(""), caseName: context.asSwiftSafeName("", .noncapitalized))
+                    try addIfUnique(id: .string(""), caseName: context.safeNameGenerator.swiftMemberName(for: ""))
                 } else {
                     guard let rawValue = anyValue as? String else {
                         throw GenericError(message: "Disallowed value for a string enum '\(typeName)': \(anyValue)")
                     }
-                    let caseName = context.asSwiftSafeName(rawValue, .noncapitalized)
+                    let caseName = context.safeNameGenerator.swiftMemberName(for: rawValue)
                     try addIfUnique(id: .string(rawValue), caseName: caseName)
                 }
             case .integer:

Sources/_OpenAPIGeneratorCore/Translator/CommonTypes/DiscriminatorExtensions.swift

@@ -80,7 +80,7 @@ extension FileTranslator {
     /// - Parameter type: The `OneOfMappedType` for which to determine the case name.
     /// - Returns: A string representing the safe Swift name for the specified `OneOfMappedType`.
     func safeSwiftNameForOneOfMappedCase(_ type: OneOfMappedType) -> String {
-        context.asSwiftSafeName(type.rawNames[0], .noncapitalized)
+        context.safeNameGenerator.swiftMemberName(for: type.rawNames[0])
     }
 }
 

Sources/_OpenAPIGeneratorCore/Translator/CommonTypes/StructBlueprint.swift

@@ -153,7 +153,7 @@ struct PropertyBlueprint {
 extension PropertyBlueprint {
 
     /// A name that is verified to be a valid Swift identifier.
-    var swiftSafeName: String { context.asSwiftSafeName(originalName, .noncapitalized) }
+    var swiftSafeName: String { context.safeNameGenerator.swiftMemberName(for: originalName) }
 
     /// The JSON path to the property.
     ///

Sources/_OpenAPIGeneratorCore/Translator/FileTranslator.swift

@@ -44,35 +44,50 @@ protocol FileTranslator {
     func translateFile(parsedOpenAPI: ParsedOpenAPIRepresentation) throws -> StructuredSwiftRepresentation
 }
 
+/// A generator that allows overriding the documented name.
+struct OverridableSafeNameGenerator: SafeNameGenerator {
+
+    /// The upstream name generator for names that aren't overriden.
+    var upstream: any SafeNameGenerator
+
+    /// A set of overrides, where the key is the documented name and the value the desired identifier.
+    var overrides: [String: String]
+
+    func swiftTypeName(for documentedName: String) -> String {
+        if let override = overrides[documentedName] { return override }
+        return upstream.swiftTypeName(for: documentedName)
+    }
+
+    func swiftMemberName(for documentedName: String) -> String {
+        if let override = overrides[documentedName] { return override }
+        return upstream.swiftMemberName(for: documentedName)
+    }
+
+    func swiftContentTypeName(for contentType: ContentType) -> String {
+        upstream.swiftContentTypeName(for: contentType)
+    }
+}
+
 extension FileTranslator {
 
     /// A new context from the file translator.
     var context: TranslatorContext {
-        let asSwiftSafeName: (String, SwiftNameOptions) -> String
+        let safeNameGenerator: any SafeNameGenerator
         switch config.namingStrategy {
-        case .defensive: asSwiftSafeName = { $0.safeForSwiftCode_defensive(options: $1) }
-        case .idiomatic: asSwiftSafeName = { $0.safeForSwiftCode_idiomatic(options: $1) }
+        case .defensive: safeNameGenerator = .defensive
+        case .idiomatic: safeNameGenerator = .idiomatic
         }
-        let overrides = config.nameOverrides
-        return TranslatorContext(
-            asSwiftSafeName: { name, options in
-                if let override = overrides[name] { return override }
-                return asSwiftSafeName(name, options)
-            },
-            namingStrategy: config.namingStrategy
+        let overridingGenerator = OverridableSafeNameGenerator(
+            upstream: safeNameGenerator,
+            overrides: config.nameOverrides
         )
+        return TranslatorContext(safeNameGenerator: overridingGenerator)
     }
 }
 
 /// A set of configuration values for concrete file translators.
 struct TranslatorContext {
 
-    /// A closure that returns a copy of the string modified to be a valid Swift identifier.
-    ///
-    /// - Parameter string: The string to convert to be safe for Swift.
-    /// - Returns: A Swift-safe version of the input string.
-    var asSwiftSafeName: (String, SwiftNameOptions) -> String
-
-    /// The naming strategy.
-    var namingStrategy: NamingStrategy
+    /// A type that generates safe names for use as Swift identifiers.
+    var safeNameGenerator: any SafeNameGenerator
 }

Sources/_OpenAPIGeneratorCore/Translator/Multipart/MultipartContentInspector.swift

@@ -120,7 +120,7 @@ extension FileTranslator {
         }
         var parts: [MultipartSchemaTypedContent] = try topLevelObject.properties.compactMap {
             (key, value) -> MultipartSchemaTypedContent? in
-            let swiftSafeName = context.asSwiftSafeName(key, .capitalized)
+            let swiftSafeName = context.safeNameGenerator.swiftTypeName(for: key)
             let typeName = typeName.appending(
                 swiftComponent: swiftSafeName + Constants.Global.inlineTypeSuffix,
                 jsonComponent: key

Sources/_OpenAPIGeneratorCore/Translator/Multipart/translateMultipart.swift

@@ -137,7 +137,7 @@ extension TypesFileTranslator {
             switch part {
             case .documentedTyped(let documentedPart):
                 let caseDecl: Declaration = .enumCase(
-                    name: context.asSwiftSafeName(documentedPart.originalName, .noncapitalized),
+                    name: context.safeNameGenerator.swiftMemberName(for: documentedPart.originalName),
                     kind: .nameWithAssociatedValues([.init(type: .init(part.wrapperTypeUsage))])
                 )
                 let decl = try translateMultipartPartContent(
@@ -404,7 +404,7 @@ extension FileTranslator {
             switch part {
             case .documentedTyped(let part):
                 let originalName = part.originalName
-                let identifier = context.asSwiftSafeName(originalName, .noncapitalized)
+                let identifier = context.safeNameGenerator.swiftMemberName(for: originalName)
                 let contentType = part.partInfo.contentType
                 let partTypeName = part.typeName
                 let schema = part.schema
@@ -613,7 +613,7 @@ extension FileTranslator {
             switch part {
             case .documentedTyped(let part):
                 let originalName = part.originalName
-                let identifier = context.asSwiftSafeName(originalName, .noncapitalized)
+                let identifier = context.safeNameGenerator.swiftMemberName(for: originalName)
                 let contentType = part.partInfo.contentType
                 let headersTypeName = part.typeName.appending(
                     swiftComponent: Constants.Operation.Output.Payload.Headers.typeName,

Sources/_OpenAPIGeneratorCore/Translator/Operations/OperationDescription.swift

@@ -83,14 +83,14 @@ extension OperationDescription {
     /// Uses the `operationID` value in the OpenAPI operation, if one was
     /// specified. Otherwise, computes a unique name from the operation's
     /// path and HTTP method.
-    var methodName: String { context.asSwiftSafeName(operationID, .noncapitalized) }
+    var methodName: String { context.safeNameGenerator.swiftMemberName(for: operationID) }
 
     /// Returns a Swift-safe type name for the operation.
     ///
     /// Uses the `operationID` value in the OpenAPI operation, if one was
     /// specified. Otherwise, computes a unique name from the operation's
     /// path and HTTP method.
-    var operationTypeName: String { context.asSwiftSafeName(operationID, .capitalized) }
+    var operationTypeName: String { context.safeNameGenerator.swiftTypeName(for: operationID) }
 
     /// Returns the identifier for the operation.
     ///
@@ -299,7 +299,7 @@ extension OperationDescription {
             }
             let newPath = OpenAPI.Path(newComponents, trailingSlash: path.trailingSlash)
             let names: [Expression] = orderedPathParameters.map { param in
-                .identifierPattern("input").dot("path").dot(context.asSwiftSafeName(param, .noncapitalized))
+                .identifierPattern("input").dot("path").dot(context.safeNameGenerator.swiftMemberName(for: param))
             }
             let arrayExpr: Expression = .literal(.array(names))
             return (newPath.rawValue, arrayExpr)

Sources/_OpenAPIGeneratorCore/Translator/Parameters/TypedParameter.swift

@@ -48,7 +48,7 @@ extension TypedParameter {
     var name: String { parameter.name }
 
     /// The name of the parameter sanitized to be a valid Swift identifier.
-    var variableName: String { context.asSwiftSafeName(name, .noncapitalized) }
+    var variableName: String { context.safeNameGenerator.swiftMemberName(for: name) }
 
     /// A Boolean value that indicates whether the parameter must be specified
     /// when performing the OpenAPI operation.

Sources/_OpenAPIGeneratorCore/Translator/RequestBody/translateRequestBody.swift

@@ -52,7 +52,7 @@ extension TypesFileTranslator {
                 bodyMembers.append(contentsOf: inlineTypeDecls)
             }
             let contentType = content.content.contentType
-            let identifier = typeAssigner.contentSwiftName(contentType)
+            let identifier = context.safeNameGenerator.swiftContentTypeName(for: contentType)
             let associatedType = content.resolvedTypeUsage.withOptional(false)
             let contentCase: Declaration = .commentable(
                 contentType.docComment(typeName: contentTypeName),
@@ -148,7 +148,7 @@ extension ClientFileTranslator {
         var cases: [SwitchCaseDescription] = try contents.map { typedContent in
             let content = typedContent.content
             let contentType = content.contentType
-            let contentTypeIdentifier = typeAssigner.contentSwiftName(contentType)
+            let contentTypeIdentifier = context.safeNameGenerator.swiftContentTypeName(for: contentType)
             let contentTypeHeaderValue = contentType.headerValueForSending
 
             let extraBodyAssignArgs: [FunctionArgumentDescription]
@@ -251,7 +251,7 @@ extension ServerFileTranslator {
                 argumentNames: ["value"],
                 body: [
                     .expression(
-                        .dot(typeAssigner.contentSwiftName(typedContent.content.contentType))
+                        .dot(context.safeNameGenerator.swiftContentTypeName(for: typedContent.content.contentType))
                             .call([.init(label: nil, expression: .identifierPattern("value"))])
                     )
                 ]