resource.go

package resource

import (
	"gopkg.in/cookieo9/resources-go.v2"
	"io/ioutil"
	"log"
	"mime"
	"net/http"
	"path/filepath"
	"strings"
	"time"
)

type Resource struct{}

var DefaultResource *Resource = &Resource{}

func (re *Resource) Load(path string) []byte {
	res, err := re.Get(path)
	if err != nil {
		return []byte{}
	}
	r, _ := res.Open()
	bf, err := ioutil.ReadAll(r)
	if err != nil {
		log.Println("read res[", path, "] failed", err.Error())
	}
	return bf
}

func (re *Resource) Get(path string) (resources.Resource, error) {
	path = strings.TrimLeft(path, "/")
	res, err := resources.Find(path)
	if err != nil {
		log.Println("load res[", path, "] failed", err.Error())
		return nil, err
	}
	return res, nil
}

func (re *Resource) HandleStatic(w http.ResponseWriter, r *http.Request, path string) {
	res, err := re.Get(path)
	if err != nil {
		http.NotFound(w, r)
		return
	}
	finfo, _ := res.Stat()
	modtime := finfo.ModTime()
	if t, err := time.Parse(http.TimeFormat, r.Header.Get("If-Modified-Since")); err == nil && modtime.Before(t.Add(1*time.Second)) {
		h := w.Header()
		delete(h, "Content-Type")
		delete(h, "Content-Length")
		w.Header().Set("Last-Modified", modtime.UTC().Format(http.TimeFormat))
		w.WriteHeader(http.StatusNotModified)
		return
	}
	mimeType := mime.TypeByExtension(filepath.Ext(path))
	if mimeType != "" {
		w.Header().Set("Content-Type", mimeType)
	}
	w.Header().Set("Last-Modified", modtime.UTC().Format(http.TimeFormat))
	w.Write(re.Load(path))
}

func ResetDefaultBundle() {
	resources.DefaultBundle = make(resources.BundleSequence, 1, 10)

	var cwd, exe_dir, exe, cur_pkg resources.Bundle
	hasZip := false
	if exe_path, err := resources.ExecutablePath(); err == nil {
		if exe, err = resources.OpenZip(exe_path); err == nil {
			//            log.Println("bundle resource zip",exe_path)
			hasZip = true
			resources.DefaultBundle = append(resources.DefaultBundle, exe)
		}
		if err != nil {
			log.Println("bundle resource  zip failed")
		}
		if !hasZip {
			exe_dir = resources.OpenFS(filepath.Dir(exe_path))
			resources.DefaultBundle = append(resources.DefaultBundle, exe_dir)
		}
	}
	if !hasZip {
		cwd = resources.OpenFS(".")
		cur_pkg = resources.OpenAutoBundle(resources.OpenCurrentPackage)
		resources.DefaultBundle = append(resources.DefaultBundle, cwd, cur_pkg)
	}
}

//func init() {
//	var cwd, cur_pkg, exe_dir, exe Bundle
//	cwd = OpenFS(".")
//	cur_pkg = OpenAutoBundle(OpenCurrentPackage)
//
//	if exe_path, err := ExecutablePath(); err == nil {
//		exe_dir = OpenFS(filepath.Dir(exe_path))
//		if exe, err = OpenZip(exe_path); err == nil {
//			DefaultBundle = append(DefaultBundle, exe)
//		}
//	}
//
//	DefaultBundle = append(DefaultBundle, cwd, exe_dir, cur_pkg, exe)
//}