Mercurial > gemma
view pkg/imports/config.go @ 1677:53304db85888
Waterway axis import: Added route for manual import.
author | Sascha L. Teichmann <sascha.teichmann@intevation.de> |
---|---|
date | Wed, 26 Dec 2018 10:46:17 +0100 |
parents | 06f08458d666 |
children | ad5e1cddaa09 |
line wrap: on
line source
// 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 imports import ( "context" "database/sql" "encoding/json" "fmt" "gemma.intevation.de/gemma/pkg/auth" "github.com/robfig/cron" ) type ( CronSpec string ImportKind string Config struct { Kind ImportKind `json:"kind"` SendEMail bool `json:"send-email"` AutoAccept bool `json:"auto-accept"` Cron *CronSpec `json:"cron"` URL *string `json:"url"` } IDConfig struct { ID int64 `json:"id"` User string `json:"user"` Kind ImportKind `json:"kind"` SendEMail bool `json:"send-email"` AutoAccept bool `json:"auto-accept"` Cron *CronSpec `json:"cron,omitempty"` URL *string `json:"url,omitempty"` } ) func (ik *ImportKind) UnmarshalJSON(data []byte) error { var s string if err := json.Unmarshal(data, &s); err != nil { return err } if !HasImportKindName(s) { return fmt.Errorf("Unknown kind '%s'", s) } *ik = ImportKind(s) return nil } func (cs *CronSpec) UnmarshalJSON(data []byte) error { var spec string if err := json.Unmarshal(data, &spec); err != nil { return err } if _, err := cron.Parse(spec); err != nil { return err } *cs = CronSpec(spec) return nil } const ( configUser = "sys_admin" loadConfigSQL = ` SELECT username, kind, send_email, auto_accept, cron, url FROM waterway.import_configuration WHERE id = $1` ) func loadIDConfig(id int64) (*IDConfig, error) { ctx := context.Background() cfg := &IDConfig{ID: id} err := auth.RunAs(ctx, configUser, func(conn *sql.Conn) error { var kind ImportKind var cron, url sql.NullString if err := conn.QueryRowContext(ctx, loadConfigSQL, id).Scan( &cfg.User, &kind, &cfg.SendEMail, &cfg.AutoAccept, &cron, &url, ); err != nil { return err } cfg.Kind = ImportKind(kind) if cron.Valid { c := CronSpec(cron.String) cfg.Cron = &c } if url.Valid { cfg.URL = &url.String } return nil }) switch { case err == sql.ErrNoRows: return nil, nil case err != nil: return nil, err } return cfg, nil }