Callbacks

Author: carlos-r-l-rodrigues

index.ts

@@ -1,24 +1,8 @@
 import { config, updateConfig, loadHandler } from "./src/config"
 import { RouteCheck } from "./src/RouteCheck"
+import { ConfigParameters } from "./src/interface";
 
-export default (options: any = {}) => {
-	// TODO: options callbacks
-	/*
-	options = {
-		callbacks: {
-			before_serve(key: string, body: string, headers: any): void {
-
-			},
-			before_cache(route: any, cache: Buffer[], headers: any): void {
-
-			},
-			after_cache(route: any, body: Buffer, headers: any): void {
-
-			}
-		}
-	}
-	*/
-
+export default (options: ConfigParameters = {}) => {
 	updateConfig(options);
 
 	const module = loadHandler(config);
@@ -41,7 +25,9 @@ export default (options: any = {}) => {
 				return next();
 			}
 			else if (reply) {
-				// TODO: callback before_serve
+				if(config.callbacks && config.callbacks.before_serve) {
+					config.callbacks.before_serve(key, reply.content, reply.header, reply.encoding);
+				}
 
 				res.set(reply.header);
 				return res.end(reply.content, reply.encoding);
@@ -66,9 +52,11 @@ export default (options: any = {}) => {
 						return;
 					}
 
-					// TESTEEEE
-					// TODO: callback before_cache
-					cache.push(Buffer.from("<hr /><b>cac<font color='red'>hed</font></b>", encoding));
+
+					let originalHeaders = Object.assign({}, res._headers);
+					if(config.callbacks && config.callbacks.before_cache) {
+						config.callbacks.before_cache(route, cache, originalHeaders, encoding);
+					}
 
 					const now = new Date();
 					let dateExpire = new Date();
@@ -78,11 +66,11 @@ export default (options: any = {}) => {
 					const buf = Buffer.concat(cache);
 
 					const headers = route.mergeHeaders(
-						res._headers,
+						originalHeaders,
 						{
 							"Content-Length": buf.length,
 							"Last-Modified": now.toUTCString(),
-							"Cache-Control": "max-age=" + (dateExpire.getTime() / 1000)
+							"Cache-Control": "max-age=" + (dateExpire.getTime() / 1000 >> 0)
 						}
 					);
 
@@ -92,8 +80,11 @@ export default (options: any = {}) => {
 						headers,
 						rawUrl,
 						expire
-					);
-					// TODO: callback after_cache
+					, (key) => {
+						if(config.callbacks && config.callbacks.after_cache) {
+							config.callbacks.after_cache(key, route, buf, headers, encoding);
+						}
+					});
 					end.call(res);
 				};
 				return next();

src/config.ts

@@ -1,9 +1,10 @@
+import { ConfigParameters } from "./interface";
 import * as path from "path";
 import * as fs from "fs"
 
 export let config: any = {}
 
-export function updateConfig(options: any) {
+export function updateConfig(options: ConfigParameters) {
 	this.config = Object.assign(this.config, options);
 }
 

src/handler/InMemory.ts

@@ -16,7 +16,7 @@ export class Handler {
 
 		return callback(null, null);
 	}
-	set(key: string, content: Buffer, header: any, rawUrl: string, expire: number = null) {
+	set(key: string, content: Buffer, header: any, rawUrl: string, expire: number = null, callback?: Function) {
 		if (this.config.prefix) {
 			key = this.config.prefix + "_" + key;
 		}
@@ -30,6 +30,8 @@ export class Handler {
 			rawUrl,
 			expire
 		}
+
+		callback && callback(key);
 	}
 
 	del(key: string) {

src/interface.ts

@@ -0,0 +1,10 @@
+export declare interface ConfigCallbacks {
+	before_serve?(key: string, body: string, headers: any, encoding?: string): void;
+	before_cache?(route: any, cache: Buffer[], headers: any, encoding?: string): void;
+	after_cache?(key: string, route: any, body: Buffer, headers: any, encoding?: string): void;
+}
+export declare interface ConfigParameters {
+	handler?: string;
+	isCacheable?(req: any): boolean;
+	callbacks?: ConfigCallbacks;
+}