init

vendor/overtrue/pinyin/src/GeneratorFileDictLoader.php

@@ -0,0 +1,142 @@
+<?php
+
+/*
+ * This file is part of the overtrue/pinyin.
+ *
+ * (c) overtrue <i@overtrue.me>
+ *
+ * This source file is subject to the MIT license that is bundled
+ * with this source code in the file LICENSE.
+ */
+
+namespace Overtrue\Pinyin;
+
+use Closure;
+use SplFileObject;
+use Generator;
+
+/**
+ * Generator syntax(yield) Dict File loader.
+ */
+class GeneratorFileDictLoader implements DictLoaderInterface
+{
+    /**
+     * Data directory.
+     *
+     * @var string
+     */
+    protected $path;
+
+    /**
+     * Words segment name.
+     *
+     * @var string
+     */
+    protected $segmentName = 'words_%s';
+
+    /**
+     * SplFileObjects.
+     *
+     * @var array
+     */
+    protected static $handles = [];
+
+    /**
+     * surnames.
+     *
+     * @var SplFileObject
+     */
+    protected static $surnamesHandle;
+
+    /**
+     * Constructor.
+     *
+     * @param string $path
+     */
+    public function __construct($path)
+    {
+        $this->path = $path;
+
+        for ($i = 0; $i < 100; ++$i) {
+            $segment = $this->path.'/'.sprintf($this->segmentName, $i);
+
+            if (file_exists($segment) && is_file($segment)) {
+                array_push(static::$handles, $this->openFile($segment));
+            }
+        }
+    }
+
+    /**
+     * Construct a new file object.
+     *
+     * @param string $filename file path
+     *
+     * @return SplFileObject
+     */
+    protected function openFile($filename, $mode = 'r')
+    {
+        return new SplFileObject($filename, $mode);
+    }
+
+    /**
+     * get Generator syntax.
+     *
+     * @param array $handles SplFileObjects
+     */
+    protected function getGenerator(array $handles)
+    {
+        foreach ($handles as $handle) {
+            $handle->seek(0);
+            while ($handle->eof() === false) {
+                $string = str_replace(['\'', ' ', PHP_EOL, ','], '', $handle->fgets());
+
+                if (strpos($string, '=>') === false) {
+                    continue;
+                }
+
+                list($string, $pinyin) = explode('=>', $string);
+
+                yield $string => $pinyin;
+            }
+        }
+    }
+
+    /**
+     * Traverse the stream.
+     *
+     * @param Generator $generator
+     * @param Closure   $callback
+     *
+     * @author Seven Du <shiweidu@outlook.com>
+     */
+    protected function traversing(Generator $generator, Closure $callback)
+    {
+        foreach ($generator as $string => $pinyin) {
+            $callback([$string => $pinyin]);
+        }
+    }
+
+    /**
+     * Load dict.
+     *
+     * @param Closure $callback
+     */
+    public function map(Closure $callback)
+    {
+        $this->traversing($this->getGenerator(static::$handles), $callback);
+    }
+
+    /**
+     * Load surname dict.
+     *
+     * @param Closure $callback
+     */
+    public function mapSurname(Closure $callback)
+    {
+        if (!static::$surnamesHandle instanceof SplFileObject) {
+            static::$surnamesHandle = $this->openFile($this->path.'/surnames');
+        }
+
+        $this->traversing($this->getGenerator([static::$surnamesHandle]), $callback);
+    }
+}