1
const express = require('express');
2
const session = require('express-session');
3
const passport = require('passport');
4
const GitHubStrategy = require('passport-github2').Strategy;
5
const mongoose = require('mongoose');
6

7
// 连接数据库
8
mongoose.connect('mongodb://localhost:27017/github-auth');
9

10
// 定义用户模型
11
const User = mongoose.model('User', new mongoose.Schema({
12
  githubId: String,
13
  name: String,
14
  email: String,
15
  avatar: String
16
}));
17

18
// 配置GitHub策略
19
passport.use(new GitHubStrategy({
20
    clientID: process.env.GITHUB_CLIENT_ID || 'YOUR_CLIENT_ID',
21
    clientSecret: process.env.GITHUB_CLIENT_SECRET || 'YOUR_CLIENT_SECRET',
22
    callbackURL: 'http://localhost:3000/auth/github/callback'
23
  },
24
  function(accessToken, refreshToken, profile, done) {
25
    User.findOne({ githubId: profile.id }, function(err, user) {
26
      if (err) return done(err);
27
      if (user) {
28
        user.name = profile.displayName;
29
        user.email = profile.emails[0].value;
30
        user.avatar = profile.photos[0].value;
31
        user.save(function(err) {
32
          if (err) return done(err);
33
          return done(null, user);
34
        });
35
      } else {
36
        const newUser = new User({
37
          githubId: profile.id,
38
          name: profile.displayName,
39
          email: profile.emails[0].value,
40
          avatar: profile.photos[0].value
41
        });
42
        newUser.save(function(err) {
43
          if (err) return done(err);
44
          return done(null, newUser);
45
        });
46
      }
47
    });
48
  }
49
));
50

51
passport.serializeUser(function(user, done) {
52
  done(null, user.id);
53
});
54

55
passport.deserializeUser(function(id, done) {
56
  User.findById(id, function(err, user) {
57
    done(err, user);
58
  });
59
});
60

61
const app = express();
62

63
app.use(session({
64
  secret: 'your-secret-key',
65
  resave: false,
66
  saveUninitialized: false
67
}));
68

69
app.use(passport.initialize());
70
app.use(passport.session());
71

72
app.get('/', function(req, res) {
73
  res.send(`
74
    <h1>GitHub登录示例</h1>
75
    ${req.user ?
76
      `<p>已登录：${req.user.name}</p><img src="${req.user.avatar}" width="100"><br><a href="/logout">登出</a>` :
77
      '<a href="/auth/github">用GitHub登录</a>'
78
    }
79
  `);
80
});
81

82
app.get('/auth/github',
83
  passport.authenticate('github', { scope: ['user:email'] }));
84

85
app.get('/auth/github/callback',
86
  passport.authenticate('github', { failureRedirect: '/' }),
87
  function(req, res) {
88
    res.redirect('/');
89
  });
90

91
app.get('/logout', function(req, res) {
92
  req.logout();
93
  res.redirect('/');
94
});
95

96
app.listen(3000, function() {
97
  console.log('服务器运行在 http://localhost:3000');
98
});