Merge pull request #384 from lbryio/speech-as-a-service-constructor

Speech as a service constructor
This commit is contained in:
Bill Bittner 2018-03-08 22:24:42 -08:00 committed by GitHub
commit 34603fbcd6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 205 additions and 207 deletions

View file

@ -1,4 +1,3 @@
// const db = require('../models'); // require our models for syncing
const logger = require('winston'); const logger = require('winston');
module.exports = { module.exports = {

View file

@ -1,9 +1,7 @@
const config = require('../config/speechConfig.js');
const logger = require('winston'); const logger = require('winston');
module.exports = function () { module.exports = (config) => {
// get the config file // get the config file
for (let configCategoryKey in config) { for (let configCategoryKey in config) {
if (config.hasOwnProperty(configCategoryKey)) { if (config.hasOwnProperty(configCategoryKey)) {
// get the final variables for each config category // get the final variables for each config category

View file

@ -1,7 +1,6 @@
const axios = require('axios'); const axios = require('axios');
const logger = require('winston'); const logger = require('winston');
const config = require('../config/speechConfig.js'); const { api: { apiHost, apiPort } } = require('../config/speechConfig.js');
const { apiHost, apiPort } = config.api;
const lbryApiUri = 'http://' + apiHost + ':' + apiPort; const lbryApiUri = 'http://' + apiHost + ':' + apiPort;
const { chooseGaLbrynetPublishLabel, sendGATimingEvent } = require('./googleAnalytics.js'); const { chooseGaLbrynetPublishLabel, sendGATimingEvent } = require('./googleAnalytics.js');

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

View file

@ -1,65 +1,66 @@
const PassportLocalStrategy = require('passport-local').Strategy; const PassportLocalStrategy = require('passport-local').Strategy;
const db = require('../models');
const logger = require('winston'); const logger = require('winston');
function returnUserAndChannelInfo (userInstance) { module.exports = (db) => {
return new Promise((resolve, reject) => { const returnUserAndChannelInfo = (userInstance) => {
let userInfo = {}; return new Promise((resolve, reject) => {
userInfo['id'] = userInstance.id; let userInfo = {};
userInfo['userName'] = userInstance.userName; userInfo['id'] = userInstance.id;
userInstance userInfo['userName'] = userInstance.userName;
.getChannel() userInstance
.then(({channelName, channelClaimId}) => { .getChannel()
userInfo['channelName'] = channelName; .then(({channelName, channelClaimId}) => {
userInfo['channelClaimId'] = channelClaimId; userInfo['channelName'] = channelName;
return db.Certificate.getShortChannelIdFromLongChannelId(channelClaimId, channelName); userInfo['channelClaimId'] = channelClaimId;
}) return db.Certificate.getShortChannelIdFromLongChannelId(channelClaimId, channelName);
.then(shortChannelId => { })
userInfo['shortChannelId'] = shortChannelId; .then(shortChannelId => {
resolve(userInfo); userInfo['shortChannelId'] = shortChannelId;
}) resolve(userInfo);
.catch(error => { })
reject(error); .catch(error => {
}); reject(error);
}); });
} });
};
module.exports = new PassportLocalStrategy( return new PassportLocalStrategy(
{ {
usernameField: 'username', usernameField: 'username',
passwordField: 'password', passwordField: 'password',
}, },
(username, password, done) => { (username, password, done) => {
return db.User return db.User
.findOne({ .findOne({
where: {userName: username}, where: {userName: username},
}) })
.then(user => { .then(user => {
if (!user) { if (!user) {
logger.debug('no user found'); logger.debug('no user found');
return done(null, false, {message: 'Incorrect username or password'}); return done(null, false, {message: 'Incorrect username or password'});
} }
return user.comparePassword(password) return user.comparePassword(password)
.then(isMatch => { .then(isMatch => {
if (!isMatch) { if (!isMatch) {
logger.debug('incorrect password'); logger.debug('incorrect password');
return done(null, false, {message: 'Incorrect username or password'}); return done(null, false, {message: 'Incorrect username or password'});
} }
logger.debug('Password was a match, returning User'); logger.debug('Password was a match, returning User');
return returnUserAndChannelInfo(user) return returnUserAndChannelInfo(user)
.then(userInfo => { .then(userInfo => {
return done(null, userInfo); return done(null, userInfo);
}) })
.catch(error => { .catch(error => {
return error; return error;
}); });
}) })
.catch(error => { .catch(error => {
return error; return error;
}); });
}) })
.catch(error => { .catch(error => {
return done(error); return done(error);
}); });
} },
); );
};

View file

@ -1,64 +1,65 @@
const db = require('../models');
const PassportLocalStrategy = require('passport-local').Strategy; const PassportLocalStrategy = require('passport-local').Strategy;
const lbryApi = require('../helpers/lbryApi.js'); const lbryApi = require('../helpers/lbryApi.js');
const logger = require('winston'); const logger = require('winston');
module.exports = new PassportLocalStrategy( module.exports = (db) => {
{ return new PassportLocalStrategy(
usernameField: 'username', {
passwordField: 'password', usernameField: 'username',
}, passwordField: 'password',
(username, password, done) => { },
logger.verbose(`new channel signup request. user: ${username} pass: ${password} .`); (username, password, done) => {
let userInfo = {}; logger.verbose(`new channel signup request. user: ${username} pass: ${password} .`);
// server-side validaton of inputs (username, password) let userInfo = {};
// server-side validaton of inputs (username, password)
// create the channel and retrieve the metadata // create the channel and retrieve the metadata
return lbryApi.createChannel(`@${username}`) return lbryApi.createChannel(`@${username}`)
.then(tx => { .then(tx => {
// create user record // create user record
const userData = { const userData = {
userName: username, userName: username,
password: password, password: password,
}; };
logger.verbose('userData >', userData); logger.verbose('userData >', userData);
// create user record // create user record
const channelData = { const channelData = {
channelName : `@${username}`, channelName : `@${username}`,
channelClaimId: tx.claim_id, channelClaimId: tx.claim_id,
}; };
logger.verbose('channelData >', channelData); logger.verbose('channelData >', channelData);
// create certificate record // create certificate record
const certificateData = { const certificateData = {
claimId: tx.claim_id, claimId: tx.claim_id,
name : `@${username}`, name : `@${username}`,
// address, // address,
}; };
logger.verbose('certificateData >', certificateData); logger.verbose('certificateData >', certificateData);
// save user and certificate to db // save user and certificate to db
return Promise.all([db.User.create(userData), db.Channel.create(channelData), db.Certificate.create(certificateData)]); return Promise.all([db.User.create(userData), db.Channel.create(channelData), db.Certificate.create(certificateData)]);
}) })
.then(([newUser, newChannel, newCertificate]) => { .then(([newUser, newChannel, newCertificate]) => {
logger.verbose('user and certificate successfully created'); logger.verbose('user and certificate successfully created');
// store the relevant newUser info to be passed back for req.User // store the relevant newUser info to be passed back for req.User
userInfo['id'] = newUser.id; userInfo['id'] = newUser.id;
userInfo['userName'] = newUser.userName; userInfo['userName'] = newUser.userName;
userInfo['channelName'] = newChannel.channelName; userInfo['channelName'] = newChannel.channelName;
userInfo['channelClaimId'] = newChannel.channelClaimId; userInfo['channelClaimId'] = newChannel.channelClaimId;
// associate the instances // associate the instances
return Promise.all([newCertificate.setChannel(newChannel), newChannel.setUser(newUser)]); return Promise.all([newCertificate.setChannel(newChannel), newChannel.setUser(newUser)]);
}) })
.then(() => { .then(() => {
logger.verbose('user and certificate successfully associated'); logger.verbose('user and certificate successfully associated');
return db.Certificate.getShortChannelIdFromLongChannelId(userInfo.channelClaimId, userInfo.channelName); return db.Certificate.getShortChannelIdFromLongChannelId(userInfo.channelClaimId, userInfo.channelName);
}) })
.then(shortChannelId => { .then(shortChannelId => {
userInfo['shortChannelId'] = shortChannelId; userInfo['shortChannelId'] = shortChannelId;
return done(null, userInfo); return done(null, userInfo);
}) })
.catch(error => { .catch(error => {
logger.error('signup error', error); logger.error('signup error', error);
return done(error); return done(error);
}); });
} }
); );
};

158
server.js
View file

@ -4,94 +4,94 @@ const bodyParser = require('body-parser');
const expressHandlebars = require('express-handlebars'); const expressHandlebars = require('express-handlebars');
const Handlebars = require('handlebars'); const Handlebars = require('handlebars');
const { populateLocalsDotUser, serializeSpeechUser, deserializeSpeechUser } = require('./helpers/authHelpers.js'); const { populateLocalsDotUser, serializeSpeechUser, deserializeSpeechUser } = require('./helpers/authHelpers.js');
const config = require('./config/speechConfig.js'); const { logging: { logLevel } } = require('./config/speechConfig.js');
const logger = require('winston'); const logger = require('winston');
const helmet = require('helmet'); const helmet = require('helmet');
const PORT = 3000; // set port
const app = express(); // create an Express application const app = express(); // create an Express application
const passport = require('passport'); const passport = require('passport');
const cookieSession = require('cookie-session'); const cookieSession = require('cookie-session');
// configure logging // configure logging
const logLevel = config.logging.logLevel;
require('./config/loggerConfig.js')(logger, logLevel); require('./config/loggerConfig.js')(logger, logLevel);
require('./config/slackConfig.js')(logger); require('./config/slackConfig.js')(logger);
// check for global config variables function SpeechServer (config) {
require('./helpers/configVarCheck.js')(); this.mysqlConfig = config.mysql;
this.siteConfig = config.siteConfig;
// trust the proxy to get ip address for us this.lbrynetConfig = config.lbrynetConfig;
app.enable('trust proxy'); this.db = require('./models')(config.mysqlConfig);
this.PORT = 3000;
// add middleware this.speak = (something) => {
app.use(helmet()); // set HTTP headers to protect against well-known web vulnerabilties
app.use(express.static(`${__dirname}/public`)); // 'express.static' to serve static files from public directory
app.use(bodyParser.json()); // 'body parser' for parsing application/json
app.use(bodyParser.urlencoded({ extended: true })); // 'body parser' for parsing application/x-www-form-urlencoded
app.use((req, res, next) => { // custom logging middleware to log all incoming http requests
logger.verbose(`Request on ${req.originalUrl} from ${req.ip}`);
next();
});
// configure passport
passport.serializeUser(serializeSpeechUser);
passport.deserializeUser(deserializeSpeechUser);
const localSignupStrategy = require('./passport/local-signup.js');
const localLoginStrategy = require('./passport/local-login.js');
passport.use('local-signup', localSignupStrategy);
passport.use('local-login', localLoginStrategy);
// initialize passport
app.use(cookieSession({
name : 'session',
keys : [config.session.sessionKey],
maxAge: 24 * 60 * 60 * 1000, // 24 hours
}));
app.use(passport.initialize());
app.use(passport.session());
// configure handlebars & register it with express app
const hbs = expressHandlebars.create({
defaultLayout: 'embed', // sets the default layout
handlebars : Handlebars, // includes basic handlebars for access to that library
});
app.engine('handlebars', hbs.engine);
app.set('view engine', 'handlebars');
// middleware to pass user info back to client (for handlebars access), if user is logged in
app.use(populateLocalsDotUser);
// start the server
const start = (config) => {
const { mysqlConfig } = config;
const db = require('./models')(mysqlConfig); // require our models for syncing
db.sequelize
// sync sequelize
.sync()
// require routes
.then(() => {
require('./routes/auth-routes.js')(app);
require('./routes/api-routes.js')(app);
require('./routes/page-routes.js')(app);
require('./routes/serve-routes.js')(app);
require('./routes/fallback-routes.js')(app);
const http = require('http');
return http.Server(app);
})
// start the server
.then(server => {
server.listen(PORT, () => {
logger.info('Trusting proxy?', app.get('trust proxy'));
logger.info(`Server is listening on PORT ${PORT}`);
});
})
.catch((error) => {
logger.error(`Startup Error:`, error);
});
};
module.exports = {
speak (something) {
console.log(something); console.log(something);
}, };
start, this.start = () => {
// trust the proxy to get ip address for us
app.enable('trust proxy');
// add middleware
app.use(helmet()); // set HTTP headers to protect against well-known web vulnerabilties
app.use(express.static(`${__dirname}/public`)); // 'express.static' to serve static files from public directory
app.use(bodyParser.json()); // 'body parser' for parsing application/json
app.use(bodyParser.urlencoded({ extended: true })); // 'body parser' for parsing application/x-www-form-urlencoded
app.use((req, res, next) => { // custom logging middleware to log all incoming http requests
logger.verbose(`Request on ${req.originalUrl} from ${req.ip}`);
next();
});
// configure passport
passport.serializeUser(serializeSpeechUser);
passport.deserializeUser(deserializeSpeechUser);
const localSignupStrategy = require('./passport/local-signup.js')(this.db);
const localLoginStrategy = require('./passport/local-login.js')(this.db);
passport.use('local-signup', localSignupStrategy);
passport.use('local-login', localLoginStrategy);
// initialize passport
app.use(cookieSession({
name : 'session',
keys : [this.siteConfig.session.sessionKey],
maxAge: 24 * 60 * 60 * 1000, // i.e. 24 hours
}));
app.use(passport.initialize());
app.use(passport.session());
// configure handlebars & register it with express app
const hbs = expressHandlebars.create({
defaultLayout: 'embed',
handlebars : Handlebars,
});
app.engine('handlebars', hbs.engine);
app.set('view engine', 'handlebars');
// middleware to pass user info back to client (for handlebars access), if user is logged in
app.use(populateLocalsDotUser); // note: I don't think I need this any more?
// set the routes on the app
require('./routes/auth-routes.js')(app);
require('./routes/api-routes.js')(app);
require('./routes/page-routes.js')(app);
require('./routes/serve-routes.js')(app);
require('./routes/fallback-routes.js')(app);
const http = require('http');
const server = http.Server(app);
this.startServer(server);
};
this.startServer = (server) => {
// print config variables
require('./helpers/configVarCheck.js')(this.config);
this.db.sequelize
// sync sequelize
.sync()
// start the server
.then(() => {
server.listen(this.PORT, () => {
logger.info(`Server is listening on PORT ${this.PORT}`);
});
})
.catch((error) => {
logger.error(`Startup Error:`, error);
});
};
}; };
module.exports = SpeechServer;