Skip to content

Rework in to a module that can be imported to other projects #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
stefanhayden wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Rework in to a module that can be imported to other projects #5

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,2 @@
config/config.js
node_modules/
npm-debug.log
tweets.txt
12 changes: 9 additions & 3 deletions bot.js
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/**
* nodeEbot v.0.2.0!
* nodeEbot v.0.2.0!
* A twitter_ebooks style bot for Node.js
* by Dave Schumaker (@davely)
* https://github.com/daveschumaker/nodeEbot
Expand All @@ -12,7 +12,13 @@
* Import the robot component.
* This is where all the business logic is handeled.
*/
var robot = require('./components/robot');
var _ = require('lodash');
var defaultConfig = require('./config');
var Robot = require('./components/robot');

// Fire up the robot!
robot.init();
module.exports = function(userConfig) {
config = _.merge(defaultConfig, userConfig);

return new Robot(config)
}
1,104 changes: 553 additions & 551 deletions components/generator/index.js
View file
Open in desktop

Large diffs are not rendered by default.

239 changes: 122 additions & 117 deletions components/robot/index.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,137 +4,142 @@
**/

// Import required npm modules to make our robot work!
var path = require('path');
var fs = require('fs');
var Promise = require('bluebird');
fs = Promise.promisifyAll(fs);

//////////
var config = require('../../config'); // Robot config, personality settings and API keys
var generator = require('../generator'); // Compiles word dictionary and builds new sentences.
var tweet = require('../tweets'); // Methods to interact with Twitter by writing and favoriting tweets and more.
var Generator = require('../generator'); // Compiles word dictionary and builds new sentences.
var Tweet = require('../tweets'); // Methods to interact with Twitter by writing and favoriting tweets and more.
var utils = require('../utilities'); // Various helper functions

// Create promises
fs = Promise.promisifyAll(fs);
generator = Promise.promisifyAll(generator);

////// Start up tasks //////

// Process a provided list of stop words.
generator.stopwords = fs.readFileAsync('./data/stopwords.txt').toString().split("\n");

// Filename to source or tweets and other content from?
tweetFile = './tweets.txt';
var Robot = function Robot(config) {
this.generator = new Generator(config);
this.generator = Promise.promisifyAll(this.generator);
this.tweet = new Tweet(this.generator, config);
this.config = config;

// Track times of various actions from our robot
var robotActions = {
lastFollowCheck: 0,
lastRandomReply: 0,
lastReply: 0,
lastRetweet: 0,
lastTweet: 0
};
// Process a provided list of stop words.
this.generator.stopwords = fs.readFileAsync(path.join(__dirname, '../../data/stopwords.txt')).toString().split("\n");

module.exports = {
// Filename to source or tweets and other content from?
tweetFile = this.config.settings.corpus || path.join(__dirname, '../../tweets.sample.txt');

// Track times of various actions from our robot
this.robotActions = {
lastFollowCheck: 0,
lastRandomReply: 0,
lastReply: 0,
lastRetweet: 0,
lastTweet: 0
};
// Initialize robot and start doing robot things.
// Example, build word dictionary. Start watching stream, etc.
init: function() {
// Load up robot settings.
console.log('\n\nNODEEBOT FOR NODEJS v.0.2.0');
console.log('by Dave Schumaker (@davely)\n');
console.log('-== CONFIG SETTINGS ==-');
console.log(' -Post to Twitter? ' + config.settings.postTweets);
console.log(' -Repond to DMs? ' + config.settings.respondDMs);
console.log(' -Repond to replies? ' + config.settings.respondReplies);
console.log(' -Random replies? ' + config.settings.randomReplies);
console.log(' -Follow new users? ' + config.settings.followUsers);
console.log(' -Mark tweets as favorites? ' + config.settings.canFavoriteTweets);
console.log(' -Tweet interval: ' + config.settings.postInterval + ' seconds');
console.log('\nAnalyzing data and creating word corpus from file \'' + tweetFile + '\'');
console.log('(This may take a few minutes to generate...)');

// Set proper context
var self = this;

// Load in text file containing raw Tweet data.
fs.readFileAsync(tweetFile)
.then(function(fileContents) {
//Split content into array, separating by line.
var content = fileContents.toString().split("\n");
return content;
})
.then(function(content){
//Build word corpus using content array above.
return generator.buildCorpus(content);
})
.then(function(data){
// Once word dictionary is built, kick off the robots actions!
self.onBoot();

/*
* There may be a better way to handle this. Right now,
* this interval runs every 5 seconds and calls the
* robotTasks function, which handles the logic that checks
* if it's time to send a new tweet, reload the Twitter stream
* etc.
*/
setInterval(function() {
self.robotTasks();
}, 5000);
});
},

/*
* These are tasks the robot should do the first time it loads up.
*/
onBoot: function() {
// Start watching the Twitter stream.
tweet.watchStream();

// Check for new followers.
if (config.settings.followUsers) {
robotActions.lastFollowCheck = Math.floor(Date.now() / 1000);
tweet.getFollowers();
}

// Check if the robot is allowed to tweet on startup.
// If so, post a tweet!
if (config.settings.tweetOnStartup) {
robotActions.lastTweet = Math.floor(Date.now() / 1000); // Update time of the last tweet.
newTweet = generator.makeTweet(140); // Create a new tweet.
tweet.postNewTweet(newTweet); // Post tweet.
console.log(utils.currentTime(), newTweet + '');
}
},

robotTasks: function() {
/*
* Check how long it's been since robot last tweeted.
* If amount of time is greater than the tweet interval
* defined in the config file. Go ahead and post a new tweet.
*/
if (Math.floor(Date.now() / 1000) - robotActions.lastTweet >= config.settings.postInterval) {
robotActions.lastTweet = Math.floor(Date.now() / 1000); // Update time of the last tweet.
newTweet = generator.makeTweet(140); // Create a new tweet.
tweet.postNewTweet(newTweet); // Post tweet.
console.log(utils.currentTime(), newTweet + '');
}

/*
* Check if the Twitter Stream dropped. If so, reinitialize it.
*/
tweet.checkStream();

/*
* Check for new followers
*/
if (config.settings.followUsers && Math.floor(Date.now() / 1000) - robotActions.lastFollowCheck >= 300) {
robotActions.lastFollowCheck = Math.floor(Date.now() / 1000);
tweet.getFollowers();
}
}

};

// Load up robot settings.
console.log('\n\nNODEEBOT FOR NODEJS v.0.2.0');
console.log('by Dave Schumaker (@davely)\n');
console.log('-== CONFIG SETTINGS ==-');
console.log(' -Post to Twitter? ' + config.settings.postTweets);
console.log(' -Repond to DMs? ' + config.settings.respondDMs);
console.log(' -Repond to replies? ' + config.settings.respondReplies);
console.log(' -Random replies? ' + config.settings.randomReplies);
console.log(' -Follow new users? ' + config.settings.followUsers);
console.log(' -Mark tweets as favorites? ' + config.settings.canFavoriteTweets);
console.log(' -Tweet interval: ' + config.settings.postInterval + ' seconds');
console.log('\nAnalyzing data and creating word corpus from file \'' + tweetFile + '\'');
console.log('(This may take a few minutes to generate...)');

// Set proper context
var self = this;

// Load in text file containing raw Tweet data.
fs.readFileAsync(tweetFile)
.then(function(fileContents) {
//Split content into array, separating by line.
var content = fileContents.toString().split("\n");
return content;
})
.then(function(content){
//Build word corpus using content array above.
return self.generator.buildCorpus(content);
})
.then(function(data){
// Once word dictionary is built, kick off the robots actions!
self.onBoot();

console.log('I\m alive and waiting for tasks!');

/*
* There may be a better way to handle this. Right now,
* this interval runs every 5 seconds and calls the
* robotTasks function, which handles the logic that checks
* if it's time to send a new tweet, reload the Twitter stream
* etc.
*/
setInterval(function() {
self.robotTasks();
}, 5000);
});

}

/*
* These are tasks the robot should do the first time it loads up.
*/
Robot.prototype.onBoot = function() {
// Start watching the Twitter stream.
this.tweet.watchStream();

// Check for new followers.
if (this.config.settings.followUsers) {
this.robotActions.lastFollowCheck = Math.floor(Date.now() / 1000);
this.tweet.getFollowers();
}

// Check if the robot is allowed to tweet on startup.
// If so, post a tweet!
if (this.config.settings.tweetOnStartup) {
this.robotActions.lastTweet = Math.floor(Date.now() / 1000); // Update time of the last tweet.
newTweet = this.generator.makeTweet(140); // Create a new tweet.
this.tweet.postNewTweet(newTweet); // Post tweet.
console.log(utils.currentTime(), newTweet + '');
}
}

Robot.prototype.robotTasks = function() {
/*
* Check how long it's been since robot last tweeted.
* If amount of time is greater than the tweet interval
* defined in the config file. Go ahead and post a new tweet.
*/
if (Math.floor(Date.now() / 1000) - this.robotActions.lastTweet >= this.config.settings.postInterval) {
this.robotActions.lastTweet = Math.floor(Date.now() / 1000); // Update time of the last tweet.
newTweet = this.generator.makeTweet(140); // Create a new tweet.
this.tweet.postNewTweet(newTweet); // Post tweet.
console.log(utils.currentTime(), newTweet + '');
}

/*
* Check if the Twitter Stream dropped. If so, reinitialize it.
*/
this.tweet.checkStream();

/*
* Check for new followers
*/
if (this.config.settings.followUsers && Math.floor(Date.now() / 1000) - this.robotActions.lastFollowCheck >= 300) {
this.robotActions.lastFollowCheck = Math.floor(Date.now() / 1000);
this.tweet.getFollowers();
}
}

module.exports = Robot;


///// DEBUG STUFF
Expand All @@ -144,4 +149,4 @@ module.exports = {
// user: {
// screen_name: 'fakeuser',
// }
// };
// };
Loading