Mercurial > gemma
diff pkg/misc/http.go @ 2194:4d6979dedb11
Imports: Deduplicted file upload code.
author | Sascha L. Teichmann <sascha.teichmann@intevation.de> |
---|---|
date | Tue, 12 Feb 2019 11:09:55 +0100 |
parents | |
children | 9cbed444b8a4 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/pkg/misc/http.go Tue Feb 12 11:09:55 2019 +0100 @@ -0,0 +1,62 @@ +// This is Free Software under GNU Affero General Public License v >= 3.0 +// without warranty, see README.md and license for details. +// +// SPDX-License-Identifier: AGPL-3.0-or-later +// License-Filename: LICENSES/AGPL-3.0.txt +// +// Copyright (C) 2018 by via donau +// – Österreichische Wasserstraßen-Gesellschaft mbH +// Software engineering by Intevation GmbH +// +// Author(s): +// * Sascha L. Teichmann <sascha.teichmann@intevation.de> + +package misc + +import ( + "bufio" + "io" + "io/ioutil" + "net/http" + "os" + "path/filepath" + + "gemma.intevation.de/gemma/pkg/config" +) + +func StoreUploadedFile(req *http.Request, field, fname string, maxSize int64) (string, error) { + + // Check for direct upload. + f, _, err := req.FormFile(field) + if err != nil { + return "", err + } + defer f.Close() + + dir, err := ioutil.TempDir(config.TmpDir(), field) + if err != nil { + return "", err + } + + o, err := os.Create(filepath.Join(dir, fname)) + if err != nil { + os.RemoveAll(dir) + return "", err + } + + out := bufio.NewWriter(o) + + if _, err = io.Copy(out, io.LimitReader(f, maxSize)); err != nil { + o.Close() + os.RemoveAll(dir) + return "", err + } + + if err = out.Flush(); err != nil { + o.Close() + os.RemoveAll(dir) + return "", err + } + + return dir, nil +}