【Swift5】文字列から特定の単語とその前後数文字だけを抽出する方法

Question

実現したいこと

文字列から特定の単語とその前後4文字だけを抽出したいです。

ソースコード

let string = "a/testKEYWORDtest/b"

let searchTerm = "KEYWORD"

上のようなコードの場合searchTermを使ってstringの中からtestKEYWORDtestだけを抽出したいです。

一気に抽出しなくてもKEYWORDの前4文字と後4文字を抜き出しておいて

前4文字 + searchTerm + 後4文字

としてもかまいません。

試したこと

indexやrangeをいろいろと試してみたのですができませんでした。

お力添えいただけますと幸いです。何卒よろしくお願いいたします。

Answer 1

extension String {
    func components(needle: String, beforCharsLength: UInt, afterCharsLength: UInt) -> [String]? {
        let pattern = "(.{(beforCharsLength)})(needle)(.{(afterCharsLength)})";
        guard let regex = try? NSRegularExpression(pattern: pattern, options:[]) else {
            return nil
        }
        if let match = regex.firstMatch(in: self, options: [], range: NSRange(location: 0, length: self.count)) {
            var result: [String] = [needle, "", ""]

            for i in 0..<match.numberOfRanges {
                if let range = Range(match.range(at: i)) {
                    let inRangeStr = String(self[self.index(self.startIndex, offsetBy: range.lowerBound)..<self.index(self.startIndex, offsetBy: range.upperBound)])
                    if i < 1 || 2 < i { continue }
                    result[i] = inRangeStr
                    if i == 2 { return result }
                }
            }
        }
        return nil;
    }
}

class ViewController: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        let arr = "a/TestKEYWORDtesT/b".components(needle: "KEYWORD", beforCharsLength: 4, afterCharsLength: 4);
        print("(arr)") // Optional(["KEYWORD", "Test", "tesT"])
    }
}

Comments

[swifty]

ご回答ありがとうございます！早速導入したところ無事問題が解決しました！凄すぎますこちらのコード。まだ自分でも読み解けていない部分がありますが読み込んで理解したいと思います。

ありがとうございました！

[thyda.eiqau]

ご提示の条件を見てから15分程度ググっただけでたどり着いたコードですので頑張ってください

[swifty]

さようでしたか。そんなにすぐに見つけられましたか..いろいろ探してはいたんですが。ありがとうございました！本当に助かりました！