main.go

package main // import "github.com/raohwork/prpr"

import (
	"log"
	"net/http"
	"os"
	"os/exec"
	"strconv"
	"strings"
	"time"
)

func main() {
	secret := os.Getenv("SECRET")
	firefox := os.Getenv("FIREFOX")
	if firefox == "" {
		firefox = "firefox"
	}
	bind := os.Getenv("BIND")
	if bind == "" {
		bind = ":9801"
	}

	ch := make(chan error)

	go runFX(firefox, ch)

	// wait few second for firefox to start
	time.Sleep(10 * time.Second)
	go runWeb(bind, secret, ch)

	log.Fatal(<-ch)
}

func runFX(firefox string, ch chan error) {
	// starting firefox
	opts := []string{
		"--marionette",
		"--safe-mode",
	}
	if os.Getenv("DEBUG_FIREFOX") == "" {
		opts = append(opts, "--headless")
	}
	if x := os.Getenv("FIREFOX_OPTS"); x != "" {
		xarr := strings.Split(x, " ")
		for _, item := range xarr {
			item = strings.TrimSpace(item)
			if item == "" {
				continue
			}
			opts = append(opts, item)
		}
	}
	cmd := exec.Command(
		firefox,
		opts...,
	)

	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	ch <- cmd.Run()
}

func runWeb(bind, secret string, ch chan error) {
	maxStr := os.Getenv("QUEUE_SIZE")
	max, err := strconv.Atoi(maxStr)
	if err != nil || max < 1 {
		max = 1
	}
	h := New("", 0, max)
	if secret != "" {
		h.Secret = secret
	}
	http.HandleFunc("/grab", h.Accept)
	log.Print("Starting prpr...")
	ch <- http.ListenAndServe(bind, nil)
}