Menci
8 years ago
17 changed files with 527 additions and 274 deletions
@ -0,0 +1,115 @@
|
||||
/* |
||||
* This file is part of SYZOJ. |
||||
* |
||||
* Copyright (c) 2016 Menci <huanghaorui301@gmail.com> |
||||
* |
||||
* SYZOJ is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as |
||||
* published by the Free Software Foundation, either version 3 of the |
||||
* License, or (at your option) any later version. |
||||
* |
||||
* SYZOJ is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public |
||||
* License along with SYZOJ. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/ |
||||
|
||||
'use strict'; |
||||
|
||||
let Sequelize = require('sequelize'); |
||||
let db = syzoj.db; |
||||
|
||||
let model = db.define('file', { |
||||
id: { type: Sequelize.INTEGER, primaryKey: true, autoIncrement: true }, |
||||
type: { type: Sequelize.STRING(80) }, |
||||
md5: { type: Sequelize.STRING(80), unique: true } |
||||
}, { |
||||
timestamps: false, |
||||
tableName: 'file', |
||||
indexes: [ |
||||
{ |
||||
fields: ['type'], |
||||
}, |
||||
{ |
||||
fields: ['md5'], |
||||
} |
||||
] |
||||
}); |
||||
|
||||
let Model = require('./common'); |
||||
class File extends Model { |
||||
static create(val) { |
||||
return File.fromRecord(File.model.build(Object.assign({ |
||||
type: '', |
||||
md5: '' |
||||
}, val))); |
||||
} |
||||
|
||||
getPath() { |
||||
return File.resolvePath(this.type, this.md5); |
||||
} |
||||
|
||||
static resolvePath(type, md5) { |
||||
return syzoj.utils.resolvePath(syzoj.config.upload_dir, type, md5); |
||||
} |
||||
|
||||
static async upload(path, type) { |
||||
let fs = Promise.promisifyAll(require('fs-extra')); |
||||
|
||||
let buf = await fs.readFileAsync(path); |
||||
|
||||
if (buf.length > syzoj.config.limit.data_size) throw new ErrorMessage('数据包太大。'); |
||||
|
||||
try { |
||||
let AdmZip = require('adm-zip'); |
||||
let zip = new AdmZip(buf); |
||||
this.unzipSize = 0; |
||||
for (let x of zip.getEntries()) this.unzipSize += x.header.size; |
||||
} catch (e) { |
||||
this.unzipSize = null; |
||||
} |
||||
|
||||
let key = syzoj.utils.md5(buf); |
||||
await fs.moveAsync(path, File.resolvePath(type, key), { overwrite: true }); |
||||
|
||||
let file = await File.findOne({ where: { md5: key } }); |
||||
if (!file) { |
||||
file = await File.create({ |
||||
type: type, |
||||
md5: key |
||||
}); |
||||
await file.save(); |
||||
} |
||||
|
||||
return file; |
||||
} |
||||
|
||||
async getUnzipSize() { |
||||
if (this.unzipSize === undefined) { |
||||
try { |
||||
let fs = Promise.promisifyAll(require('fs-extra')); |
||||
let buf = await fs.readFileAsync(this.getPath()); |
||||
|
||||
let AdmZip = require('adm-zip'); |
||||
let zip = new AdmZip(buf); |
||||
|
||||
this.unzipSize = 0; |
||||
for (let x of zip.getEntries()) this.unzipSize += x.header.size; |
||||
} catch (e) { |
||||
this.unzipSize = null; |
||||
} |
||||
} |
||||
|
||||
if (this.unzipSize === null) throw new ErrorMessage('无效的 ZIP 文件。'); |
||||
else return this.unzipSize; |
||||
} |
||||
|
||||
getModel() { return model; } |
||||
} |
||||
|
||||
File.model = model; |
||||
|
||||
module.exports = File; |
||||
@ -1,64 +0,0 @@
|
||||
/* |
||||
* This file is part of SYZOJ. |
||||
* |
||||
* Copyright (c) 2016 Menci <huanghaorui301@gmail.com> |
||||
* |
||||
* SYZOJ is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as |
||||
* published by the Free Software Foundation, either version 3 of the |
||||
* License, or (at your option) any later version. |
||||
* |
||||
* SYZOJ is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public |
||||
* License along with SYZOJ. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/ |
||||
|
||||
'use strict'; |
||||
|
||||
let Sequelize = require('sequelize'); |
||||
let db = syzoj.db; |
||||
|
||||
let model = db.define('file', { |
||||
id: { type: Sequelize.INTEGER, primaryKey: true, autoIncrement: true }, |
||||
filename: { type: Sequelize.STRING(80), unique: true }, |
||||
md5: { type: Sequelize.STRING(80), unique: true } |
||||
}, { |
||||
timestamps: false, |
||||
tableName: 'file', |
||||
indexes: [ |
||||
{ |
||||
fields: ['filename'], |
||||
}, |
||||
{ |
||||
fields: ['md5'], |
||||
} |
||||
] |
||||
}); |
||||
|
||||
let Model = require('./common'); |
||||
class TestData extends Model { |
||||
static create(val) { |
||||
return TestData.fromRecord(TestData.model.build(Object.assign({ |
||||
filename: '', |
||||
md5: '' |
||||
}, val))); |
||||
} |
||||
|
||||
getPath() { |
||||
return TestData.resolvePath(this.md5); |
||||
} |
||||
|
||||
static resolvePath(md5) { |
||||
return syzoj.utils.resolvePath(syzoj.config.upload_dir, 'testdata', md5); |
||||
} |
||||
|
||||
getModel() { return model; } |
||||
} |
||||
|
||||
TestData.model = model; |
||||
|
||||
module.exports = TestData; |
||||
Loading…
Reference in new issue