-
Notifications
You must be signed in to change notification settings - Fork 1
/
device_integration_repo.go
142 lines (119 loc) · 4.01 KB
/
device_integration_repo.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
//go:generate mockgen -source device_integration_repo.go -destination mocks/device_integration_repo_mock.go -package mocks
package repositories
import (
"context"
"database/sql"
"fmt"
"github.com/DIMO-Network/device-definitions-api/internal/infrastructure/db/models"
"github.com/DIMO-Network/device-definitions-api/internal/infrastructure/exceptions"
"github.com/DIMO-Network/shared/db"
"github.com/pkg/errors"
"github.com/segmentio/ksuid"
"github.com/volatiletech/sqlboiler/v4/boil"
)
type DeviceIntegrationRepository interface {
Create(ctx context.Context, deviceDefinitionID string, integrationID string, region string, features []map[string]interface{}) (*models.DeviceIntegration, error)
}
type deviceIntegrationRepository struct {
DBS func() *db.ReaderWriter
}
type PowertrainType string
const (
ICE PowertrainType = "ICE"
HEV PowertrainType = "HEV"
PHEV PowertrainType = "PHEV"
BEV PowertrainType = "BEV"
FCEV PowertrainType = "FCEV"
)
func (p PowertrainType) String() string {
return string(p)
}
// IntegrationsMetadata represents json stored in integrations table metadata jsonb column
type IntegrationsMetadata struct {
AutoPiDefaultTemplateID int `json:"autoPiDefaultTemplateId"`
AutoPiPowertrainToTemplateID map[PowertrainType]int `json:"autoPiPowertrainToTemplateId,omitempty"`
}
func NewDeviceIntegrationRepository(dbs func() *db.ReaderWriter) DeviceIntegrationRepository {
return &deviceIntegrationRepository{DBS: dbs}
}
func (r *deviceIntegrationRepository) Create(ctx context.Context, deviceDefinitionID string, integrationID string, region string, features []map[string]interface{}) (*models.DeviceIntegration, error) {
const (
AutoPiVendor = "AutoPi"
autoPiType = models.IntegrationTypeHardware
autoPiStyle = models.IntegrationStyleAddon
defaultTemplateID = 10
)
// Validate if the integrationID was sent.
if len(integrationID) > 0 {
integration, err := models.Integrations(models.IntegrationWhere.ID.EQ(integrationID)).
One(ctx, r.DBS().Reader)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return nil, &exceptions.NotFoundError{
Err: fmt.Errorf("could not find integration id: %s", integrationID),
}
}
}
integrationID = integration.ID
} else {
integration, err := models.Integrations(models.IntegrationWhere.Vendor.EQ(AutoPiVendor),
models.IntegrationWhere.Style.EQ(autoPiStyle), models.IntegrationWhere.Type.EQ(autoPiType)).
One(ctx, r.DBS().Reader)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
// create
im := IntegrationsMetadata{AutoPiDefaultTemplateID: defaultTemplateID}
integration = &models.Integration{
ID: ksuid.New().String(),
Vendor: AutoPiVendor,
Type: autoPiType,
Style: autoPiStyle,
}
_ = integration.Metadata.Marshal(im)
err = integration.Insert(ctx, r.DBS().Writer, boil.Infer())
if err != nil {
return nil, &exceptions.InternalError{
Err: errors.Wrap(err, "error inserting autoPi integration"),
}
}
} else {
return nil, &exceptions.InternalError{
Err: errors.Wrap(err, "error fetching autoPi integration from database"),
}
}
}
integrationID = integration.ID
}
di, err := models.DeviceIntegrations(models.DeviceIntegrationWhere.DeviceDefinitionID.EQ(deviceDefinitionID),
models.DeviceIntegrationWhere.IntegrationID.EQ(integrationID), models.DeviceIntegrationWhere.Region.EQ(region)).
One(ctx, r.DBS().Reader)
if err != nil {
if !errors.Is(err, sql.ErrNoRows) {
return nil, &exceptions.InternalError{
Err: err,
}
}
}
if di == nil {
di = &models.DeviceIntegration{
DeviceDefinitionID: deviceDefinitionID,
IntegrationID: integrationID,
Region: region,
}
if features != nil {
err = di.Features.Marshal(features)
if err != nil {
return nil, &exceptions.InternalError{
Err: err,
}
}
}
err = di.Insert(ctx, r.DBS().Writer, boil.Infer())
if err != nil {
return nil, &exceptions.InternalError{
Err: err,
}
}
}
return di, nil
}