index.js

#!/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'

let user
let file
/**
 * 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')

		file = await new File(dbName)

		await file.deleteExpired()
		user = await new User(dbName)

		await user.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

		file = await new File(dbName)
		await file.upload(extension, name, size, userName, path)

		ctx.redirect('/myfiles?msg=Your file has been uploaded!')
	} catch (err) {
		await ctx.render('error', { message: err.message })
	} finally {
		file.tearDown()
	}
})

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
		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})
	} finally {
		user.tearDown()
	}
})


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
		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})
	} finally {
		user.tearDown()
	}
})

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

		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 })
	} finally {
		file.tearDown()
	}
})

router.get('/download/:usrName/:name', async ctx => {
	try {
		const fileName = ctx.params.name
		const userName = ctx.params.usrName

		file = await new File(dbName)

		const fileData = await file.getFileInfo(fileName, userName)
		//generate link with a direct path to file
		const link = await file.generateLink(fileName, userName, false)

		const data = {}
		data.fileName = `${fileData[0]['FileName']}.${fileData[0]['FileType']}`
		data.fileHash = fileData[0]['FileHash']

		await ctx.render('download', { link: link, file: data })
	} catch (err) {
		await ctx.render('error', { message: err.message })
	} finally {
		file.tearDown()
	}
})

router.post('/download', async ctx => {
	try {
		const downloaded = ctx.request.body.downloaded
		const fileHash = ctx.request.body.fileHash

		if (downloaded) {
			file = await new File(dbName)
			await file.updateValue(fileHash)
		}

		return ctx.redirect('/')
	} 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}`))