Extension syntax in Swift

Introduction

Extensions let you add new features to existing types without changing their original code. This helps keep your code organized and easy to update.

You want to add a new function to a type you did not create, like String or Int.

You want to group related functions together to keep your code clean.

You want to add computed properties to a type without subclassing.

You want to make an existing type conform to a protocol.

You want to add convenience initializers to a type.

Syntax

Swift

extension TypeName {
    // new functionality here
}

Replace TypeName with the name of the type you want to extend.

You can add methods, computed properties, initializers, and make the type conform to protocols.

Examples

Adds a method squared() to the Int type that returns the number multiplied by itself.

Swift

extension Int {
    func squared() -> Int {
        return self * self
    }
}

Adds a computed property reversedString to String that returns the reversed version of the string.

Swift

extension String {
    var reversedString: String {
        return String(self.reversed())
    }
}

Makes Double conform to the CustomStringConvertible protocol by adding a description property.

Swift

extension Double: CustomStringConvertible {
    var description: String {
        return "Value: \(self)"
    }
}

Sample Program

This program adds a new method squared() to the Int type. It then uses this method on the number 5 and prints the result.

Swift

extension Int {
    func squared() -> Int {
        return self * self
    }
}

let number = 5
print("\(number) squared is \(number.squared())")

Important Notes

Extensions cannot add stored properties to types, only computed properties or methods.

You can use extensions to organize your code better by grouping related functionality.

Extensions work with classes, structs, enums, and protocols.

Summary

Extensions add new features to existing types without changing their original code.

You can add methods, computed properties, initializers, and protocol conformance.

Extensions help keep your code clean and organized.

Practice

1. What is the main purpose of using extension in Swift?

easy

A. To add new functionality to an existing type without modifying its original code

B. To create a new subclass from an existing class

C. To replace the original implementation of a type

D. To delete methods from a type

Solution

1. Step 1: Understand what extensions do

   Extensions allow adding new features to existing types without changing their original source code.

2. Step 2: Compare options with this definition

   Only adding new functionality to an existing type without modifying its original code correctly describes this purpose. Replacing original implementation refers to overriding, creating subclasses is inheritance, and deleting methods is unrelated to extensions.

3. Final Answer:

   To add new functionality to an existing type without modifying its original code -> Option A

4. Quick Check:

   Extension purpose = Add features without changing original code [OK]

Hint: Extensions add features without changing original code [OK]

Common Mistakes:

• Confusing extensions with subclassing
• Thinking extensions replace original code
• Believing extensions can remove methods

2. Which of the following is the correct syntax to start an extension for the String type in Swift?

easy

A. extend String { }

B. extension String { }

C. extension String() { }

D. extension String[] { }

Solution

1. Step 1: Recall Swift extension syntax

   The correct syntax starts with the keyword extension followed by the type name without parentheses or brackets.

2. Step 2: Check each option

   extension String { } uses correct syntax. extend String { } uses invalid keyword. extension String() { } adds invalid parentheses. extension String[] { } uses invalid array notation.

3. Final Answer:

   extension String { } -> Option B

4. Quick Check:

   Extension syntax = extension TypeName { } [OK]

Hint: Use 'extension TypeName { }' without parentheses or brackets [OK]

Common Mistakes:

• Using 'extend' instead of 'extension'
• Adding parentheses after type name
• Using brackets after type name

3. What will be the output of this Swift code?

extension Int {
    func squared() -> Int {
        return self * self
    }
}

let number = 4
print(number.squared())

medium

A. 16

B. 8

C. 4

D. Compile error

Solution

1. Step 1: Understand the extension method

   The extension adds a method squared() to Int that returns the number multiplied by itself.

2. Step 2: Calculate the output for number = 4

   Calling 4.squared() returns 4 * 4 = 16.

3. Final Answer:

   16 -> Option A

4. Quick Check:

   4 squared = 16 [OK]

Hint: Multiply the number by itself in the extension method [OK]

Common Mistakes:

• Confusing squared with doubling
• Expecting syntax error due to extension
• Forgetting to return the value

4. Identify the error in this Swift extension code:

extension Double {
    func half() -> Double {
        self / 2
    }
}

medium

A. Missing return keyword in the method

B. Extension cannot add methods to Double

C. Method name 'half' is invalid

D. No error, code is correct

Solution

1. Step 1: Check method syntax in extension

   In Swift, single-expression functions can omit the return keyword.

2. Step 2: Verify if extension on Double is allowed

   Extensions can add methods to Double. The method name 'half' is valid.

3. Final Answer:

   No error, code is correct -> Option D

4. Quick Check:

   Single-expression methods can omit return [OK]

Hint: Single-expression methods can omit 'return' keyword [OK]

Common Mistakes:

• Assuming 'return' is always required
• Thinking extensions can't add methods to built-in types
• Believing method names have restrictions

5. You want to add a computed property isEven to Int using an extension that returns true if the number is even, otherwise false. Which code correctly implements this?

hard

A. extension Int { var isEven: Bool { if self % 2 == 0 { return true } } }

B. extension Int { func isEven() -> Bool { return self % 2 == 0 } }

C. extension Int { var isEven: Bool { self % 2 == 0 } }

D. extension Int { var isEven: Bool { self / 2 == 0 } }

Solution

1. Step 1: Understand computed property syntax

   Computed properties use var with a getter. extension Int { var isEven: Bool { self % 2 == 0 } } uses correct concise syntax; the func version declares a method, the if version lacks an else branch causing compile error, the division version has wrong logic.

2. Step 2: Check logic correctness

   extension Int { var isEven: Bool { self % 2 == 0 } } uses concise expression self % 2 == 0 which returns true if even. extension Int { var isEven: Bool { if self % 2 == 0 { return true } } } omits else branch, causing compiler error. extension Int { var isEven: Bool { self / 2 == 0 } } incorrectly uses division instead of modulo.

3. Step 3: Verify distractors

   The method requires parentheses to call like num.isEven(). Incomplete if fails to compile. Division fails for even numbers like 2 (2/2 == 1 != 0).

4. Final Answer:

   extension Int { var isEven: Bool { self % 2 == 0 } } -> Option C

5. Quick Check:

   Computed property with modulo check = extension Int { var isEven: Bool { self % 2 == 0 } } [OK]

Hint: Use computed property with 'self % 2 == 0' for even check [OK]

Common Mistakes:

• Using division instead of modulo
• Confusing method with computed property
• Forgetting to return a Bool value