index.js

#!/usr/bin/env node

/**
 * Routes File
 */

'use strict'

/* MODULE IMPORTS */
const bcrypt = require('bcrypt-promise')
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 sqlite = require('sqlite-async')
const fs = require('fs-extra')
const mime = require('mime-types')
const validator = require('email-validator')
//var fs = require('fs');
//const jimp = require('jimp')

/* IMPORT CUSTOM MODULES */
const User = require('./modules/user')
const Appliance = require('./modules/appliance')
const Log = require('./modules/logs')
const validateRegister = require('./modules/checkRegister')
const validateLogin = require('./modules/checkLogin')

const app = new Koa()
const router = new Router()

/* CONFIGURING THE MIDDLEWARE */
app.keys = ['darkSecret']
app.use(staticDir('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'
const saltRounds = 10

/**
 * The secure home page.
 *
 * @name Home Page
 * @route {GET} /
 * @authentication This route requires cookie-based authentication.
 */
router.get('/', koaBody, 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
		let filterStatus = ctx.query.status
		if(filterStatus === undefined) filterStatus = 'empty'
		const username = ctx.session.userNameCookie
		const applainces = await new Appliance(dbName)
		const joinJob = await applainces.getCustomerAppliances(username, filterStatus)
		await ctx.render('index', {jobs: joinJob, query: filterStatus, username})
	} 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 {
		const body = ctx.request.body
		console.log(body)
		if(validateRegister.checkTitle(body.title) === false)
			return ctx.redirect('/register?msg=Title cannot be empty')
		if(validateRegister.checkUsername(body.user) === false)
			return ctx.redirect('/register?msg=Username cannot be empty')
		if(validateRegister.checkEmail(body.email) === false)
			return ctx.redirect('/register?msg=Email is invalid')
		if(validateRegister.checkPassword(body.pass) === false) return ctx.redirect('/register?msg=Password cannot be empty')
		const user = await new User(dbName)
		await user.register(body.user, body.pass, body.title, body.email, body.house, body.postCode)
		// redirect to the home page
		ctx.redirect(`/?msg=new user "${body.name}" added`)
	} catch(err) {
		await ctx.render('error', {message: err.message})
	}
})
/**
 * The script to render login page.
 *
 * @name Login page
 * @route {GET} /login
 */
router.get('/login', async ctx => {
	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)
})
/**
 * The script to process user login.
 *
 * @name Login Script
 * @route {GET} /login
 */
router.post('/login', async ctx => {
	try {
		const body = ctx.request.body
		if(validateLogin.checkUsername(body.user) === false)
			return ctx.redirect('/login?msg=Username cannot be empty')
		if(validateLogin.checkPassword(body.pass) === false)
			return ctx.redirect('/?msg=Password cannot be empty')
		const user = await new User(dbName)
		const validUsername = await user.login(body.user, body.pass)
		if(validUsername === false) return ctx.redirect('/login?msg=Invalid login Credentials')
		ctx.session.authorised = true
		ctx.session.userNameCookie = body.user
		return ctx.redirect('/?msg=you are now logged in...')
	} catch(err) {
		await ctx.render('error', {message: err.message})
	}
})
/**
 * The script to render report page.
 *
 * @name Report page
 * @route {GET} /report
 * @authentication This route requires cookie-based authentication.
 */
router.get('/report', koaBody, async ctx => {
	if(ctx.session.authorised !== true) return ctx.redirect('/login?msg=you need to log in')
	await ctx.render('report')
})
/**
 * The script to process appliance report.
 *
 * @name Report Script
 * @route {POST} /report
 * @authentication This route requires cookie-based authentication.
 */
router.post('/report', koaBody, async ctx => {
	const body = ctx.request.body
	const username = ctx.session.userNameCookie
	console.log(`UserName in Cookie: ${ username}`)
	const appliance = await new Appliance(dbName)
	await appliance.report(body.type, body.age, body.manufacturer, body.description, username)
	return ctx.redirect('/?msg=Your appliance has been added...')
})
/**
 * The script to render log page.
 *
 * @name Log page
 * @route {GET} /log
 * @params applianceID
 * @authentication This route requires cookie-based authentication.
 */
router.get('/log/:applianceID', async ctx => {
	const logs = await new Log(dbName)
	const applianceID = ctx.params.applianceID
	const data = await logs.getLog(applianceID)
	await ctx.render('logs', {log: data, id: applianceID})
})
/**
 * The script to logout user.
 * @name Logout Script
 * @route {GET} /logout
 */
router.get('/logout', async ctx => {
	ctx.session.authorised = null
	ctx.redirect('/?msg=you are now logged out')
})

app.use(router.routes())
module.exports = app.listen(port, async() => console.log(`listening on port ${port}`))