package main

import (
	"io"
	"net/http"
	"path/filepath"
	"time"

	"github.com/NIR3X/filecache"
	"github.com/NIR3X/filewatcher"
	"github.com/NIR3X/logger"
)

const (
	fileCacheMaxSize          = 2 * 1024 * 1024
	fileWatcherUpdateInterval = 1 * time.Second
	rootDir                   = "www"
)

func main() {
	fileCache := filecache.NewFileCache(fileCacheMaxSize)
	multiSender := NewMultiSender(fileCache)
	fileWatcher := filewatcher.NewFileWatcher(fileWatcherUpdateInterval, func(path string, isDir bool) { // created
		if isDir {
			return
		}
		err := fileCache.Update(path)
		if err != nil {
			logger.Eprintln(err)
		}
	}, func(path string, isDir bool) { // removed
		if isDir {
			return
		}
		fileCache.Delete(path)
	}, func(path string, isDir bool) { // modified
		if isDir {
			return
		}
		err := fileCache.Update(path)
		if err != nil {
			logger.Eprintln(err)
		}
	})
	defer fileWatcher.Close()
	err := fileWatcher.Watch(rootDir)
	if err != nil {
		logger.Eprintln(err)
		return
	}
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		path := r.URL.Path
		if path == "/" {
			path = "/index.html"
		}
		path = filepath.Join(rootDir, path)
		reader, ident := fileCache.GetCached(path)
		switch ident {
		case filecache.Cached:
			_, err = io.Copy(w, reader)
			if err != nil {
				logger.Eprintln(err)
			}
		case filecache.Piped:
			multiSenderWriter := multiSender.Add(path, w)
			multiSenderWriter.Wait()
		}
	})
	logger.Println("Listening on port 8000...")
	err = http.ListenAndServe(":8000", nil)
	if err != nil {
		logger.Eprintln(err)
		return
	}
}