2017-02-01 18:31:09 -06:00
import dotenv from 'dotenv'
import express from 'express'
import redis from 'redis'
import pg from 'pg'
2017-02-02 06:56:14 -06:00
import log from 'npmlog'
2017-02-01 18:31:09 -06:00
2017-02-02 09:11:36 -06:00
const env = process . env . NODE _ENV || 'development'
dotenv . config ( {
path : env === 'production' ? '.env.production' : '.env'
} )
2017-02-01 18:31:09 -06:00
const pgConfigs = {
development : {
database : 'mastodon_development' ,
host : '/var/run/postgresql' ,
max : 10
} ,
production : {
user : process . env . DB _USER || 'mastodon' ,
password : process . env . DB _PASS || '' ,
database : process . env . DB _NAME || 'mastodon_production' ,
host : process . env . DB _HOST || 'localhost' ,
port : process . env . DB _PORT || 5432 ,
max : 10
}
}
const app = express ( )
const pgPool = new pg . Pool ( pgConfigs [ env ] )
const authenticationMiddleware = ( req , res , next ) => {
const authorization = req . get ( 'Authorization' )
if ( ! authorization ) {
err = new Error ( 'Missing access token' )
err . statusCode = 401
return next ( err )
}
const token = authorization . replace ( /^Bearer / , '' )
pgPool . connect ( ( err , client , done ) => {
if ( err ) {
2017-02-02 06:56:14 -06:00
log . error ( err )
2017-02-01 18:31:09 -06:00
return next ( err )
}
2017-02-02 08:20:31 -06:00
client . query ( 'SELECT oauth_access_tokens.resource_owner_id, users.account_id FROM oauth_access_tokens INNER JOIN users ON oauth_access_tokens.resource_owner_id = users.id WHERE oauth_access_tokens.token = $1 LIMIT 1' , [ token ] , ( err , result ) => {
2017-02-01 18:31:09 -06:00
done ( )
if ( err ) {
2017-02-02 06:56:14 -06:00
log . error ( err )
2017-02-01 18:31:09 -06:00
return next ( err )
}
if ( result . rows . length === 0 ) {
err = new Error ( 'Invalid access token' )
err . statusCode = 401
return next ( err )
}
req . accountId = result . rows [ 0 ] . account _id
next ( )
} )
} )
}
const errorMiddleware = ( err , req , res , next ) => {
res . writeHead ( err . statusCode || 500 , { 'Content-Type' : 'application/json' } )
2017-02-02 06:56:14 -06:00
res . end ( JSON . stringify ( { error : err . statusCode ? ` ${ err } ` : 'An unexpected error occured' } ) )
2017-02-01 18:31:09 -06:00
}
2017-02-02 06:56:14 -06:00
const streamFrom = ( id , req , res , needsFiltering = false ) => {
log . verbose ( ` Starting stream from ${ id } for ${ req . accountId } ` )
2017-02-01 18:31:09 -06:00
res . setHeader ( 'Content-Type' , 'text/event-stream' )
res . setHeader ( 'Transfer-Encoding' , 'chunked' )
2017-02-02 08:29:41 -06:00
const redisClient = redis . createClient ( {
host : process . env . REDIS _HOST || '127.0.0.1' ,
port : process . env . REDIS _PORT || 6379 ,
password : process . env . REDIS _PASSWORD
} )
2017-02-01 18:31:09 -06:00
redisClient . on ( 'message' , ( channel , message ) => {
const { event , payload } = JSON . parse ( message )
2017-02-02 06:56:14 -06:00
if ( needsFiltering ) {
pgPool . connect ( ( err , client , done ) => {
if ( err ) {
log . error ( err )
return
}
const unpackedPayload = JSON . parse ( payload )
const targetAccountIds = [ unpackedPayload . account . id ] + unpackedPayload . mentions . map ( item => item . id ) + ( unpackedPayload . reblog ? unpackedPayload . reblog . account . id : [ ] )
client . query ( 'SELECT target_account_id FROM blocks WHERE account_id = $1 AND target_account_id IN ($2)' , [ req . accountId , targetAccountIds ] , ( err , result ) => {
done ( )
if ( err ) {
log . error ( err )
return
}
if ( result . rows . length > 0 ) {
return
}
res . write ( ` event: ${ event } \n ` )
res . write ( ` data: ${ payload } \n \n ` )
} )
} )
} else {
res . write ( ` event: ${ event } \n ` )
res . write ( ` data: ${ payload } \n \n ` )
}
2017-02-01 18:31:09 -06:00
} )
2017-02-02 08:20:31 -06:00
const heartbeat = setInterval ( ( ) => res . write ( ':thump\n' ) , 15000 )
req . on ( 'close' , ( ) => {
log . verbose ( ` Ending stream from ${ id } for ${ req . accountId } ` )
clearInterval ( heartbeat )
redisClient . quit ( )
} )
2017-02-01 18:31:09 -06:00
redisClient . subscribe ( id )
}
app . use ( authenticationMiddleware )
app . use ( errorMiddleware )
2017-02-02 06:56:14 -06:00
app . get ( '/api/v1/streaming/user' , ( req , res ) => streamFrom ( ` timeline: ${ req . accountId } ` , req , res ) )
app . get ( '/api/v1/streaming/public' , ( req , res ) => streamFrom ( 'timeline:public' , req , res , true ) )
app . get ( '/api/v1/streaming/hashtag' , ( req , res ) => streamFrom ( ` timeline:hashtag: ${ req . params . tag } ` , req , res , true ) )
log . level = 'verbose'
log . info ( ` Starting HTTP server on port ${ process . env . PORT || 4000 } ` )
2017-02-01 18:31:09 -06:00
2017-02-02 06:56:14 -06:00
app . listen ( process . env . PORT || 4000 )