up: update the sql to markdown gen, support language

inhere · Oct 26, 2022 · 7c4b74d · 7c4b74d
1 parent b10966f
commit 7c4b74d
Showing 1 changed file with 49 additions and 1 deletion.
diff --git a/app/Lib/Parser/DBTable.php b/app/Lib/Parser/DBTable.php
@@ -25,8 +25,27 @@ class DBTable
         'comment'   => '',
     ];
 
+    public const LANG_EN = 'en';
+    public const LANG_CN = 'zh-CN';
+
+    /**
+     * @var string
+     */
     protected string $source = '';
 
+    /**
+     * @var string
+     */
+    protected string $lang = self::LANG_CN;
+
+    /**
+     * @var array<string, string>
+     */
+    protected array $langTitles = [
+        'zh-CN' => ' 字段名 | 类型 | 是否为空 | 默认值 | 注释 ',
+        'en'    => ' Field | Type | Allow Null | Default | Comments ',
+    ];
+
     /**
      * @var string
      */
@@ -164,6 +183,11 @@ public function addIndexExpr(string $expr): self
         return $this;
     }
 
+    public function getLangTitle(): string
+    {
+        return $this->langTitles[$this->lang] ?? $this->langTitles[self::LANG_CN];
+    }
+
     /**
      * @return string
      */
@@ -232,7 +256,7 @@ public function toCreateSQL(): string
     public function toMDTable(): string
     {
         $mdNodes = [
-            ' 字段名 | 类型 | 是否为空 | 默认值 | 注释 ',
+            $this->getLangTitle(),
             '-------|------|---------|--------|-----'
         ];
 
@@ -355,4 +379,28 @@ public function setTableComment(string $tableComment): void
     {
         $this->tableComment = $tableComment;
     }
+
+    /**
+     * @param string $lang
+     *
+     * @return DBTable
+     */
+    public function setLang(string $lang): self
+    {
+        if ($lang) {
+            $this->lang = $lang;
+        }
+        return $this;
+    }
+
+    /**
+     * @param array $langTitles
+     *
+     * @return DBTable
+     */
+    public function setLangTitles(array $langTitles): self
+    {
+        $this->langTitles = array_merge($this->langTitles, $langTitles);
+        return $this;
+    }
 }