Add ReIndexDictionary feature

Author: subins2000

c-shared-general.go

@@ -0,0 +1,16 @@
+package main
+
+/*
+#include "c-shared.h"
+#include "c-shared-varray.h"
+#include "stdlib.h"
+*/
+import "C"
+
+//export varnam_reindex_dictionary
+func varnam_reindex_dictionary(varnamHandleID C.int) C.int {
+	handle := getVarnamHandle(varnamHandleID)
+	handle.err = handle.varnam.ReIndexDictionary()
+
+	return checkError(handle.err)
+}

cli/main.go

@@ -35,6 +35,8 @@ func main() {
 	debugFlag := flag.Bool("debug", false, "Enable debugging outputs")
 	schemeFlag := flag.String("s", "", "Scheme ID")
 
+	reIndexFlag := flag.Bool("reindex", false, "Reindex user dictionary database")
+
 	learnFlag := flag.Bool("learn", false, "Learn a word")
 	unlearnFlag := flag.Bool("unlearn", false, "Unlearn a word")
 	trainFlag := flag.Bool("train", false, "Train a word with a particular pattern. 2 Arguments: Pattern & Word")
@@ -77,7 +79,13 @@ func main() {
 
 	args := flag.Args()
 
-	if *trainFlag {
+	if *reIndexFlag {
+		err := varnam.ReIndexDictionary()
+		if err != nil {
+			log.Fatal(err.Error())
+		}
+		fmt.Println("Successfully re-indexed dictionary.")
+	} else if *trainFlag {
 		pattern := args[0]
 		word := args[1]
 

govarnam/dictionary.go

@@ -96,6 +96,12 @@ func (varnam *Varnam) InitDict(dictPath string) error {
 	return err
 }
 
+// ReIndexDictionary re-indexes dictionary
+func (varnam *Varnam) ReIndexDictionary() error {
+	_, err := varnam.dictConn.Exec("INSERT INTO words_fts(words_fts) VALUES('rebuild');")
+	return err
+}
+
 type searchDictionaryType int32
 
 const (

govarnamgo/general.go

@@ -0,0 +1,11 @@
+package govarnamgo
+
+// #cgo pkg-config: govarnam
+// #include "libgovarnam.h"
+// #include "stdlib.h"
+import "C"
+
+func (handle *VarnamHandle) ReIndexDictionary() error {
+	err := C.varnam_reindex_dictionary(handle.connectionID)
+	return handle.checkError(err)
+}

govarnamgo/general_test.go

@@ -0,0 +1,10 @@
+package govarnamgo
+
+import "testing"
+
+func TestReIndex(t *testing.T) {
+	varnam := getVarnamInstance("ml")
+
+	err := varnam.ReIndexDictionary()
+	assertEqual(t, err, nil)
+}