mls-elo/commands/create.js

Calculate ELO rankings for Major League Soccer

const db = require('sqlite');
const fs = require('fs-extra');
const path = require('path');

module.exports = async (argv) => {
  const {dbPath, verbose} = argv;
  const directory = path.dirname(dbPath);
  if (verbose) {
    console.info(`creating directory for db file in ${directory}`);
  }
  fs.mkdirp(directory);

  if (verbose) {
    console.info(`creating db in ${dbPath}`);
  }

  await db.open(dbPath, { Promise });
  await Promise.all([
    'teams',
    'matches',
    'competitions',
    'rankings',
    'valuations',
    'players',
    'outfieldPlayerMatchLog',
    'goalkeeperMatchLog',
  ].map(table => db.run(`DROP TABLE IF EXISTS ${table}`)));
  await Promise.all([
    db.run(`CREATE TABLE teams (
      teamid INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
      teamname TEXT NOT NULL,
      location TEXT NOT NULL,
      abbreviation TEXT NOT NULL,
      start INTEGER NOT NULL,
      end INTEGER,
      founded INTEGER NOT NULL,
      dissolved INTEGER
    )`),
    db.run(`CREATE TABLE competitions (
      competitionid INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
      competitionname TEXT NOT NULL,
      k INTEGER NOT NULL,
      m INTEGER NOT NULL
    )`),
    db.run(`CREATE TABLE valuations (
      valuationid INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
      valuationteamid INTEGER NOT NULL,
      year INTEGER NOT NULL,
      value REAL NOT NULL,
      revenue REAL NOT NULL,
      income REAL NOT NULL,
      FOREIGN KEY(valuationteamid) REFERENCES teams(teamid)
    );`).then(() => db.run(`CREATE UNIQUE INDEX idx_uniquevaluations ON valuations(valuationteamid, year);`)),
    db.run(`CREATE TABLE matches (
      matchid INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
      hometeam INTEGER NOT NULL,
      awayteam INTEGER NOT NULL,
      matchcompetition INTEGER NOT NULL,
      homegoals INTEGER,
      awaygoals INTEGER,
      date INTEGER NOT NULL,
      FOREIGN KEY(hometeam) REFERENCES teams(teamid),
      FOREIGN KEY(awayteam) REFERENCES teams(teamid),
      FOREIGN KEY(matchcompetition) REFERENCES competitions(competitionid)
    );`).then(() => db.run(`CREATE UNIQUE INDEX idx_uniquematches ON matches(hometeam, awayteam, date);`)),
    db.run(`CREATE TABLE rankings (
      rankingteamid INTEGER NOT NULL,
      elo INTEGER NOT NULL,
      trueskillmu INTEGER NOT NULL,
      trueskillsigma INTEGER NOT NULL,
      date INTEGER NOT NULL,
      FOREIGN KEY(rankingteamid) REFERENCES teams(teamid)
    );`).then(() => db.run(`CREATE UNIQUE INDEX idx_uniquerankings ON rankings(rankingteamid, date);`)),
    db.run(`CREATE TABLE players (
      playerid INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
      name TEXT NOT NULL,
      realname TEXT NOT NULL,
      height INTEGER,
      weight INTEGER,
      birthdate INTEGER NOT NULL,
      birthplace TEXT,
      hometown TEXT,
      twitter TEXT,
      position TEXT NOT NULL,
      jersey INTEGER
    );`),
    db.run(`CREATE TABLE outfieldPlayerMatchLog (
      playerid INTEGER NOT NULL,
      matchid INTEGER NOT NULL,
      result TEXT NOT NULL,
      appearance TEXT NOT NULL,
      minutes INTEGER NOT NULL,
      goals INTEGER NOT NULL,
      assists INTEGER NOT NULL,
      shots INTEGER NOT NULL,
      shotsongoal INTEGER NOT NULL,
      foulscommitted INTEGER NOT NULL,
      foulsssuffered INTEGER NOT NULL,
      yellows INTEGER NOT NULL,
      reds INTEGER NOT NULL,
      fantasypoints INTEGER NOT NULL,
      FOREIGN KEY(playerid) REFERENCES players(playerid),
      FOREIGN KEY(matchid) REFERENCES match(matchid),
      PRIMARY KEY (playerid, matchid)
    );`),
    db.run(`CREATE TABLE goalkeeperMatchLog (
      playerid INTEGER NOT NULL,
      matchid INTEGER NOT NULL,
      result TEXT NOT NULL,
      appearance TEXT NOT NULL,
      minutes INTEGER NOT NULL,
      goalsfor INTEGER NOT NULL,
      goalsagainst INTEGER NOT NULL,
      shotsongoal INTEGER NOT NULL,
      saves INTEGER NOT NULL,
      penaltykicksagainst INTEGER NOT NULL,
      penaltykickgoals INTEGER NOT NULL,
      penaltykicksaves INTEGER NOT NULL,
      fantasypoints INTEGER NOT NULL,
      FOREIGN KEY(playerid) REFERENCES players(playerid),
      FOREIGN KEY(matchid) REFERENCES match(matchid),
      PRIMARY KEY (playerid, matchid)
    );`),
  ]);
}