Introduction & Context Setting

In our previous lesson, you learned how to eliminate duplicated code through function extraction and the refactoring of magic numbers. This lesson builds upon that foundation by introducing another crucial refactoring technique: the Extract Method. This technique is vital for transforming long, complex functions into smaller, more manageable ones, enhancing both readability and maintainability. As we delve into this lesson, remember that our goal is to follow the Test-Driven Development (TDD) workflow: Red, Green, Refactor. This iterative cycle ensures that we can leverage our tests when refactoring to confirm that we have not changed anything about the behavior. If you change behavior, it is not a successful refactor.

Understanding the Problem with Long Methods

Long methods are a code smell that can hinder efficient development as they often become difficult to understand, test, and maintain. A method might be considered long if it handles multiple responsibilities, making the code harder to track and debug. This complexity can impede our ability to effectively employ the TDD cycle, as isolated testing of functionalities becomes more challenging. Our task is to identify such cumbersome methods and employ the Extract Method technique to break them down into smaller, focused sub-methods, each with a single responsibility.

An Example of a Long Method

Take a look at the following function. Notice how it is not only long, but it is also responsible for doing a lot of things. Can you identify the different tasks this function is handling?

func processUserRegistration(userData: [String: Any]) -> [String: Any] {
    do {
        guard let username = userData["username"] as? String, username.count >= 3, username.count <= 20 else {
            return [
                "success": false,
                "message": "Invalid username. Must be between 3 and 20 characters."
            ]
        }

        guard let email = userData["email"] as? String, email.contains("@"), email.contains(".") else {
            return ["success": false, "message": "Invalid email format."]
        }

        guard let password = userData["password"] as? String, password.count >= 8,
              password.rangeOfCharacter(from: .uppercaseLetters) != nil,
              password.rangeOfCharacter(from: .decimalDigits) != nil,
              password.rangeOfCharacter(from: CharacterSet(charactersIn: "!@#$%^&*")) != nil else {
            return [
                "success": false,
                "message": "Password must be at least 8 characters and contain uppercase, number, and special character."
            ]
        }

        guard let dateOfBirthString = userData["date_of_birth"] as? String,
              let birthDate = ISO8601DateFormatter().date(from: dateOfBirthString) else {
            return [
                "success": false,
                "message": "Invalid date of birth format."
            ]
        }
        
        let today = Date()
        let ageComponents = Calendar.current.dateComponents([.year], from: birthDate, to: today)
        guard let age = ageComponents.year, age >= 18, age <= 120 else {
            return [
                "success": false,
                "message": "Invalid date of birth or user must be 18+"
            ]
        }

        guard let address = userData["address"] as? [String: String],
              let street = address["street"], street.count >= 5,
              let city = address["city"], city.count >= 2,
              let country = address["country"], country.count >= 2,
              let postalCode = address["postal_code"], postalCode.count >= 3, postalCode.rangeOfCharacter(from: .alphanumerics) != nil else {
            return ["success": false, "message": "Invalid address details."]
        }

        let userId = "USER_" + (0..<9).map { _ in "abcdefghijklmnopqrstuvwxyz0123456789".randomElement()! }.joined()

        let normalizedData: [String: Any] = [
            "username": username.lowercased(),
            "email": email.lowercased(),
            "password": password,
            "date_of_birth": ISO8601DateFormatter().string(from: birthDate),
            "address": [
                "street": street.trimmingCharacters(in: .whitespaces),
                "city": city.trimmingCharacters(in: .whitespaces),
                "country": country.uppercased(),
                "postal_code": postalCode.uppercased()
            ]
        ]

        // Assuming dataStore is a property of the class
        try dataStore.store(normalizedData)

        return [
            "success": true,
            "message": "User registered successfully",
            "user_id": userId
        ]
    } catch {
        return [
            "success": false,
            "message": "Registration failed: \(error.localizedDescription)"
        ]
    }
}

The processUserRegistration function performs multiple tasks:

  • User Validation: Checks that the username, email, and password meet specific criteria.
  • Date Validation: Verifies that the user’s date of birth is valid and within a specific age range.
  • Address Validation: Ensures that each part of the address (e.g., street, city, country, postal code) follows certain rules.
  • Data Transformation: Normalizes data (e.g., converting email and username to lowercase).
  • Data Storage: Saves the user data to the datastore.
  • Error Handling: Catches and returns any errors encountered.

By isolating each of these responsibilities into separate methods, we can improve readability and reusability and make our code easier to maintain.

Sign up
Join the 1M+ learners on CodeSignal
Be a part of our community of 1M+ users who develop and demonstrate their skills on CodeSignal