view pkg/imports/modelconvert.go @ 5718:3d497077f888 uploadwg

Implemented direct file upload as alternative import method for WG. For testing and data corrections it is useful to be able to import waterway gauges data directly by uploading a xml file.
author Sascha Wilde <wilde@sha-bang.de>
date Thu, 18 Apr 2024 19:23:19 +0200
parents 6270951dda28
children
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, 2020, 2021 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 (
	"gemma.intevation.de/gemma/pkg/models"
)

var kindToImportModel = map[JobKind]func() any{
	BNJobKind:          func() any { return new(models.BottleneckImport) },
	GMJobKind:          func() any { return new(models.GaugeMeasurementImport) },
	FAJobKind:          func() any { return new(models.FairwayAvailabilityImport) },
	WXJobKind:          func() any { return FindJobCreator(WXJobKind).Create() },
	WAJobKind:          func() any { return new(models.WaterwayAreaImport) },
	WGJobKind:          func() any { return new(models.WaterwayGaugeImport) },
	DMVJobKind:         func() any { return new(models.DistanceMarksVirtualImport) },
	FDJobKind:          func() any { return new(models.FairwayDimensionImport) },
	DMAJobKind:         func() any { return new(models.DistanceMarksAshoreImport) },
	BCNLATHYDROJobKind: func() any { return FindJobCreator(BCNLATHYDROJobKind).Create() },
	BCNLATIENCJobKind:  func() any { return FindJobCreator(BCNLATIENCJobKind).Create() },
	BOYCARJobKind:      func() any { return FindJobCreator(BOYCARJobKind).Create() },
	BOYLATHYDROJobKind: func() any { return FindJobCreator(BOYLATHYDROJobKind).Create() },
	BOYLATIENCJobKind:  func() any { return FindJobCreator(BOYLATIENCJobKind).Create() },
	BOYSAWJobKind:      func() any { return FindJobCreator(BOYSAWJobKind).Create() },
	BOYSPPJobKind:      func() any { return FindJobCreator(BOYSPPJobKind).Create() },
	DAYMARHYDROJobKind: func() any { return FindJobCreator(DAYMARHYDROJobKind).Create() },
	DAYMARIENCJobKind:  func() any { return FindJobCreator(DAYMARIENCJobKind).Create() },
	LIGHTSJobKind:      func() any { return FindJobCreator(LIGHTSJobKind).Create() },
	RTPBCNJobKind:      func() any { return FindJobCreator(RTPBCNJobKind).Create() },
	TOPMARJobKind:      func() any { return FindJobCreator(TOPMARJobKind).Create() },
	NOTMRKJobKind:      func() any { return FindJobCreator(NOTMRKJobKind).Create() },
	STJobKind:          func() any { return new(models.StretchImport) },
	SECJobKind:         func() any { return new(models.SectionImport) },
	DSECJobKind:        func() any { return new(models.SectionDelete) },
	DSTJobKind:         func() any { return new(models.StretchDelete) },
	DSRJobKind:         func() any { return new(models.SoundingResultDelete) },
	ReportJobKind:      func() any { return FindJobCreator(ReportJobKind).Create() },
	StatsUpdateJobKind: func() any { return FindJobCreator(StatsUpdateJobKind).Create() },
}

// ImportModelForJobKind returns the constructor function to
// de-serialize an incoming JSON REST represention of an import.
func ImportModelForJobKind(kind JobKind) func() any {
	return kindToImportModel[kind]
}

var convertModel = map[JobKind]func(any) any{

	BNJobKind: func(input any) any {
		bi := input.(*models.BottleneckImport)
		return &Bottleneck{
			URL:       bi.URL,
			Tolerance: bi.Tolerance,
			Insecure:  bi.Insecure,
		}
	},

	GMJobKind: func(input any) any {
		gi := input.(*models.GaugeMeasurementImport)
		return &GaugeMeasurement{
			URL:      gi.URL,
			Insecure: gi.Insecure,
		}
	},

	FAJobKind: func(input any) any {
		fai := input.(*models.FairwayAvailabilityImport)
		return &FairwayAvailability{
			URL:      fai.URL,
			Insecure: fai.Insecure,
		}
	},

	WAJobKind: func(input any) any {
		wai := input.(*models.WaterwayAreaImport)
		return &WaterwayArea{
			URL:         wai.URL,
			FeatureType: wai.FeatureType,
			SortBy:      nilString(wai.SortBy),
			User:        nilString(wai.User),
			Password:    nilString(wai.Password),
		}
	},

	WGJobKind: func(input any) any {
		wgi := input.(*models.WaterwayGaugeImport)
		return &WaterwayGauge{
			URL:      wgi.URL,
			Username: nilString(wgi.User),
			Password: nilString(wgi.Password),
			Insecure: wgi.Insecure,
		}
	},

	DMVJobKind: func(input any) any {
		dmvi := input.(*models.DistanceMarksVirtualImport)
		return &DistanceMarksVirtual{
			URL:      dmvi.URL,
			Username: nilString(dmvi.User),
			Password: nilString(dmvi.Password),
			Insecure: dmvi.Insecure,
		}
	},

	FDJobKind: func(input any) any {
		fdi := input.(*models.FairwayDimensionImport)
		return &FairwayDimension{
			URL:                fdi.URL,
			FeatureType:        fdi.FeatureType,
			SortBy:             nilString(fdi.SortBy),
			LOS:                fdi.LOS,
			MinWidth:           fdi.MinWidth,
			MaxWidth:           fdi.MaxWidth,
			Depth:              fdi.Depth,
			SourceOrganization: fdi.SourceOrganization,
			User:               nilString(fdi.User),
			Password:           nilString(fdi.Password),
		}
	},

	DMAJobKind: func(input any) any {
		dmai := input.(*models.DistanceMarksAshoreImport)
		return &DistanceMarksAshore{
			URL:         dmai.URL,
			FeatureType: dmai.FeatureType,
			SortBy:      nilString(dmai.SortBy),
			User:        nilString(dmai.User),
			Password:    nilString(dmai.Password),
		}
	},

	STJobKind: func(input any) any {
		sti := input.(*models.StretchImport)
		return &Stretch{
			Name:      sti.Name,
			From:      sti.From,
			To:        sti.To,
			Tolerance: sti.Tolerance,
			ObjNam:    sti.ObjNam,
			NObjNam:   sti.NObjNam,
			Source:    sti.Source,
			Date:      sti.Date,
			Countries: sti.Countries,
		}
	},

	SECJobKind: func(input any) any {
		seci := input.(*models.SectionImport)
		return &Section{
			Name:      seci.Name,
			From:      seci.From,
			To:        seci.To,
			Tolerance: seci.Tolerance,
			ObjNam:    seci.ObjNam,
			NObjNam:   seci.NObjNam,
			Source:    seci.Source,
			Date:      seci.Date,
		}
	},

	DSECJobKind: func(input any) any {
		dsec := input.(*models.SectionDelete)
		return &DeleteSection{ID: dsec.ID}
	},

	DSTJobKind: func(input any) any {
		dst := input.(*models.StretchDelete)
		return &DeleteStretch{ID: dst.ID}
	},

	DSRJobKind: func(input any) any {
		dsr := input.(*models.SoundingResultDelete)
		return &DeleteSoundingResult{
			BottleneckID: dsr.BottleneckID,
			Date:         dsr.Date,
		}
	},
}

func nilString(s *string) string {
	if s != nil {
		return *s
	}
	return ""
}

// ConvertToInternal converts an external JSON REST represention
// of an import into the internal one store in the import queue.
func ConvertToInternal(kind JobKind, src any) any {
	fn := convertModel[kind]
	if fn == nil {
		return src
	}
	return fn(src)
}