createTable.js

// Copyright 2018 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

'use strict';

const Knex = require('knex');

const createTable = async (config) => {
  const socketPath = process.env.DB_SOCKET_PATH || "/cloudsql";
  
  // Connect to the database
  if (config.dbHost){
    const dbSocketAddr = config.dbHost.split(":");
    config.host = dbSocketAddr[0];
    config.port = dbSocketAddr[1];
  } else {
    config.host = `${socketPath}/${config.connectionName}`;
  }
  
  const knex = Knex({client: 'pg', connection: config});

  // Create the "votes" table
  try {
    await knex.schema.createTable('votes', (table) => {
      table.bigIncrements('vote_id').notNull();
      table.timestamp('time_cast').notNull();
      table.specificType('candidate', 'CHAR(6) NOT NULL');
    });

    console.log(`Successfully created 'votes' table.`);
    return knex.destroy();
  } catch (err) {
    console.error(`Failed to create 'votes' table:`, err);
    if (knex) {
      knex.destroy();
    }
  }
};

require('yargs')
  .command(
    '* <user> <password> <database> <connectionName> [dbHost]',
    'Create a "votes" table',
    {},
    createTable
  )
  .help().argv;