feat: add directory index page option

Author: marjune163

README.md

@@ -52,6 +52,10 @@ server [options]
     Use virtual empty directory as root directory.
     Useful to share alias directories only.
 
+-I|--dir-index <file> ...
+    Specify default index file for directory.
+    Directly mounted files by alias are not considered.
+
 -a|--alias <separator><url-path><separator><fs-path> ...
     Set path alias. e.g. ":/doc:/usr/share/doc"
 

src/param/cli.go

@@ -26,6 +26,9 @@ func init() {
 	err = options.AddFlags("emptyroot", []string{"-R", "--empty-root"}, "GHFS_EMPTY_ROOT", "use virtual empty root directory")
 	serverErrHandler.CheckFatal(err)
 
+	err = options.AddFlagsValues("dirindexes", []string{"-I", "--dir-index"}, "GHFS_DIR_INDEX", nil, "default index page for directory")
+	serverErrHandler.CheckFatal(err)
+
 	err = options.AddFlagsValues("aliases", []string{"-a", "--alias"}, "", nil, "set alias path, <sep><url><sep><path>, e.g. :/doc:/usr/share/doc")
 	serverErrHandler.CheckFatal(err)
 
@@ -197,6 +200,10 @@ func doParseCli() []*Param {
 		root, _ = util.NormalizeFsPath(root)
 		param.Root = root
 
+		// dir indexes
+		dirIndexes, _ := result.GetStrings("dirindexes")
+		param.DirIndexes = normalizeFilenames(dirIndexes)
+
 		// certificate
 		key, _ := result.GetString("key")
 		cert, _ := result.GetString("cert")

src/param/main.go

@@ -14,7 +14,8 @@ type Param struct {
 	Root      string
 	EmptyRoot bool
 
-	Aliases map[string]string
+	DirIndexes []string
+	Aliases    map[string]string
 
 	GlobalUpload bool
 	UploadUrls   []string

src/param/util.go

@@ -3,6 +3,7 @@ package param
 import (
 	"../util"
 	"path"
+	"path/filepath"
 	"strings"
 	"unicode/utf8"
 )
@@ -75,3 +76,25 @@ func normalizeFsPaths(inputs []string) []string {
 
 	return outputs
 }
+
+func normalizeFilenames(inputs []string) []string {
+	outputs := make([]string, 0, len(inputs))
+
+	for _, input := range inputs {
+		if len(input) == 0 {
+			continue
+		}
+
+		if strings.IndexByte(input, '/') >= 0 {
+			continue
+		}
+
+		if filepath.Separator != '/' && strings.IndexByte(input, filepath.Separator) >= 0 {
+			continue
+		}
+
+		outputs = append(outputs, input)
+	}
+
+	return outputs
+}

src/serverHandler/main.go

@@ -15,7 +15,8 @@ type handler struct {
 	emptyRoot bool
 	urlPrefix string
 
-	aliases aliases
+	dirIndexes []string
+	aliases    aliases
 
 	globalUpload bool
 	uploadUrls   []string
@@ -133,7 +134,8 @@ func NewHandler(
 		emptyRoot: emptyRoot,
 		urlPrefix: urlPrefix,
 
-		aliases: aliases,
+		dirIndexes: p.DirIndexes,
+		aliases:    aliases,
 
 		globalUpload: p.GlobalUpload,
 		uploadUrls:   p.UploadUrls,

src/serverHandler/responseData.go

@@ -274,6 +274,23 @@ func getStatusByErr(err error) int {
 	}
 }
 
+func (h *handler) stateIndexFile(baseItem os.FileInfo, baseDir string) (file *os.File, item os.FileInfo, err error) {
+	if baseItem == nil || !baseItem.IsDir() || h.emptyRoot || len(h.dirIndexes) == 0 {
+		return
+	}
+
+	for _, index := range h.dirIndexes {
+		file, item, err = stat(baseDir+"/"+index, true)
+		if err != nil && os.IsNotExist(err) {
+			continue
+		} else {
+			return
+		}
+	}
+
+	return nil, nil, nil
+}
+
 func (h *handler) getResponseData(r *http.Request) (data *responseData) {
 	requestUri := r.URL.Path
 	tailSlash := requestUri[len(requestUri)-1] == '/'
@@ -303,6 +320,20 @@ func (h *handler) getResponseData(r *http.Request) (data *responseData) {
 		status = getStatusByErr(_statErr)
 	}
 
+	indexFile, indexItem, _statIdxErr := h.stateIndexFile(item, reqFsPath)
+	if _statIdxErr != nil {
+		errs = append(errs, _statIdxErr)
+		status = getStatusByErr(_statIdxErr)
+	} else if indexFile != nil {
+		if indexItem != nil {
+			file.Close()
+			file = indexFile
+			item = indexItem
+		} else {
+			indexFile.Close()
+		}
+	}
+
 	itemName := getItemName(item, r)
 
 	subInfos, _readdirErr := readdir(file, item, needResponseBody(r.Method))