Modify slice from string to string function to return last index if to string not found

Question

I have a slice function which I got here. I was wondering how I can modify it so that if the to string is not found, but it found from it will return the end index of the entire string (.count-1). Right now it's obviously crashing if I call .slice and there is no to string found.

extension String {

    func slice(from: String, to: String) -> String? {

        return (range(of: from)?.upperBound).flatMap { substringFrom in
            (range(of: to, range: substringFrom..<endIndex)?.lowerBound).map { substringTo in
                String(self[substringFrom..<substringTo])
            }
        }
    }
}

rmaddy · Accepted Answer · 2018-05-15 03:05:36Z

3

Here's one possible solution:

extension String {
    func slice(from: String, to: String) -> String? {
        if let fromRng = range(of: from) {
            if let toRng = range(of: to, range: fromRng.upperBound..<endIndex) {
                // "from" and "to" found, get parts between
                return String(self[fromRng.upperBound..<toRng.lowerBound])
            } else {
                // "to" not found, return everything after "from"
                return String(self[fromRng.upperBound...])
            }
        } else {
            // "from" not found
            return nil
        }
    }
}

It's not as "fancy" as the original but personally I think the logic is much easier to read.

answered May 15, 2018 at 3:05

rmaddy

319k44 gold badges548 silver badges591 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Martin R Over a year ago

Or if let toRng = self[fromRng.upperBound...].range(of: to) { ... to save some keystrokes :)

Collectives™ on Stack Overflow

Modify slice from string to string function to return last index if to string not found

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related