初始上传

vendor/lizhichao/word/Lib/VicDict.php

@@ -0,0 +1,136 @@
+<?php
+
+/**
+ * Add word to dict.
+ */
+
+namespace Lizhichao\Word;
+
+class VicDict
+{
+    private $word = [];
+
+    private $code = 'utf-8';
+
+    private $end = ['\\' => 1];
+
+    private $default_end = ['\\' => 1];
+
+    private $end_key = '\\';
+
+    private $type = '';
+
+    private $dictPath = '';
+
+    /**
+     * VicDict constructor.
+     * @param string $path 词库地址
+     * @throws \Exception
+     */
+    public function __construct($path = '')
+    {
+        if($path === ''){
+            $this->dictPath = dirname(__DIR__) . '/Data/dict.json';
+        }else{
+            $this->dictPath = $path;
+        }
+        $this->type = pathinfo($this->dictPath)['extension'];
+
+        if ( ! \file_exists($this->dictPath)) {
+            throw new \Exception("Invalid dict file: {$this->dictPath}");
+        }
+
+        // check dict type
+        switch ($this->type) {
+            case 'igb':
+                if ( ! \function_exists('\\igbinary_unserialize')) {
+                    throw new \Exception('Requires igbinary PHP extension.');
+                }
+
+                $this->word = \igbinary_unserialize(\file_get_contents($this->dictPath));
+                break;
+            case 'json':
+                $this->word = \json_decode(\file_get_contents($this->dictPath), true);
+                break;
+            default:
+                throw new \Exception('Invalid dict type.');
+        }
+    }
+
+    /**
+     * @param string      $word
+     * @param null|string $x    词性
+     *
+     * @return bool
+     */
+    public function add($word, $x = null)
+    {
+        $this->end = ['\\x' => $x] + $this->default_end;
+        $word      = $this->filter($word);
+        if ($word) {
+            return $this->merge($word);
+        }
+
+        return false;
+    }
+
+    public function save()
+    {
+        if ('igb' === $this->type) {
+            $str = \igbinary_serialize($this->word);
+        } else {
+            $str = \json_encode($this->word);
+        }
+
+        return \file_put_contents($this->dictPath, $str);
+    }
+
+    private function merge($word)
+    {
+        $ar = $this->toArr($word);
+        $br = $ar;
+        $wr = &$this->word;
+        foreach ($ar as $i => $v) {
+            \array_shift($br);
+            if ( ! isset($wr[$v])) {
+                $wr[$v] = $this->dict($br, $this->end);
+
+                return true;
+            }
+            $wr = &$wr[$v];
+        }
+        if ( ! isset($wr[$this->end_key])) {
+            foreach ($this->end as $k => $v) {
+                $wr[$k] = $v;
+                $wr[$k] = $v;
+            }
+        }
+
+        return true;
+    }
+
+    private function filter($word)
+    {
+        return \str_replace(["\n", "\t", "\r"], '', $word);
+    }
+
+    private function dict($arr, $v, $i = 0)
+    {
+        if (isset($arr[$i])) {
+            return [$arr[$i] => $this->dict($arr, $v, $i + 1)];
+        }
+
+        return $v;
+    }
+
+    private function toArr($str)
+    {
+        $l = \mb_strlen($str, $this->code);
+        $r = [];
+        for ($i = 0; $i < $l; ++$i) {
+            $r[] = \mb_substr($str, $i, 1, $this->code);
+        }
+
+        return $r;
+    }
+}