Mirrors/sftpgo
1
0
Fork 0
mirror of https://github.com/drakkan/sftpgo.git synced 2025-12-06 14:20:55 +03:00
Code Issues Packages Projects Releases Wiki Activity

WebDAV: allow to define custom MIME type mappings

Browse Source 
Fixes #1154

Signed-off-by: Nicola Murino <nicola.murino@gmail.com>
This commit is contained in:
Nicola Murino
2023-01-23 18:43:25 +01:00
parent 61199172d0
commit 2066ad7c83
11 changed files with 183 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
internal/config/config_test.go
View File

@@ -38,6 +38,7 @@ import (
"github.com/drakkan/sftpgo/v2/internal/sftpd"
"github.com/drakkan/sftpgo/v2/internal/smtp"
"github.com/drakkan/sftpgo/v2/internal/util"
"github.com/drakkan/sftpgo/v2/internal/webdavd"
)
const (
@@ -1031,6 +1032,72 @@ func TestFTPDBindingsFromEnv(t *testing.T) {
require.Equal(t, "cert.key", bindings[1].CertificateKeyFile)
}
func TestWebDAVMimeCache(t *testing.T) {
reset()
err := config.LoadConfig(configDir, "")
assert.NoError(t, err)
webdavdConf := config.GetWebDAVDConfig()
webdavdConf.Cache.MimeTypes.CustomMappings = []webdavd.CustomMimeMapping{
{
Ext: ".custom",
Mime: "application/custom",
},
}
cfg := map[string]any{
"webdavd": webdavdConf,
}
data, err := json.Marshal(cfg)
assert.NoError(t, err)
confName := tempConfigName + ".json"
configFilePath := filepath.Join(configDir, confName)
err = os.WriteFile(configFilePath, data, 0666)
assert.NoError(t, err)
reset()
err = config.LoadConfig(configDir, confName)
assert.NoError(t, err)
mappings := config.GetWebDAVDConfig().Cache.MimeTypes.CustomMappings
if assert.Len(t, mappings, 1) {
assert.Equal(t, ".custom", mappings[0].Ext)
assert.Equal(t, "application/custom", mappings[0].Mime)
}
// now add from env
os.Setenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__1__EXT", ".custom1")
os.Setenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__1__MIME", "application/custom1")
t.Cleanup(func() {
os.Unsetenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__0__EXT")
os.Unsetenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__0__MIME")
os.Unsetenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__1__EXT")
os.Unsetenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__1__MIME")
})
reset()
err = config.LoadConfig(configDir, confName)
assert.NoError(t, err)
mappings = config.GetWebDAVDConfig().Cache.MimeTypes.CustomMappings
if assert.Len(t, mappings, 2) {
assert.Equal(t, ".custom", mappings[0].Ext)
assert.Equal(t, "application/custom", mappings[0].Mime)
assert.Equal(t, ".custom1", mappings[1].Ext)
assert.Equal(t, "application/custom1", mappings[1].Mime)
}
// override from env
os.Setenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__0__EXT", ".custom0")
os.Setenv("SFTPGO_WEBDAVD__CACHE__MIME_TYPES__CUSTOM_MAPPINGS__0__MIME", "application/custom0")
reset()
err = config.LoadConfig(configDir, confName)
assert.NoError(t, err)
mappings = config.GetWebDAVDConfig().Cache.MimeTypes.CustomMappings
if assert.Len(t, mappings, 2) {
assert.Equal(t, ".custom0", mappings[0].Ext)
assert.Equal(t, "application/custom0", mappings[0].Mime)
assert.Equal(t, ".custom1", mappings[1].Ext)
assert.Equal(t, "application/custom1", mappings[1].Mime)
}
err = os.Remove(configFilePath)
assert.NoError(t, err)
}
func TestWebDAVBindingsFromEnv(t *testing.T) {
reset()