From 0270978b12fe9885eb4df971abaedb95d55b99dc Mon Sep 17 00:00:00 2001
From: Jay <admin@trj.tw>
Date: Wed, 21 Feb 2018 14:14:31 +0800
Subject: [PATCH] add dbtool

---
 bin/dbTool.js      | 69 ++++++++++++++++++++++++++++++++++++++++++++++
 bin/dbVersion.json |  5 ++++
 route/index.js     |  3 ++
 views/test.ejs     |  4 ++-
 4 files changed, 80 insertions(+), 1 deletion(-)
 create mode 100644 bin/dbTool.js
 create mode 100644 bin/dbVersion.json

diff --git a/bin/dbTool.js b/bin/dbTool.js
new file mode 100644
index 0000000..f6030e9
--- /dev/null
+++ b/bin/dbTool.js
@@ -0,0 +1,69 @@
+const pg = require('pg')
+const config = require('../config')
+const path = require('path')
+const fs = require('fs')
+
+const versions = require('./dbVersion.json').versions
+const schemaPath = path.resolve(__dirname, '../schema')
+
+if (!Array.isArray(versions) || versions.length === 0) {
+  throw new Error('Schema version empty')
+}
+
+const client = new pg.Client({
+  host: config.database.host,
+  port: config.database.port,
+  user: config.database.user,
+  password: config.database.pass,
+  database: config.database.db_name
+})
+
+// auto start function
+;
+!(async function () {
+  let flag = false
+  await client.connect()
+
+  await client.query(`select now()`)
+  console.log('Database Connected')
+
+  let version = -1
+  let checkTable = await client.query(`select exists(select 1 from "information_schema"."tables" where "table_schema" = $1 and "table_name" = $2) as exists`, ['public', 'version_ctrl'])
+  if (checkTable.rows.length > 0 && checkTable.rows[0].exists === true) {
+    let checkVersion = await client.query(`select max(version) as version from "public"."version_ctrl"`)
+    if (checkVersion.rows.length > 0 && 'version' in checkVersion.rows[0] && isFinite(checkVersion.rows[0].version)) {
+      version = checkVersion.rows[0].version
+    }
+  }
+
+  let runVer = versions.filter(t => t.version > version).sort((a, b) => a.version - b.version)
+
+  if (runVer.length === 0) return
+
+  await client.query('begin')
+  try {
+    // write table query
+    for (let i in runVer) {
+      let qstr = fs.readFileSync(path.resolve(schemaPath, runVer[i].file)).toString()
+      await client.query(qstr)
+      let query = `insert into "public"."version_ctrl" ("version", "ctime", "querystr") values ($1, now(), $2)`
+      let param = [runVer[i].version, qstr]
+      await client.query(query, param)
+    }
+
+    await client.query('commit')
+  } catch (err) {
+    flag = true
+    console.log(err)
+    await client.query('rollback')
+  }
+
+  await client.end()
+  if (flag) throw new Error('Not Finish')
+})().then(() => {
+  console.log('Finish')
+  process.exit(0)
+}).catch(err => {
+  console.error(err)
+  process.exit(1)
+})
\ No newline at end of file
diff --git a/bin/dbVersion.json b/bin/dbVersion.json
new file mode 100644
index 0000000..20a56b4
--- /dev/null
+++ b/bin/dbVersion.json
@@ -0,0 +1,5 @@
+{
+  "versions":[
+    {"file": "main.sql", "version": 1}
+  ]
+}
\ No newline at end of file
diff --git a/route/index.js b/route/index.js
index 5633e73..5a23089 100644
--- a/route/index.js
+++ b/route/index.js
@@ -90,6 +90,9 @@ r.get('/twitch/oauth', async (c, n) => {
         resolve(null)
         return
       }
+      try {
+        body = JSON.parse(body)
+      }catch (err){}
       resolve(body)
     })
   })
diff --git a/views/test.ejs b/views/test.ejs
index 2a0559a..d5bdd80 100644
--- a/views/test.ejs
+++ b/views/test.ejs
@@ -1 +1,3 @@
-<%- JSON.stringify(data) %>
\ No newline at end of file
+<pre>
+<%- JSON.stringify(data, null, 2) %>
+</pre>
\ No newline at end of file