NEW! Pre-order my latest book, Testing Swift! >>

How to create a custom OptionSet

Paul Hudson       @twostraws

Option sets are similar to enums, but they are designed to work as a set so you can use more than one at a time. For example, when using the range(of:) method of a string, you can specify .caseInsensitive to have the search ignore case, you can specify .backwards to have the search start from the end of the string, or you can combine the two:

let string = "The rain in Spain"
let range = string.range(of: "rain", options: [.caseInsensitive, .backwards])

That searches through the string backwards and ignoring case – we provided both options at the same time. This functionality looks like an enum, but it can also be treated as an array; Swift figures it out for you.

You can write your own by making a custom struct that conforms to the OptionSet protocol, and it doesn’t take much:

  1. Create a constant describing what the underlying value is – it’s normally an integer, but you need to be specific.
  2. Make static instances of your struct for each option you want to represent.
  3. Each of those should have a unique raw value, so its common to use bit-shifting to avoid mistakes.
  4. Add any groups of those instances as new statics.

To demonstrate this, lets create a UserRoles struct that defines roles a user might have in a GitHub account: they can create things, destroy things, and get the status of things.

All three of those roles need unique raw values, so we’re going to use bit shifting – 1 << 0, 1 << 1, and so on – to make that clear.

Here’s how it looks in Swift:

struct UserRoles: OptionSet {
    let rawValue: Int

    static let status = UserRoles(rawValue: 1 << 0)
    static let create = UserRoles(rawValue: 1 << 1)
    static let destroy = UserRoles(rawValue: 1 << 2)

You can now use any of those roles by themselves or in an array:

let roles1: UserRoles = [.create]
let roles2: UserRoles = [.create, .destroy]
let roles3: UserRoles = [.create, .destroy, .status]

You can also add an extra static instance to the UserRoles struct to represent all roles at once, like this:

static let all: UserRoles = [.status, .create, .destroy]

Available from iOS 8.0

Did this solution work for you? Please pass it on!

Other people are reading…

About the Swift Knowledge Base

This is part of the Swift Knowledge Base, a free, searchable collection of solutions for common iOS questions.

Test your Swift today

Think you know Swift? My book Swift Coding Challenges will put your Swift skills to the test – it's perfect for job interviews and more!

Buy Testing Swift Buy Practical iOS 12 Buy Pro Swift Buy Swift Design Patterns Buy Swift Coding Challenges Buy Server-Side Swift (Vapor Edition) Buy Server-Side Swift (Kitura Edition) Buy Hacking with macOS Buy Advanced iOS Volume One Buy Advanced iOS Volume Two Buy Hacking with watchOS Buy Hacking with tvOS Buy Hacking with Swift Buy Dive Into SpriteKit Buy Swift in Sixty Seconds Buy Objective-C for Swift Developers Buy Beyond Code

Was this page useful? Let me know!

Click here to visit the Hacking with Swift store >>