import express from 'express'
import cors from 'cors'
import bodyParser from 'body-parser'
import expressWs from 'express-ws'
import { Provider, ProviderOptions } from './provider'
import { log } from './utils/logs'
const app = express()

export type CliOptions = {
  rpc?: boolean,
  port: number
  ip: string
}

export class Server {
  provider

  constructor (options?: ProviderOptions) {
    this.provider = new Provider(options)
    this.provider.init().then(() => {
      log('Provider initiated')
    }).catch((error) => {
      log(error)
    })    
  }

  start (cliOptions: CliOptions) {
    const wsApp = expressWs(app)

    app.use(cors())
    app.use(bodyParser.urlencoded({ extended: true }))
    app.use(bodyParser.json())

    app.get('/', (req, res) => {
      res.send('Welcome to remix-simulator')
    })

    if (cliOptions.rpc) {
      app.use((req, res) => {
        this.provider.sendAsync(req.body, (err, jsonResponse) => {
          if (err) {
            return res.send(JSON.stringify({ error: err }))
          }
          res.send(jsonResponse)
        })
      })
    } else {
      wsApp.app.ws('/', (ws, req) => {
        ws.on('message', (msg) => {
          this.provider.sendAsync(JSON.parse(msg.toString()), (err, jsonResponse) => {
            if (err) {
              return ws.send(JSON.stringify({ error: err }))
            }
            ws.send(JSON.stringify(jsonResponse))
          })
        })

        this.provider.on('data', (result) => {
          ws.send(JSON.stringify(result))
        })
      })
    }

    app.listen(cliOptions.port, cliOptions.ip, () => {
      if (!cliOptions.rpc) {
        log('Remix Simulator listening on ws://' + cliOptions.ip + ':' + cliOptions.port)
        log('http json-rpc is deprecated and disabled by default. To enable it use --rpc')
      } else {
        log('Remix Simulator listening on http://' + cliOptions.ip + ':' + cliOptions.port)
      }
    })
  }
}