Yuri Tseretyan
d15e1ad8d0
* introduce ExtraConfigsCrypto * delete getLastConfiguration and move all code into method Get * update legacy store to encrypt ExtraConfigs same way we do in MultiOrgAlertmanager * update legacy store init * delete PersistConfig * remove unnecessary unmarshalling * add better error * add tests for legacy store * add tests that cover extra config validation * add integration test for conflicting routes --------- Signed-off-by: Yuri Tseretyan <yuriy.tseretyan@grafana.com>
94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
package legacy_storage
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/services/ngalert/api/tooling/definitions"
|
|
"github.com/grafana/grafana/pkg/services/ngalert/models"
|
|
)
|
|
|
|
type crypto interface {
|
|
EncryptExtraConfigs(ctx context.Context, config *definitions.PostableUserConfig) error
|
|
DecryptExtraConfigs(ctx context.Context, config *definitions.PostableUserConfig) error
|
|
}
|
|
|
|
type amConfigStore interface {
|
|
GetLatestAlertmanagerConfiguration(ctx context.Context, orgID int64) (*models.AlertConfiguration, error)
|
|
UpdateAlertmanagerConfiguration(ctx context.Context, cmd *models.SaveAlertmanagerConfigurationCmd) error
|
|
}
|
|
|
|
func DeserializeAlertmanagerConfig(config []byte) (*definitions.PostableUserConfig, error) {
|
|
result := definitions.PostableUserConfig{}
|
|
if err := json.Unmarshal(config, &result); err != nil {
|
|
return nil, makeErrBadAlertmanagerConfiguration(err)
|
|
}
|
|
return &result, nil
|
|
}
|
|
|
|
func SerializeAlertmanagerConfig(config definitions.PostableUserConfig) ([]byte, error) {
|
|
return json.Marshal(config)
|
|
}
|
|
|
|
type ConfigRevision struct {
|
|
Config *definitions.PostableUserConfig
|
|
ConcurrencyToken string
|
|
Version string
|
|
}
|
|
type alertmanagerConfigStoreImpl struct {
|
|
store amConfigStore
|
|
crypto crypto
|
|
}
|
|
|
|
func NewAlertmanagerConfigStore(store amConfigStore, crypto crypto) *alertmanagerConfigStoreImpl {
|
|
return &alertmanagerConfigStoreImpl{store: store, crypto: crypto}
|
|
}
|
|
|
|
func (a alertmanagerConfigStoreImpl) Get(ctx context.Context, orgID int64) (*ConfigRevision, error) {
|
|
alertManagerConfig, err := a.store.GetLatestAlertmanagerConfiguration(ctx, orgID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if alertManagerConfig == nil {
|
|
return nil, ErrNoAlertmanagerConfiguration.Errorf("")
|
|
}
|
|
|
|
concurrencyToken := alertManagerConfig.ConfigurationHash
|
|
cfg, err := DeserializeAlertmanagerConfig([]byte(alertManagerConfig.AlertmanagerConfiguration))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = a.crypto.DecryptExtraConfigs(ctx, cfg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to decrypt extra configurations: %w", err)
|
|
}
|
|
|
|
return &ConfigRevision{
|
|
Config: cfg,
|
|
ConcurrencyToken: concurrencyToken,
|
|
Version: alertManagerConfig.ConfigurationVersion,
|
|
}, nil
|
|
}
|
|
|
|
func (a alertmanagerConfigStoreImpl) Save(ctx context.Context, revision *ConfigRevision, orgID int64) error {
|
|
err := a.crypto.EncryptExtraConfigs(ctx, revision.Config)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to encrypt extra configurations: %w", err)
|
|
}
|
|
|
|
serialized, err := SerializeAlertmanagerConfig(*revision.Config)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return a.store.UpdateAlertmanagerConfiguration(ctx, &models.SaveAlertmanagerConfigurationCmd{
|
|
AlertmanagerConfiguration: string(serialized),
|
|
ConfigurationVersion: revision.Version,
|
|
FetchedConfigurationHash: revision.ConcurrencyToken,
|
|
Default: false,
|
|
OrgID: orgID,
|
|
})
|
|
}
|