sliproad/files/disk.go

package files

import (
	"io"
	"io/ioutil"
	"os"
	"strings"
)

type DiskProvider struct{
	FileProvider
}

func (dp *DiskProvider) Setup(args map[string]string) bool {
	return true
}

func (dp *DiskProvider) GetDirectory(path string) Directory {
	rp := strings.Join([]string{dp.Location,path}, "/")
	fileDir, err := ioutil.ReadDir(rp)
	if err != nil {
		_ = os.MkdirAll(path, 0644)
	}
	var fileList []FileInfo

	for _, file := range fileDir {
		info := FileInfo{
			IsDirectory: file.IsDir(),
			Name: file.Name(),
		}
		if !info.IsDirectory {
			split := strings.Split(file.Name(), ".")
			info.Extension = split[len(split) - 1]
		}
		fileList = append(fileList, info)
	}

	return Directory{
		Path: rp,
		Files: fileList,
	}
}

func (dp *DiskProvider) ViewFile(path string, w io.Writer) {
	file := strings.Join([]string{dp.Location,path}, "/")
	fileReader, err := os.Open(file)
	if err != nil {
		return
	}
	_, err = io.Copy(w, fileReader)
	if err != nil {
		return
	}
}

func (dp *DiskProvider) SaveFile(contents []byte, path string) bool {
	err := ioutil.WriteFile(path, contents, 0600)
	if err != nil {
		return false
	}
	return true
}

func (dp *DiskProvider) DetermineType(path string) string {
	rp := strings.Join([]string{dp.Location,path}, "/")
	file, err := os.Stat(rp)
	if err != nil {
		return ""
	}
	if file.IsDir() {
		return "directory"
	}

	return "file"
}