golang-fave/main.go

package main

import (
	"flag"
	"fmt"
	"io"
	"log"
	"net/http"
	"os"
	"strconv"
	"strings"

	"golang-fave/engine/wrapper"
)

const C_Debug = false

var FParamHost string
var FParamPort int
var FParamWwwDir string
var FVhostHomeDir string

func init() {
	flag.StringVar(&FParamHost, "host", "0.0.0.0", "server host")
	flag.IntVar(&FParamPort, "port", 8080, "server port")
	flag.StringVar(&FParamWwwDir, "dir", "", "virtual hosts directory")
	flag.Parse()
}

func main() {
	if _, err := os.Stat(FParamWwwDir); os.IsNotExist(err) {
		fmt.Println("Virtual hosts directory is not exists")
		fmt.Println("Example: ./fave -host 127.0.0.1 -port 80 -dir ./hosts")
		return
	}
	if FParamWwwDir[len(FParamWwwDir)-1] != '/' {
		FParamWwwDir = FParamWwwDir + "/"
	}
	http.HandleFunc("/", handler)
	fmt.Println("Starting server at " + FParamHost + ":" + strconv.Itoa(FParamPort))
	log.Fatal(http.ListenAndServe(FParamHost+":"+strconv.Itoa(FParamPort), nil))
}

func vhExists(vhosthome string) bool {
	if fi, err := os.Stat(vhosthome); !os.IsNotExist(err) {
		if err == nil {
			fmode := fi.Mode()
			if fmode.IsDir() {
				return true
			}
		}
	}
	return false
}

func handler(w http.ResponseWriter, r *http.Request) {
	// Build vhost home dir
	host := r.Host
	port := strconv.Itoa(FParamPort)
	index := strings.Index(host, ":")
	if index > -1 {
		port = host[index+1:]
		host = host[0:index]
	}

	// Cut "www" if exists
	if strings.HasPrefix(host, "www.") {
		host = host[4:]
	}
	FVhostHomeDir = FParamWwwDir + host

	// Check if virtual host exists
	if !vhExists(FVhostHomeDir) {
		host = "localhost"
		FVhostHomeDir = FParamWwwDir + host
	}

	// Set protocol
	r.URL.Scheme = "http"

	// Set server name
	w.Header().Set("Server", "fave.pro")

	// Create and start engine
	wrapper.New(&w, r, host, port, FParamWwwDir, FVhostHomeDir, C_Debug).
		Run(func(e *wrapper.Wrapper) bool {
			if e.R.URL.Path == "/" {
				if !e.Session.IsSetInt("CounterTest") {
					e.Session.SetInt("CounterTest", 1)
				}

				cc, err := e.Session.GetInt("CounterTest")
				if err != nil {
					cc = 1
				}

				(*e.W).Header().Set("Content-Type", "text/html")
				io.WriteString(*e.W, "Home<br />")
				io.WriteString(*e.W, "<a href=\"/static.html\">Static Page</a><br />")
				io.WriteString(*e.W, "<a href=\"/robots.txt\">robots.txt</a><br />")
				io.WriteString(*e.W, "<a href=\"/static404\">Page 404</a><br />")
				io.WriteString(*e.W, "Counter: "+strconv.Itoa(cc))

				e.Session.SetInt("CounterTest", cc+1)

				return true
			}
			return false
		})
}