算法评测平台前端。
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

238 lines
6.4 KiB

8 years ago
let fs = require('fs'),
path = require('path');
6 years ago
const serializejs = require('serialize-javascript');
8 years ago
const commandLineArgs = require('command-line-args');
const optionDefinitions = [
{ name: 'config', alias: 'c', type: String, defaultValue: './config.json' },
];
const options = commandLineArgs(optionDefinitions);
8 years ago
global.syzoj = {
7 years ago
rootDir: __dirname,
config: require('object-assign-deep')({}, require('./config-example.json'), require(options.config)),
languages: require('./language-config.json'),
configDir: options.config,
8 years ago
models: [],
modules: [],
db: null,
log(obj) {
console.log(obj);
},
async run() {
8 years ago
let Express = require('express');
global.app = Express();
syzoj.production = app.get('env') === 'production';
let winstonLib = require('./libs/winston');
winstonLib.configureWinston(!syzoj.production);
app.server = require('http').createServer(app);
if (!module.parent) {
// Loaded by `require()`, not node CLI.
app.server.listen(parseInt(syzoj.config.port), syzoj.config.hostname, () => {
this.log(`SYZOJ is listening on ${syzoj.config.hostname}:${parseInt(syzoj.config.port)}...`);
});
}
8 years ago
// Set assets dir
app.use(Express.static(__dirname + '/static', { maxAge: syzoj.production ? '1y' : 0 }));
8 years ago
// Set template engine ejs
app.set('view engine', 'ejs');
// Use body parser
let bodyParser = require('body-parser');
app.use(bodyParser.urlencoded({
extended: true,
limit: '50mb'
8 years ago
}));
app.use(bodyParser.json({ limit: '50mb' }));
8 years ago
// Use cookie parser
app.use(require('cookie-parser')());
6 years ago
app.locals.serializejs = serializejs;
8 years ago
let multer = require('multer');
app.multer = multer({ dest: syzoj.utils.resolvePath(syzoj.config.upload_dir, 'tmp') });
// This should before load api_v2, to init the `res.locals.user`
this.loadHooks();
// Trick to bypass CSRF for APIv2
app.use((() => {
let router = new Express.Router();
app.apiRouter = router;
require('./modules/api_v2');
return router;
})());
await this.connectDatabase();
if (!module.parent) {
await this.lib('judger').connect();
}
8 years ago
this.loadModules();
},
async connectDatabase() {
8 years ago
let Sequelize = require('sequelize');
let Op = Sequelize.Op;
let operatorsAliases = {
$eq: Op.eq,
$ne: Op.ne,
$gte: Op.gte,
$gt: Op.gt,
$lte: Op.lte,
$lt: Op.lt,
$not: Op.not,
$in: Op.in,
$notIn: Op.notIn,
$is: Op.is,
$like: Op.like,
$notLike: Op.notLike,
$iLike: Op.iLike,
$notILike: Op.notILike,
$regexp: Op.regexp,
$notRegexp: Op.notRegexp,
$iRegexp: Op.iRegexp,
$notIRegexp: Op.notIRegexp,
$between: Op.between,
$notBetween: Op.notBetween,
$overlap: Op.overlap,
$contains: Op.contains,
$contained: Op.contained,
$adjacent: Op.adjacent,
$strictLeft: Op.strictLeft,
$strictRight: Op.strictRight,
$noExtendRight: Op.noExtendRight,
$noExtendLeft: Op.noExtendLeft,
$and: Op.and,
$or: Op.or,
$any: Op.any,
$all: Op.all,
$values: Op.values,
$col: Op.col
};
8 years ago
this.db = new Sequelize(this.config.db.database, this.config.db.username, this.config.db.password, {
host: this.config.db.host,
6 years ago
dialect: 'mariadb',
logging: syzoj.production ? false : syzoj.log,
timezone: require('moment')().format('Z'),
operatorsAliases: operatorsAliases
8 years ago
});
global.Promise = Sequelize.Promise;
this.db.countQuery = async (sql, options) => (await this.db.query(`SELECT COUNT(*) FROM (${sql}) AS \`__tmp_table\``, options))[0][0]['COUNT(*)'];
this.loadModels();
8 years ago
},
loadModules() {
fs.readdir('./modules/', (err, files) => {
if (err) {
this.log(err);
return;
}
files.filter((file) => file.endsWith('.js'))
.forEach((file) => this.modules.push(require(`./modules/${file}`)));
});
},
loadModels() {
fs.readdir('./models/', (err, files) => {
if (err) {
this.log(err);
return;
}
files.filter((file) => file.endsWith('.js'))
.forEach((file) => require(`./models/${file}`));
this.db.sync();
});
},
7 years ago
lib(name) {
return require(`./libs/${name}`);
},
8 years ago
model(name) {
return require(`./models/${name}`);
8 years ago
},
loadHooks() {
let Session = require('express-session');
let FileStore = require('session-file-store')(Session);
8 years ago
let sessionConfig = {
secret: this.config.session_secret,
cookie: { httpOnly: false },
8 years ago
rolling: true,
saveUninitialized: true,
resave: true,
store: new FileStore
8 years ago
};
if (syzoj.production) {
8 years ago
app.set('trust proxy', 1);
sessionConfig.cookie.secure = false;
8 years ago
}
app.use(Session(sessionConfig));
app.use((req, res, next) => {
res.locals.useLocalLibs = !!parseInt(req.headers['syzoj-no-cdn']);
let User = syzoj.model('user');
8 years ago
if (req.session.user_id) {
User.fromID(req.session.user_id).then((user) => {
res.locals.user = user;
next();
8 years ago
}).catch((err) => {
this.log(err);
res.locals.user = null;
req.session.user_id = null;
next();
});
8 years ago
} else {
if (req.cookies.login) {
let obj;
try {
obj = JSON.parse(req.cookies.login);
User.findOne({
where: {
username: obj[0],
password: obj[1]
}
}).then(user => {
if (!user) throw null;
res.locals.user = user;
req.session.user_id = user.id;
next();
}).catch(err => {
console.log(err);
res.locals.user = null;
req.session.user_id = null;
next();
});
} catch (e) {
res.locals.user = null;
req.session.user_id = null;
next();
}
} else {
res.locals.user = null;
req.session.user_id = null;
next();
}
8 years ago
}
});
// Active item on navigator bar
app.use((req, res, next) => {
res.locals.active = req.path.split('/')[1];
next();
});
app.use((req, res, next) => {
res.locals.req = req;
res.locals.res = res;
next();
});
},
utils: require('./utility')
};
syzoj.run();