slate/server.js

import * as Middleware from '~/common/middleware';
import * as Strings from '~/common/strings';
import * as Utilities from '~/node_common/utilities';
import * as Constants from '~/node_common/constants';

import { createPow, ffs } from '@textile/powergate-client';
const PowerGate = createPow({ host: Constants.POWERGATE_HOST });

import FS from 'fs-extra';
import WebSocketServer from 'ws';
import express from 'express';
import formidable from 'formidable';
import next from 'next';
import bodyParser from 'body-parser';
import compression from 'compression';

// TODO(jim): Support multiple desktop applications.
let client = null;
let state = null;

const production = process.env.NODE_ENV === 'production';
const port = process.env.PORT || 1337;
const wsPort = process.env.WS_PORT || 2448;
const app = next({ dev: !production, quiet: false });
const nextRequestHandler = app.getRequestHandler();

const setIntervalViewerUpdatesUnsafe = async () => {
  if (client) {
    try {
      console.log('[ prototype ] polling: new viewer state.');
      state = await Utilities.emitState({
        state,
        client,
        PG: PowerGate,
      });

      console.log('[ prototype ] polling: new library state.');
      state = await Utilities.refreshLibrary({
        state,
        PG: PowerGate,
        FFS: ffs,
      });
    } catch (e) {
      console.log(e);
    }
  }

  setTimeout(setIntervalViewerUpdatesUnsafe, Constants.POLLING_RATE);
};

app.prepare().then(async () => {
  console.log('[ prototype ] initializing ');

  state = {
    production,
    port,
    wsPort,
    token: null,
    library: null,
    status: null,
    messageList: null,
    peersList: null,
    addrsList: null,
    info: null,
  };

  if (!production) {
    // TODO(jim): Remove later.
    // We wipe all of the local data each time you run the application.
    try {
      await Utilities.resetFileSystem();

      const updates = await Utilities.refresh({ PG: PowerGate });
      state = await Utilities.updateStateData(state, updates);
      console.log('[ prototype ] updated without token');

      // NOTE(jim): This is a configuration folder with all of the client tokens.
      if (!FS.existsSync(`./.data`)) {
        FS.mkdirSync(`./.data`, { recursive: true });
      }

      // NOTE(jim): This will create a token for authentication with powergate.
      if (!FS.existsSync('./.data/powergate-token')) {
        const FFS = await PowerGate.ffs.create();
        state.token = FFS.token ? FFS.token : null;

        // NOTE(jim): Write a new token file.
        if (state.token) {
          FS.writeFileSync('./.data/powergate-token', state.token);
        }
      } else {
        state.token = FS.readFileSync('./.data/powergate-token', 'utf8');
      }

      if (state.token) {
        console.log('[ prototype ] powergate token:', state.token);
        PowerGate.setToken(state.token);
      }

      const tokenUpdates = await Utilities.refreshWithToken({
        PG: PowerGate,
      });
      state = await Utilities.updateStateData(state, tokenUpdates);
      console.log('[ prototype ] updated with token');

      // TODO(jim): Needs to support nested folders in the future.
      if (!FS.existsSync('./.data/library.json')) {
        const librarySchema = {
          library: [
            {
              ...Utilities.createFolder({ id: Constants.FILE_STORAGE_URL }),
              file: 'Files',
              name: 'Files',
            },
          ],
        };

        FS.writeFileSync('./.data/library.json', JSON.stringify(librarySchema));
        state.library = librarySchema.library;
      } else {
        const parsedLibrary = FS.readFileSync('./.data/library.json', 'utf8');
        state.library = JSON.parse(parsedLibrary).library;
      }
    } catch (e) {
      console.log('[ prototype ] you can not run the filecoin client, only web views allowed.');
    }
  }

  const server = express();
  const WSS = new WebSocketServer.Server({ port: wsPort });

  WSS.on('connection', (s) => {
    // TODO(jim): Suppport more than one client.
    client = s;

    s.on('close', function () {
      s.send(JSON.stringify({ action: null, data: 'closed' }));
    });

    s.send(JSON.stringify({ action: null, data: 'connected' }));
  });

  if (production) {
    server.use(compression());
  }

  server.use(Middleware.CORS);
  server.use('/public', express.static('public'));
  server.use(bodyParser.json());
  server.use(
    bodyParser.urlencoded({
      extended: false,
    })
  );

  server.post('/_/viewer', async (req, res) => {
    let data = state;

    if (!production) {
      const updates = await Utilities.refresh({ PG: PowerGate });
      const updatesWithToken = await Utilities.refreshWithToken({
        PG: PowerGate,
      });
      data = await Utilities.updateStateData(data, {
        ...updates,
        ...updatesWithToken,
      });
    }

    return res.status(200).send({ success: true, data });
  });

  server.post('/_/deals/storage', async (req, res) => {
    if (Strings.isEmpty(req.body.src)) {
      return res.status(500).send({ success: false });
    }

    const localPath = `.${req.body.src}`;
    const buffer = FS.readFileSync(localPath);
    const { cid } = await PowerGate.ffs.addToHot(buffer);
    const { jobId } = await PowerGate.ffs.pushConfig(cid);

    // TODO(jim): Refactor this so we repeat this less often.
    let write = false;
    for (let i = 0; i < state.library.length; i++) {
      for (let j = 0; j < state.library[i].children.length; j++) {
        if (localPath === state.library[i].children[j].id) {
          state.library[i].children[j].job_id = jobId;
          state.library[i].children[j].cid = cid;
          state.library[i].children[j].storage_status = 1;
          write = true;
        }
      }
    }

    if (write) {
      FS.writeFileSync('./.data/library.json', JSON.stringify({ library: state.library }));
    }

    state = await Utilities.emitState({ state, client, PG: PowerGate });
    return res.status(200).send({ success: true, cid, jobId });
  });

  server.post('/_/storage/:file', async (req, res) => {
    const form = formidable({
      multiples: true,
      uploadDir: Constants.FILE_STORAGE_URL,
    });

    form.parse(req, async (error, fields, files) => {
      if (error) {
        return res.status(500).send({ error });
      } else {
        // TODO(jim): Need to support other file types.
        if (!files.image) {
          console.error('[ prototype ] File type unspported', files);
          return res.status(500).send({ error: 'File type unsupported', files });
        }

        const newPath = form.uploadDir + req.params.file;
        FS.rename(files.image.path, newPath, function (err) {});

        const localFile = Utilities.createFile({
          id: newPath,
          data: files.image,
        });

        // TODO(jim): Messy, refactor.
        let pushed = false;
        for (let i = 0; i < state.library.length; i++) {
          if (!pushed) {
            state.library[i].children.push(localFile);
            pushed = true;
            break;
          }
        }

        if (pushed) {
          FS.writeFileSync('./.data/library.json', JSON.stringify({ library: state.library }));
        }

        state = await Utilities.emitState({
          state,
          client,
          PG: PowerGate,
        });

        return res.status(200).send({ success: true, file: localFile });
      }
    });
  });

  server.post('/_/upload/avatar', async (req, res) => {
    const form = formidable({
      multiples: true,
      uploadDir: Constants.AVATAR_STORAGE_URL,
    });

    form.parse(req, async (error, fields, files) => {
      if (error) {
        return res.status(500).send({ error });
      } else {
        const newPath = form.uploadDir + 'avatar.png';
        FS.rename(files.image.path, newPath, function (err) {});

        state = await Utilities.emitState({
          state,
          client,
          PG: PowerGate,
        });
        return res.status(200).send({ success: true });
      }
    });
  });

  server.post('/_/settings', async (req, res) => {
    let data;
    try {
      data = await PowerGate.ffs.setDefaultConfig(req.body.config);
    } catch (e) {
      return res.status(500).send({ error: e.message });
    }

    state = await Utilities.emitState({ state, client, PG: PowerGate });
    return res.status(200).send({ success: true, data });
  });

  server.post('/_/wallet/create', async (req, res) => {
    let data;
    try {
      data = await PowerGate.ffs.newAddr(req.body.name, req.body.type, req.body.makeDefault);
    } catch (e) {
      return res.status(500).send({ error: e.message });
    }

    state = await Utilities.emitState({ state, client, PG: PowerGate });
    return res.status(200).send({ success: true, data });
  });

  server.post('/_/wallet/send', async (req, res) => {
    let data;
    try {
      data = await PowerGate.ffs.sendFil(req.body.source, req.body.target, req.body.amount);
    } catch (e) {
      return res.status(500).send({ error: e.message });
    }

    state = await Utilities.emitState({ state, client, PG: PowerGate });
    return res.status(200).send({ success: true, data: { ...data, ...req.body } });
  });

  server.get('/', async (req, res) => {
    if (production || !state.token) {
      return res.redirect('/system');
    }

    return app.render(req, res, '/', {
      production,
      wsPort,
    });
  });

  server.get('*', async (req, res) => {
    return nextRequestHandler(req, res, req.url);
  });

  server.listen(port, async (err) => {
    if (err) {
      throw err;
    }

    console.log(`[ prototype ] client: http://localhost:${port}`);
    console.log(`[ prototype ] constants:`, Constants);

    await setIntervalViewerUpdatesUnsafe();
  });
});