mirror-yaf/jaf.go

package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"log"
	"math/rand"
	"net/http"
	"time"
)

const allowedChars = "0123456789ABCDEFGHIJKLMNOPQRTSUVWXYZabcdefghijklmnopqrstuvwxyz"

var (
	savedFileNames = NewSet()
	config         Config
)

type parameters struct {
	configFile string
}

func parseParams() *parameters {
	configFile := flag.String("configFile", "jaf.conf", "path to config file")
	flag.Parse()

	retval := &parameters{}
	retval.configFile = *configFile
	return retval
}

func main() {
	rand.Seed(time.Now().UnixNano())
	log.SetPrefix("jaf > ")

	params := parseParams()

	// Read config
	config, err := ConfigFromFile(params.configFile)
	if err != nil {
		log.Fatalf("could not read config file: %s\n", err.Error())
	}

	files, err := ioutil.ReadDir(config.FileDir)
	if err != nil {
		log.Fatalf("could not read file root %s: %s\n", config.FileDir, err.Error())
	}

	// Cache taken file names on start-up
	for _, fileInfo := range files {
		savedFileNames.Insert(fileInfo.Name())
	}

	// Start server
	uploadServer := &http.Server{
		ReadTimeout:  30 * time.Second,
		WriteTimeout: 30 * time.Second,
		Addr:         fmt.Sprintf(":%d", config.Port),
	}

	log.Printf("starting jaf on port %d\n", config.Port)
	http.Handle("/upload", &uploadHandler{config: config})
	uploadServer.ListenAndServe()
}