Как записать слова вокруг шаблона регулярных выражений?

Question

У меня есть строка, которая:

let s : String = "The dog ate the cat." 

со следующей схемой:

let p : NSRegularExpression = try! NSRegularExpression(pattern: "\\w()?dog()?\\w"), options: .CaseInsensitive) 

Мой вопрос: Как я могу получить слова до и после появления паттерна?

Это моя текущая реализация:

let t : [NSTextCheckingResults] = p.matchesInString(s, options: [], NSMakeRange(0, NSString(string: s).length)) 
print(String(t[0].URL!)) 

Однако это не возвращает желаемый результат.

Что не так с моим регулярным выражением?

Answer 1

Попробуйте это:

extension String { 
    subscript (range: NSRange) -> String { 
     let startIndex = self.startIndex.advancedBy(range.location) 
     let endIndex = startIndex.advancedBy(range.length) 

     return self[startIndex..<endIndex] 
    } 
} 

let s = "The dog ate the cat." 

let pattern = try! NSRegularExpression(pattern: "(\\w+)?\\s?dog\\s?(\\w+)?", options: [.CaseInsensitive]) 

let matches = pattern.matchesInString(s, options: [], range: NSMakeRange(0, s.characters.count)) 

for m in matches { 
    print(s[m.rangeAtIndex(0)]) 

    let range1 = m.rangeAtIndex(1) 
    if range1.location != NSNotFound { 
     print("capture group 1:", s[range1]) 
    } 

    let range2 = m.rangeAtIndex(2) 
    if range2.location != NSNotFound { 
     print("capture group 2:", s[range2]) 
    } 
} 

Расширение для String не является обязательным, но позволяет работать с NSRange гораздо проще. В 2 захвате группа были сделана необязательной с (\\w+)?, чтобы она соответствовала следующим строкам тоже:

// The dog ate the cat 
The dog ate 
capture group 1: the 
capture group 2: ate 

// dog ate my home work 
dog ate 
capture group 2: ate 

// I love my dog 
my dog 
capture group 1: my 

// dog 
dog