index.js

/**
 * Created by patrykmazurkiewicz on 15/06/2017.
 */
const express = require('express');
const http = require('http');
const bodyParser = require('body-parser');
const morgan = require('morgan');
const mongoose = require('mongoose');
const rateLimit = require('express-rate-limit');
const helmet = require('helmet');
const { execute, subscribe } = require('graphql');
const { graphqlExpress, graphiqlExpress } = require('apollo-server-express');
const { SubscriptionServer } = require('subscriptions-transport-ws');

const graphQl = require('./graphql');
const initFirebase = require('./services/firebaseApi').init;
const saveBeerWords = require('./services/firebaseApi').saveBeerWords;
const router = require('./router');
const websockets = require('./controllers/ws').websockets;
const getFirstTweetId = require('./services/dataScraping').getFirstTweet;
const scrape = require('./services/dataScraping').scrape;
const config = require('./config');

process.setMaxListeners(0);

const limiter = rateLimit({
  windowMs: 60 * 1000,
  max: 20,
});

//app
const app = express();
app.set('trust proxy', 1);
app.use(morgan('combined'));
app.use(helmet());
app.use(limiter);
app.use((req, res, next) => {
  res.header('Access-Control-Allow-Origin', '*');
  res.header(
    'Access-Control-Allow-Headers',
    'Origin, X-Requested-With, Content-Type, Accept',
  );
  res.header('Access-Control-Allow-Methods', 'GET,PUT,POST,DELETE');
  next();
});
app.use(bodyParser.json({ type: '*/*', limit: '10000kb' }));

//db

if (process.env.NODE_ENV === 'docker') {
  mongoose.connect('mongodb://mongo:auth/public-api');
} else if (process.env.NODE_ENV === 'local') {
  mongoose.connect('mongodb://localhost/public-api');
} else {
  mongoose.connect(config.mLabMyApi, {
    useMongoClient: true,
  });
}

router(app);

//server
const PORT = process.env.PORT || 3090;
const server = http.createServer(app);

//firebase
initFirebase();

//websocket
websockets(server);

//scraping
// getFirstTweetId(config.scrapingTargetUrl);
setInterval(() => {
  scrape('craft beer', 1, saveBeerWords);
}, config.scrapingInterval);

server.listen(PORT);

//graphql
SubscriptionServer.create(
  {
    schema: graphQl.schema,
    execute,
    subscribe,
  },
  {
    server: server,
    path: '/graphqlsubs',
  },
);
app.use(
  '/graphql',
  graphqlExpress({
    schema: graphQl.schema,
  }),
);
app.get(
  '/graphiql',
  graphiqlExpress({
    endpointURL: '/graphql',
    subscriptionsEndpoint: `wss://api.codebooyah.com/graphqlsubs`,
  }),
);
graphQl.listenForNewRank();

console.log('API listen on port: ', PORT);