app_server.n

#!/usr/bin/env node --harmony one_node.js
/**
 *  ONEJS Static nodejs server
 * 
 *  Copyright (C) 2014 ONEJS 
 *
 *  MIT license: Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 * 
 */
import node
import htmlscraper
import math

var live = 0
if(live){
	var addr = "0.0.0.0"
	var port = 80
}
else{
	var addr = "0.0.0.0"
	var port = 2000
}

var fs_root = Path.resolve(__dirname)

class StaticServer extends HttpServer{

	// lets package our app
	package( file, root ){
		var html = await fs.readFile(file)
		var tree = HTMLScraper.parse(html)

		// lets inline each script tag
		for(var script from tree.html.head.script){
			if(script.$src){
				if(script.$src == 'reloader.js'){
					delete script.$src
					continue
				}
				var file = await fs.readFile(script.$src)
				delete script.$src
				script.text = file.toString()
			}
		}

		// recursively find all our dependencies
		var onejs = {}
		function fetch(file){
			var code = await fs.readFile(file + '.n')
			onejs[file] = code.toString()
			var ast = parse('->{' + code + '\n}')
			for(var dep from ast.getDependencies()){
				await fetch(dep)
			}
		}
		await fetch(root)

		// add new tags 
		for(var file in onejs) tree.html.head.appendChild() do {
			tagname:'script'
			$type: 'text/onejs'
			$id: file == root ? 'main' : file
			text: onejs[file]
		}
		// ok first lets stringify it again.
		return tree.toString()
	}

	route("/_reloader_") do( req, res ){
		setTimeout(15000) on(){
			res.writeHead(204)
			res.end()			
		}
		(pending ?= []).push(res)
	}

	route("/pull") do( req, res ){
		out('Updating and clearing cache!')
		child_process.exec('git pull')
		child_process.exec('rm -rf cache_*')
		res.writeHead(200)
		res.end()
	}

	if(live) route("/") do( req, res ){
		var host = req.headers.host
		var root = ''
		if(host.indexOf('localhost') == 0) root = 'index'

		else root = host.slice(0, host.indexOf('.'))

		var cache_file = 'cache_'+root
		var exist = await fs.exists(cache_file)
		var header = {
			"Content-Type": "text/html"
			"Content-encoding":"deflate"
		}
		if(exist){
			out('Loading cached ' + cache_file)
			var stream = fs.createReadStream(cache_file)
			res.writeHead(200, header)
			stream.pipe(res)
			return
		}

		// lets fetch the root
		var pkg = await package('index.html', root).catch do(){
			out("Exception!")
			res.writeHead(501)
			res.end()
		}

		// deflate it
		var data = await zlib.deflate(pkg)
		res.writeHead(200, header)
		res.write(data)
		res.end()
		fs.writeFile(cache_file, data)
	}

	route(/.*/) do( req, res ){
		var path = req.url.pathname
		if(path == '/') path = 'index.html'
		var fullpath = Path.join(fs_root, path)

		out(`Reading ${path}`)

		// read and return the file
		var exist = await fs.exists(fullpath)
		if(!exist){
			res.writeHead(404)
			res.end('File not found')
			return
		}

		var stream = fs.createReadStream(fullpath)
		res.writeHead(200, {"Content-Type": mime.fromFilename(fullpath)})
		stream.pipe(res)
		watch(fullpath)
	}

	watch( path ){

		files ?= {}
		if(files[path]) return

		files[path] = fs.watchFileFast(path) do(){
			out("---- " + path + " changed, sending reload to frontend ----" )
			for(var res of pending){
				res.writeHead(200)
				res.end('Reload')
			}
			pending = []
		}
	}

	init( port, addr ){
		out(`Starting webserver on ${addr}:${port}`)
		super()
	}
}

new StaticServer( port, addr )