From 94ec873159122787b21ddaafd157d5efde457c2c Mon Sep 17 00:00:00 2001
From: katsuta-a-wap <katsuta_a+sudachi@worksap.co.jp>
Date: Wed, 1 Jul 2026 15:04:31 +0900
Subject: [PATCH] add index and plugin for CompletionFilter

---
 .../index/SudachiCompletionFilterFactory.kt   | 44 +++++++++++++++++++
 .../sudachi/plugin/AnalysisSudachiPlugin.kt   |  1 +
 test-scripts/10-init-index.json               | 22 ++++++++++
 3 files changed, 67 insertions(+)
 create mode 100644 search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/index/SudachiCompletionFilterFactory.kt

diff --git a/search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/index/SudachiCompletionFilterFactory.kt b/search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/index/SudachiCompletionFilterFactory.kt
new file mode 100644
index 00000000..8e847fb2
--- /dev/null
+++ b/search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/index/SudachiCompletionFilterFactory.kt
@@ -0,0 +1,44 @@
+/*
+ * Copyright (c) 2020-2026 Works Applications Co., Ltd.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.worksap.nlp.elasticsearch.sudachi.index
+
+import com.worksap.nlp.lucene.sudachi.ja.SudachiCompletionFilter
+import com.worksap.nlp.tools.EnumFlag
+import org.apache.lucene.analysis.TokenStream
+import org.elasticsearch.common.settings.Settings
+import org.elasticsearch.env.Environment
+import org.elasticsearch.index.IndexSettings
+import org.elasticsearch.index.analysis.AbstractTokenFilterFactory
+
+class SudachiCompletionFilterFactory(
+    indexSettings: IndexSettings?,
+    environment: Environment?,
+    name: String?,
+    settings: Settings?,
+) : AbstractTokenFilterFactory(name) {
+
+  private val mode = Mode.fromString(settings?.get(Mode.name, null))
+
+  override fun create(tokenStream: TokenStream): TokenStream {
+    return SudachiCompletionFilter(tokenStream, mode)
+  }
+
+  companion object {
+    private object Mode :
+        EnumFlag<SudachiCompletionFilter.Mode>("mode", SudachiCompletionFilter.DEFAULT_MODE)
+  }
+}
diff --git a/search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/plugin/AnalysisSudachiPlugin.kt b/search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/plugin/AnalysisSudachiPlugin.kt
index f4f19ffe..6abea1a1 100644
--- a/search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/plugin/AnalysisSudachiPlugin.kt
+++ b/search/src/main/java/com/worksap/nlp/elasticsearch/sudachi/plugin/AnalysisSudachiPlugin.kt
@@ -68,6 +68,7 @@ class AnalysisSudachiPlugin(settings: Settings?) : Plugin(), AnalysisPlugin, Ext
         "sudachi_readingform" to provider(::SudachiReadingFormFilterFactory),
         "sudachi_split" to provider(::SudachiSplitFilterFactory),
         "sudachi_ja_stop" to provider(::SudachiStopTokenFilterFactory),
+        "sudachi_completion" to provider(::SudachiCompletionFilterFactory),
     )
   }
 
diff --git a/test-scripts/10-init-index.json b/test-scripts/10-init-index.json
index 96c166d9..eb7863c7 100644
--- a/test-scripts/10-init-index.json
+++ b/test-scripts/10-init-index.json
@@ -20,6 +20,14 @@
         "katakana_readingform": {
           "type": "sudachi_readingform",
           "use_romaji": false
+        },
+        "sudachi_completion_index": {
+          "type": "sudachi_completion",
+          "mode": "index"
+        },
+        "sudachi_completion_query": {
+          "type": "sudachi_completion",
+          "mode": "query"
         }
       },
       "analyzer": {
@@ -75,6 +83,20 @@
           "filter": [],
           "type": "custom",
           "tokenizer": "sudachi_a_tokenizer"
+        },
+        "sudachi_completion_index": {
+          "filter": [
+            "sudachi_completion_index"
+          ],
+          "type": "custom",
+          "tokenizer": "sudachi_tokenizer"
+        },
+        "sudachi_completion_query": {
+          "filter": [
+            "sudachi_completion_query"
+          ],
+          "type": "custom",
+          "tokenizer": "sudachi_tokenizer"
         }
       },
       "tokenizer": {