shell-mirror 1.0.0

package/server.js

@@ -0,0 +1,171 @@
+const express = require('express');
+const http = require('http');
+const WebSocket = require('ws');
+const pty = require('node-pty');
+const os = require('os');
+const path = require('path');
+const session = require('express-session');
+const passport = require('passport');
+
+// Load environment configuration and auth setup
+require('dotenv').config();
+require('./auth'); // Configure passport strategies
+
+const app = express();
+const server = http.createServer(app);
+const wss = new WebSocket.Server({ server });
+
+// 1. Session Configuration
+const sessionParser = session({
+  secret: process.env.SESSION_SECRET || 'a-secure-default-session-secret',
+  resave: false,
+  saveUninitialized: false,
+  cookie: { secure: process.env.NODE_ENV === 'production' } // Use secure cookies in production
+});
+app.use(sessionParser);
+
+// 2. Passport Initialization
+app.use(passport.initialize());
+app.use(passport.session());
+
+// Use the user's default shell.
+const shell = os.platform() === 'win32' ? 'powershell.exe' : (process.env.SHELL || 'bash');
+
+// --- Persistent Terminal Session ---
+const term = pty.spawn(shell, [], {
+  name: 'xterm-color',
+  cols: 80,
+  rows: 30,
+  cwd: process.cwd(),
+  env: process.env
+});
+
+// --- Terminal History Buffer ---
+let history = [];
+const MAX_HISTORY_LINES = 100;
+term.on('data', (data) => {
+    if (history.length > MAX_HISTORY_LINES * 2) {
+        history = history.slice(history.length - MAX_HISTORY_LINES);
+    }
+    history.push(data);
+});
+
+// Serve static files from 'public'
+app.use(express.static('public'));
+
+// Route for the terminal application (protected)
+app.get('/app', (req, res) => {
+  if (!req.isAuthenticated()) {
+    return res.redirect('/auth/google');
+  }
+  res.sendFile(path.join(__dirname, 'public', 'app', 'terminal.html'));
+});
+
+// 3. Authentication Routes
+app.get('/auth/google', passport.authenticate('google', { scope: ['profile', 'email'] }));
+
+app.get('/auth/google/callback', 
+  passport.authenticate('google', { failureRedirect: '/login-failed.html' }),
+  function(req, res) {
+    // Successful authentication, redirect home.
+    res.redirect('/');
+  }
+);
+
+app.get('/api/auth/status', (req, res) => {
+  if (req.isAuthenticated()) {
+    res.json({ authenticated: true, user: req.user });
+  } else {
+    res.json({ authenticated: false });
+  }
+});
+
+app.post('/api/auth/logout', (req, res) => {
+  req.logout(function(err) {
+    if (err) { return next(err); }
+    res.redirect('/');
+  });
+});
+
+// Environment validation
+const requiredEnvVars = ['BASE_URL', 'GOOGLE_CLIENT_ID', 'GOOGLE_CLIENT_SECRET', 'SESSION_SECRET'];
+const missingEnvVars = requiredEnvVars.filter(varName => !process.env[varName]);
+
+if (missingEnvVars.length > 0) {
+  console.error('❌ Missing required environment variables:');
+  missingEnvVars.forEach(varName => {
+    console.error(`   - ${varName}`);
+  });
+  console.error('\nPlease create a .env file based on .env.example and fill in the required values.');
+  console.error('See README.md for instructions on getting Google OAuth credentials.');
+  process.exit(1);
+}
+
+// 4. WebSocket Connection Handler with Authentication
+wss.on('connection', function connection(ws, request) {
+  console.log('New WebSocket connection attempt');
+  
+  // Parse session from the WebSocket request
+  sessionParser(request, {}, () => {
+    if (!request.session || !request.session.passport || !request.session.passport.user) {
+      console.log('❌ Unauthorized WebSocket connection rejected');
+      ws.close(1008, 'Authentication required');
+      return;
+    }
+    
+    console.log(`✅ Authenticated WebSocket connection established for user: ${request.session.passport.user.displayName}`);
+    
+    // Send terminal history to newly connected client
+    if (history.length > 0) {
+      const recentHistory = history.slice(-MAX_HISTORY_LINES).join('');
+      ws.send(recentHistory);
+    }
+    
+    // Forward terminal output to this WebSocket connection
+    const terminalDataHandler = (data) => {
+      if (ws.readyState === WebSocket.OPEN) {
+        ws.send(data);
+      }
+    };
+    term.on('data', terminalDataHandler);
+    
+    // Handle incoming WebSocket messages
+    ws.on('message', function message(data) {
+      try {
+        const message = JSON.parse(data);
+        
+        if (message.type === 'input' && message.data) {
+          // Forward user input to terminal
+          term.write(message.data);
+        } else if (message.type === 'resize' && message.cols && message.rows) {
+          // Resize terminal
+          term.resize(message.cols, message.rows);
+          console.log(`Terminal resized to ${message.cols}x${message.rows}`);
+        }
+      } catch (error) {
+        console.error('Invalid WebSocket message:', error);
+      }
+    });
+    
+    // Handle WebSocket disconnection
+    ws.on('close', function close() {
+      console.log('WebSocket connection closed');
+      term.removeListener('data', terminalDataHandler);
+    });
+    
+    // Handle WebSocket errors
+    ws.on('error', function error(err) {
+      console.error('WebSocket error:', err);
+      term.removeListener('data', terminalDataHandler);
+    });
+  });
+});
+
+const PORT = process.env.PORT || 3000;
+const HOST = process.env.HOST || '0.0.0.0';
+server.listen(PORT, HOST, () => {
+  console.log(`✅ Shell Mirror server is running on ${process.env.BASE_URL}`);
+  if (process.env.BASE_URL.includes('localhost')) {
+    console.log(`   Local access: http://localhost:${PORT}`);
+  }
+});