react-native-jieba 0.1.0

package/src/NativeJieba.ts

@@ -0,0 +1,19 @@
+import { TurboModuleRegistry, type TurboModule } from 'react-native';
+
+export interface Spec extends TurboModule {
+  setDictPath(path: string): void;
+  cut(sentence: string, hmm: boolean): string[];
+  cutAll(sentence: string): string[];
+  cutForSearch(sentence: string, hmm: boolean): string[];
+  cutHMM(sentence: string): string[];
+  cutSmall(sentence: string, maxWordLen: number): string[];
+  tag(sentence: string): Array<{ word: string; tag: string }>;
+  extract(
+    sentence: string,
+    topK: number
+  ): Array<{ word: string; weight: number }>;
+  insertUserWord(word: string, tag: string): boolean;
+  find(word: string): boolean;
+}
+
+export default TurboModuleRegistry.getEnforcing<Spec>('Jieba');

package/src/NativeJiebaAndroidHelper.ts

@@ -0,0 +1,7 @@
+import { TurboModuleRegistry, type TurboModule } from 'react-native';
+
+export interface Spec extends TurboModule {
+  prepareDictDir(): Promise<string>;
+}
+
+export default TurboModuleRegistry.get<Spec>('JiebaAndroidHelper');

package/src/index.tsx

@@ -0,0 +1,13 @@
+export {
+  cut,
+  cutAll,
+  cutForSearch,
+  cutHMM,
+  cutSmall,
+  tag,
+  extract,
+  insertUserWord,
+  find,
+} from './jieba';
+export type { Tagged, Keyword } from './jieba';
+export { initJieba } from './init';

package/src/init.ts

@@ -0,0 +1,25 @@
+import { Platform } from 'react-native';
+import Jieba from './NativeJieba';
+import JiebaAndroidHelper from './NativeJiebaAndroidHelper';
+
+let initPromise: Promise<void> | null = null;
+
+export function initJieba(): Promise<void> {
+  if (initPromise) return initPromise;
+  initPromise = (async () => {
+    if (Platform.OS === 'android') {
+      if (!JiebaAndroidHelper) {
+        throw new Error(
+          'react-native-jieba: JiebaAndroidHelper native module is missing.'
+        );
+      }
+      const path = await JiebaAndroidHelper.prepareDictDir();
+      Jieba.setDictPath(path);
+    }
+    // iOS: OnLoad.mm already configured the dict path from the bundle.
+  })().catch((err) => {
+    initPromise = null;
+    throw err;
+  });
+  return initPromise;
+}

package/src/jieba.ts

@@ -0,0 +1,43 @@
+import Jieba from './NativeJieba';
+
+export type Tagged = { word: string; tag: string };
+export type Keyword = { word: string; weight: number };
+
+export function cut(sentence: string, hmm: boolean = true): string[] {
+  return Jieba.cut(sentence, hmm);
+}
+
+export function cutAll(sentence: string): string[] {
+  return Jieba.cutAll(sentence);
+}
+
+export function cutForSearch(
+  sentence: string,
+  hmm: boolean = true
+): string[] {
+  return Jieba.cutForSearch(sentence, hmm);
+}
+
+export function cutHMM(sentence: string): string[] {
+  return Jieba.cutHMM(sentence);
+}
+
+export function cutSmall(sentence: string, maxWordLen: number): string[] {
+  return Jieba.cutSmall(sentence, maxWordLen);
+}
+
+export function tag(sentence: string): Tagged[] {
+  return Jieba.tag(sentence);
+}
+
+export function extract(sentence: string, topK: number = 5): Keyword[] {
+  return Jieba.extract(sentence, topK);
+}
+
+export function insertUserWord(word: string, tag: string = ''): boolean {
+  return Jieba.insertUserWord(word, tag);
+}
+
+export function find(word: string): boolean {
+  return Jieba.find(word);
+}