package main

import (
	"html/template"
	"net/http"
	"path"

	"github.com/gorilla/mux"
	minio "github.com/minio/minio-go"
)

// ObjectWithIcon is a minio object with an added icon
type ObjectWithIcon struct {
	minio.ObjectInfo
	Icon string
}

// BucketPage defines the details page of a bucket
type BucketPage struct {
	BucketName string
	Objects    []ObjectWithIcon
}

// BucketViewHandler shows the details page of a bucket
func BucketViewHandler(s3 S3Client) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		bucketName := mux.Vars(r)["bucketName"]
		var objs []ObjectWithIcon

		l := path.Join("templates", "layout.html.tmpl")
		p := path.Join("templates", "bucket.html.tmpl")

		t, err := template.ParseFiles(l, p)
		if err != nil {
			msg := "error parsing templates"
			handleHTTPError(w, msg, err, http.StatusInternalServerError)
			return
		}

		doneCh := make(chan struct{})
		defer close(doneCh)
		objectCh := s3.ListObjectsV2(bucketName, "", true, doneCh)
		for object := range objectCh {
			if object.Err != nil {
				msg := "error listing objects"
				code := http.StatusInternalServerError
				if object.Err.Error() == "The specified bucket does not exist." {
					msg = "bucket not found"
					code = http.StatusNotFound
				}

				handleHTTPError(w, msg, object.Err, code)
				return
			}
			objectWithIcon := ObjectWithIcon{object, icon(object.Key)}
			objs = append(objs, objectWithIcon)
		}

		bucketPage := BucketPage{
			BucketName: bucketName,
			Objects:    objs,
		}

		err = t.ExecuteTemplate(w, "layout", bucketPage)
		if err != nil {
			msg := "error executing template"
			handleHTTPError(w, msg, err, http.StatusInternalServerError)
			return
		}
	})
}

// icon returns an icon for a file type
func icon(fileName string) string {
	e := path.Ext(fileName)

	switch e {
	case ".tgz":
		return "archive"
	case ".png", ".jpg", ".gif", ".svg":
		return "photo"
	case ".mp3":
		return "music_note"
	}

	return "insert_drive_file"
}