From 4853df2f7a91c4a4e34351a4afbe60dd4bbf1b44 Mon Sep 17 00:00:00 2001
From: Sahat Yalkabov <sakhat@gmail.com>
Date: Tue, 15 Apr 2014 11:47:54 -0400
Subject: [PATCH] Add/remove linkedin generated app routes from app.js

---
 generator.js | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/generator.js b/generator.js
index 973312e871..8e33ea7264 100644
--- a/generator.js
+++ b/generator.js
@@ -743,6 +743,11 @@ inquirer.prompt({
           userModel.splice(index - 1, 0, linkedinModel);
           fs.writeFileSync(userModelFile, userModel.join('\n'));
 
+          // Add LinkedIn to app.js
+          index = app.indexOf(' * OAuth routes for sign-in.');
+          app.splice(index + 3, 0, linkedinRoutes);
+          fs.writeFileSync(appFile, app.join('\n'));
+
           console.log('✓ LinkedIn authentication has been added.'.info);
         } else {
           console.log('✓ LinkedIn authentication is already active.'.data);
@@ -769,6 +774,11 @@ inquirer.prompt({
         profileTemplate.splice(index - 1, 5);
         fs.writeFileSync(profileTemplateFile, profileTemplate.join('\n'));
 
+        // Remove LinkedIn from app.js
+        index = app.indexOf("app.get('/auth/linkedin', passport.authenticate('linkedin', { state: 'SOME STATE' }));");
+        app.splice(index, 4);
+        fs.writeFileSync(appFile, app.join('\n'));
+
         // Remove LinkedIn from User.js
         index = userModel.indexOf('  linkedin: String,');
         userModel.splice(index, 1);