katago-analyze-sgf-cli.js

#!/bin/node

const net = require('net')

const SERVER_PORT = 6364

function main() {
  const argv = require('yargs').command(
    '$0 COMMAND [PARAMS] [OPTIONS]',
    'Process SGF files using the KataGo analysis engine - client.',
    (yargs) => {
      yargs.option('port', {
        describe: 'Port that the katago-analyze-sgf daemon is listening on.',
        type: 'number',
        default: SERVER_PORT,
      })
      yargs.positional('COMMAND', {
        describe: 'The command. Options: "submit", "list-jobs"',
        type: 'string',
      })
      yargs.positional('PARAMS', {
        describe: 'Parameters to the command.',
        type: 'string',
      })
    }
  ).argv

  const {port, COMMAND, PARAMS} = argv
  let params
  if (PARAMS) {
    try {
      params = JSON.parse(PARAMS)
    } catch {
      console.error(`Error: PARAMS must be a valid JSON string.`)
      return
    }
  }

  const connection = net.createConnection(port)

  connection.on('error', (error) => {
    switch (error.code) {
      case 'ECONNREFUSED':
        console.error(
          `Error: Could not connect to server port ${port}. ` +
            'Make sure the KataGo analysis server is running.'
        )
        break
      default:
        console.error(error)
    }
  })

  connection.on('data', (data) => {
    const response = JSON.parse(data)
    const {result, error} = response
    if (error) {
      console.error('Error:')
      console.error(error)
    } else {
      console.log(result)
    }
    connection.destroy()
  })

  connection.on('ready', () => {
    const request = {method: COMMAND, params, id: 0}
    connection.write(JSON.stringify(request))
  })
}

if (module === require.main) {
  main()
}