Permalink
Cannot retrieve contributors at this time
Name already in use
A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
File-Sharing-Website/index.js
Go to fileThis commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
239 lines (203 sloc)
6.24 KB
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env node | |
//Routes File | |
'use strict' | |
/* MODULE IMPORTS */ | |
const Koa = require('koa') | |
const Router = require('koa-router') | |
const views = require('koa-views') | |
const staticDir = require('koa-static') | |
const bodyParser = require('koa-bodyparser') | |
const koaBody = require('koa-body')({multipart: true, uploadDir: '.'}) | |
const session = require('koa-session') | |
const mime = require('mime-types') | |
const Database = require('sqlite-async') | |
/* IMPORT CUSTOM MODULES */ | |
const User = require('./modules/user') | |
const Mailer = require('./modules/mailer') | |
const File = require('./modules/upload') | |
const app = new Koa() | |
const router = new Router() | |
/* CONFIGURING THE MIDDLEWARE */ | |
app.keys = ['darkSecret'] | |
app.use(staticDir('public')) | |
app.use(staticDir('private')) //TODO: is this considered 'not public'? | |
app.use(bodyParser()) | |
app.use(session(app)) | |
app.use(views(`${__dirname}/views`, { extension: 'handlebars' }, {map: { handlebars: 'handlebars' }})) | |
const defaultPort = 8080 | |
const port = process.env.PORT || defaultPort | |
const dbName = 'website.db' | |
/** | |
* The secure home page. | |
* | |
* @name Home Page | |
* @route {GET} / | |
* @authentication This route requires cookie-based authentication. | |
*/ | |
router.get('/', async ctx => { | |
try { | |
// if(ctx.session.authorised !== true) return ctx.redirect('/login?msg=you need to log in') | |
const data = {} | |
if(ctx.query.msg) data.msg = ctx.query.msg | |
await ctx.render('home') | |
} catch(err) { | |
await ctx.render('error', {message: err.message}) | |
} | |
}) | |
/** | |
* The Users Interaction Page and Script | |
* @name User Page | |
* @route {GET} /myfiles | |
*/ | |
// eslint-disable-next-line max-lines-per-function | |
router.get('/myfiles', async ctx => { | |
try { | |
if(ctx.session.authorised !== true) return ctx.redirect('/login?msg=You need to log in') | |
await new File(dbName) | |
// await ctx.render('myfiles') | |
const display = await new User(dbName) | |
await display.UserFileDisplay(ctx, dbName, Database) | |
// console.log(data) | |
} catch(err) { | |
await ctx.render('error', {message: err.message}) | |
} | |
}) | |
router.post('/myfiles', koaBody, async ctx => { | |
try { | |
const userName = ctx.session.userName | |
// extract the data from the request | |
const { path, type } = ctx.request.files._file | |
const extension = mime.extension(type) | |
let name = ctx.request.files._file.name | |
name = name.replace(`.${extension}`, '') | |
const size = ctx.request.files._file.size | |
const date = ctx.request.files._file.lastModifiedDate | |
const file = await new File(dbName) | |
await file.upload(extension, name, size, date, userName, path) | |
ctx.redirect('/myfiles?msg=Your file has been uploaded!') | |
} catch (err) { | |
await ctx.render('error', { message: err.message }) | |
} | |
}) | |
router.get('/shared', async ctx => { | |
try { | |
await ctx.render('shared') | |
} catch (err) { | |
await ctx.render('error', { message: err.message }) | |
} | |
}) | |
router.get('/about', async ctx => { | |
try { | |
await ctx.render('AboutUs') | |
} catch (err) { | |
await ctx.render('error', { message: err.message }) | |
} | |
}) | |
router.get('/contact', async ctx => { | |
try { | |
await ctx.render('ContactUs') | |
} catch (err) { | |
await ctx.render('error', { message: err.message }) | |
} | |
}) | |
/** | |
* The user registration page. | |
* | |
* @name Register Page | |
* @route {GET} /register | |
*/ | |
router.get('/register', async ctx => { | |
await ctx.render('register') | |
}) | |
/** | |
* The script to process new user registrations. | |
* | |
* @name Register Script | |
* @route {POST} /register | |
*/ | |
router.post('/register', koaBody, async ctx => { | |
try { | |
// extract the data from the request | |
const body = ctx.request.body | |
// call the functions in the module | |
const user = await new User(dbName) | |
await user.register(body.user, body.pass) | |
// await user.uploadPicture(path, type) | |
// redirect to the home page | |
ctx.redirect(`/?msg=new user "${body.name}" added`) | |
} catch(err) { | |
await ctx.render('error', {message: err.message}) | |
} | |
}) | |
router.get('/login', async ctx => { | |
if(ctx.session.authorised === true) return ctx.redirect('myfiles') | |
const data = {} | |
if(ctx.query.msg) data.msg = ctx.query.msg | |
if(ctx.query.user) data.user = ctx.query.user | |
await ctx.render('login', data) | |
}) | |
router.post('/login', async ctx => { | |
try { | |
const body = ctx.request.body | |
const user = await new User(dbName) | |
const _userName = await user.login(body.user, body.pass) | |
ctx.session.authorised = true | |
ctx.session.userName = _userName | |
return ctx.redirect('/myfiles') | |
// return ctx.redirect('/?msg=you are now logged in...') | |
} catch(err) { | |
await ctx.render('error', {message: err.message}) | |
} | |
}) | |
router.get('/logout', async ctx => { | |
ctx.session.authorised = null | |
ctx.redirect('/') | |
// ctx.redirect('/?msg=you are now logged out') | |
}) | |
router.get('/transfer', async ctx => { | |
try { | |
const data = {} | |
if (ctx.query.name) data.cleanFileName = ctx.query.name | |
if (ctx.query.ex) data.fileType = ctx.query.ex | |
if (ctx.query.msg) data.msg = ctx.query.msg | |
await ctx.render('transfer', data) | |
} catch (err) { | |
await ctx.render('error', { message: err.message }) | |
} | |
}) | |
router.post('/transfer', koaBody, async ctx => { | |
try { | |
const userName = ctx.session.userName | |
// extract the data from the request | |
const filename = ctx.request.body.fileName | |
const userMessage = ctx.request.body.message | |
const email = ctx.request.body.targetMail | |
const file = await new File(dbName) | |
//Generate the link to download page | |
const link = await file.generateLink(filename, userName, true) | |
console.log(`Download link: ${link}`) | |
//TODO: username passed in email | |
//Send the link in email | |
await Mailer.sendMail(email, link, userMessage) | |
ctx.redirect('/myfiles?msg=Your file has been transfered!') | |
} catch (err) { | |
await ctx.render('error', { message: err.message }) | |
} | |
}) | |
router.get('/download/:usrName/:name', async ctx => { | |
try { | |
let fileName = ctx.params.name | |
const userName = ctx.params.usrName | |
const file = await new File(dbName) | |
const fileData = await file.getFileInfo(fileName, userName) | |
console.log(fileData) | |
//generate link with a direct path to file | |
const link = await file.generateLink(fileName, userName, false) | |
fileName = `${fileData[0]['FileName']}.${fileData[0]['FileType']}` | |
await ctx.render('download', { link: link, file: fileName }) | |
} catch (err) { | |
await ctx.render('error', { message: err.message }) | |
} | |
}) | |
app.use(router.routes()) | |
module.exports = app.listen(port, async() => console.log(`listening on port ${port}`)) |