From 72b255dd4b0b182e3529a3cead0015e73be81680 Mon Sep 17 00:00:00 2001
From: Pacien
Date: Sat, 29 Jun 2013 13:03:56 +0200
Subject: Add custom parameters: parsable extensions, save as

---
 main.go | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 49b27fa..2c0cf81 100644
--- a/main.go
+++ b/main.go
@@ -22,6 +22,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"strings"
 )
 
 var settings struct {
@@ -29,6 +30,8 @@ var settings struct {
 	sourceDir *string
 	outputDir *string // for compiled site
 	port      *string // for the integrated web server (dynamic mode only)
+	exts      []string
+	saveAs    *string
 }
 
 func init() {
@@ -37,7 +40,13 @@ func init() {
 	settings.sourceDir = flag.String("source", ".", "Path to sources directory.")
 	settings.outputDir = flag.String("output", "./out", "[compiled mode] Path to output directory.")
 	settings.port = flag.String("port", "8080", "[dynamic mode] Port to listen.")
+	exts := flag.String("exts", "html, txt, md", "List parsable file extensions. Separated by commas.")
+	settings.saveAs = flag.String("saveAs", "index.html", "[compiled and interactive modes] Save compiled files as named.")
 	flag.Parse()
+	settings.exts = strings.Split(*exts, ",")
+	for i, ext := range settings.exts {
+		settings.exts[i] = "." + strings.Trim(ext, ". ")
+	}
 }
 
 func main() {
@@ -49,9 +58,9 @@ func main() {
 
 	switch *settings.mode {
 	case "compiled":
-		compiled(*settings.sourceDir, *settings.outputDir)
+		compiled(*settings.sourceDir, *settings.outputDir, settings.exts, *settings.saveAs)
 	case "interactive":
-		interactive(*settings.sourceDir, *settings.outputDir)
+		interactive(*settings.sourceDir, *settings.outputDir, settings.exts, *settings.saveAs)
 	case "dynamic":
 		dynamic(*settings.port)
 	default:
-- 
cgit v1.2.3