Merge pull request #6 from ensan-hcl/add_remove_api

Add remove api
ensan-hcl · Aug 25, 2021 · 8e3e46f · 8e3e46f
2 parents 87ef611 + 1c30cf4
commit 8e3e46f
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 17 deletions.
diff --git a/Sources/KaomojiParser/KaomojiParser.swift b/Sources/KaomojiParser/KaomojiParser.swift
@@ -278,4 +278,19 @@ public struct KaomojiParser {
         let result = splited.flatMap{kaomojiExtract(from: String($0), G: tolerance.gValue, L: minimumLength)}
         return Set(result)
     }
+
+    /// Remove kaomoji inside the text.
+    /// - Parameters:
+    ///   - string: target string to search kaomoji.
+    ///   - tolerance: how torelantly judge whether a part of text is kaomoji or not
+    ///   - minimumLength: minimum length of a kaomoji
+    /// - Returns: found kaomoji set
+    public func removeKaomoji(from string: String, tolerance: Tolerance = .normal, minimumLength: Int = 3) -> String {
+        let splited = string.split(separator: "\n")
+        var result = string
+        for kaomoji in Set(splited.flatMap{kaomojiExtract(from: String($0), G: tolerance.gValue, L: minimumLength)}) {
+            result = result.replacingOccurrences(of: kaomoji, with: "")
+        }
+        return result
+    }
 }
diff --git a/Tests/KaomojiParserTests/KaomojiParserTests.swift b/Tests/KaomojiParserTests/KaomojiParserTests.swift
@@ -1,20 +1,30 @@
-    import XCTest
-    @testable import KaomojiParser
+import XCTest
+@testable import KaomojiParser
 
-    final class KaomojiParserTests: XCTestCase {
-        func testHalfKatakana() {
-            let parser = KaomojiParser()
-            XCTAssertEqual(parser.isHalfKatakana("ﾊ"), true)
-            XCTAssertEqual(parser.isHalfKatakana("レ"), false)
-            XCTAssertEqual(parser.isHalfKatakana("a"), false)
-        }
+final class KaomojiParserTests: XCTestCase {
+    func testHalfKatakana() {
+        let parser = KaomojiParser()
+        XCTAssertEqual(parser.isHalfKatakana("ﾊ"), true)
+        XCTAssertEqual(parser.isHalfKatakana("レ"), false)
+        XCTAssertEqual(parser.isHalfKatakana("a"), false)
+    }
+
+    func testSearch() {
+        let parser = KaomojiParser()
+        XCTAssertEqual(parser.search(in: "嬉しいです(≧▽≦)"), ["(≧▽≦)"])
+        XCTAssertEqual(parser.search(in: "地震だ！┗(^o^;)┓"), ["┗(^o^;)┓"])
+        XCTAssertEqual(parser.search(in: "またね(・Д・)ノ"), ["(・Д・)ノ"])
+        XCTAssertEqual(parser.search(in: "ホームランｷﾀ━━━━(ﾟ∀ﾟ)━━━━‼︎"), ["ｷﾀ━━━━(ﾟ∀ﾟ)━━━━‼︎"])
+        XCTAssertEqual(parser.search(in: "嫌い(｀ε´)　絶交しよ( ￣っ￣)"), ["(｀ε´)　", "( ￣っ￣)"])
+    }
 
-        func testParser() {
-            let parser = KaomojiParser()
-            XCTAssertEqual(parser.search(in: "嬉しいです(≧▽≦)"), ["(≧▽≦)"])
-            XCTAssertEqual(parser.search(in: "地震だ！┗(^o^;)┓"), ["┗(^o^;)┓"])
-            XCTAssertEqual(parser.search(in: "またね(・Д・)ノ"), ["(・Д・)ノ"])
-            XCTAssertEqual(parser.search(in: "ホームランｷﾀ━━━━(ﾟ∀ﾟ)━━━━‼︎"), ["ｷﾀ━━━━(ﾟ∀ﾟ)━━━━‼︎"])
-            XCTAssertEqual(parser.search(in: "嫌い(｀ε´)　絶交しよ( ￣っ￣)"), ["(｀ε´)　", "( ￣っ￣)"])
-        }
+    func testRemove() {
+        let parser = KaomojiParser()
+        XCTAssertEqual(parser.removeKaomoji(from: "嬉しいです(≧▽≦)"), "嬉しいです")
+        XCTAssertEqual(parser.removeKaomoji(from: "地震だ！┗(^o^;)┓"), "地震だ！")
+        XCTAssertEqual(parser.removeKaomoji(from: "またね(・Д・)ノ"), "またね")
+        XCTAssertEqual(parser.removeKaomoji(from: "ホームランｷﾀ━━━━(ﾟ∀ﾟ)━━━━‼︎"), "ホームラン")
+        XCTAssertEqual(parser.removeKaomoji(from: "嫌い(｀ε´)　絶交しよ( ￣っ￣)"), "嫌い絶交しよ")
     }
+
+}