// Copyright 2015, 2016 Ethcore (UK) Ltd.
// This file is part of Parity.

// Parity is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Parity 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 General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Parity.  If not, see <http://www.gnu.org/licenses/>.

use std::io::Write;
use hyper::uri::RequestUri;
use hyper::server;
use hyper::header;
use hyper::status::StatusCode;
use parity_webapp::WebApp;

pub trait Page : Send + Sync {
	fn serve_file(&self, mut path: &str, mut res: server::Response);
}

pub struct PageHandler<T : WebApp> {
	pub app: T
}

impl<T: WebApp> Page for PageHandler<T> {
	fn serve_file(&self, mut path: &str, mut res: server::Response) {
		// Support index file
		if path == "" {
			path = "index.html"
		}
		let file = self.app.file(path);
		if let Some(f) = file {
			*res.status_mut() = StatusCode::Ok;
			res.headers_mut().set(header::ContentType(f.content_type.parse().unwrap()));

			let _ = match res.start() {
				Ok(mut raw_res) => {
					for chunk in f.content.chunks(1024 * 20) {
						let _ = raw_res.write(chunk);
					}
					raw_res.end()
				},
				Err(_) => {
					println!("Error while writing response.");
					Ok(())
				}
			};
		}
	}
}

impl server::Handler for Page {
	fn handle(&self, req: server::Request, mut res: server::Response) {
		*res.status_mut() = StatusCode::NotFound;

		if let RequestUri::AbsolutePath(ref path) = req.uri {
			self.serve_file(path, res);
		}
	}
}