// webshit is a fork of uriel's cgd
package main

import (
	"flag"
	"log"
	"net"
	"net/http"
	"net/http/cgi"
	"net/http/fcgi"
	"os"
)

var (
	cmd = flag.String("c", "", "CGI program to run")
	pwd = flag.String("w", "", "Working dir for CGI")
	serveFcgi = flag.Bool("f", false, "Run as a FCGI 'server' instead of HTTP")
	address = flag.String("a", ":3333", "Listen address")
)

func usage() {
	os.Stderr.WriteString("usage: webshit [-s] -c prog [-w wdir] [-a addr]\n")
	flag.PrintDefaults()
	os.Exit(2)
}

func main() {
	flag.Usage = usage
	flag.Parse()

	if *cmd == "" {
		usage()
	}

	// This is a hack to make p9p's rc happier for some unknown reason.
	c := *cmd
	if c[0] != '/' {
		c = "./" + c
	}

	os.Setenv("PATH", os.Getenv("PATH")+":.")

	h := &cgi.Handler{
		Path:       c,
		Root:       "/",
		Dir:        *pwd,
		InheritEnv: []string{"PATH", "PLAN9"},
	}

	var err error
	if *serveFcgi {
		if l, err := net.Listen("tcp", *address); err == nil {
			log.Println("Starting FastCGI daemon listening on", *address)
			err = fcgi.Serve(l, h)
		}

	} else {
		log.Println("Starting HTTP server listening on", *address)
		err = http.ListenAndServe(*address, h)
	}

	if err != nil {
		log.Fatal(err)
	}
}