fix: parsing required attribute (krzysztofzablocki#287)

Author: ilyapuchka

CHANGELOG.md

@@ -22,6 +22,7 @@
 - Fixed creating cache folder when cache is disabled
 - Fixed parsing multiple enum cases annotations
 - Fixed parsing inline annotations when there is an access level or attribute
+- Fixed parsing `required` attribute
 
 ### Internal changes
 

Sourcery/Parsing/FileParser.swift

@@ -696,12 +696,13 @@ extension FileParser {
             attributesValue.map({ $0.values }).joined()
                 .flatMap(Attribute.Identifier.init(identifier:))
                 .forEach {
-                    let attributeRange = prefix.range(of: $0.description, options: .backwards)
-
+                    var attributeRange = prefix.range(of: $0.description, options: .backwards)
                     // we expect all attributes to be prefixed with `@`
-                    // but `convenience` attribute does not need it...
-                    if $0 == Attribute.Identifier.convenience {
+                    // but `convenience` and `required` attribute does not need it...
+                    if $0 == Attribute.Identifier.convenience || $0 == Attribute.Identifier.required {
                         prefix = prefix.replacingCharacters(in: attributeRange, with: "@\($0)") as NSString
+                        attributeRange.length += 1
+                        attributeRange.location -= 1
                     }
                     ranges.append(attributeRange)
             }

SourceryRuntime/Sources/Attribute.swift

@@ -86,6 +86,8 @@ public class Attribute: NSObject, AutoCoding, AutoEquatable, AutoDiffable, AutoJ
             switch self {
             case .convenience:
                 return "convenience"
+            case .required:
+                return "required"
             default:
                 return "@\(name)"
             }

SourceryTests/Parsing/FileParser + AttributesSpec.swift

@@ -58,9 +58,10 @@ class FileParserAttributesSpec: QuickSpec {
                     "objc": Attribute(name: "objc", arguments: ["some": NSNumber(value: true)], description: "@objc(some)")
                     ]))
 
-                expect(parse("class Foo { @nonobjc convenience init() {} }").first?.initializers.first?.attributes).to(equal([
+                expect(parse("class Foo { @nonobjc convenience required init() {} }").first?.initializers.first?.attributes).to(equal([
                     "nonobjc": Attribute(name: "nonobjc"),
-                    "convenience": Attribute(name: "convenience", description: "convenience")
+                    "convenience": Attribute(name: "convenience", description: "convenience"),
+                    "required": Attribute(name: "required", description: "required")
                     ]))
             }