NotificationSettingsRepository.go

/*
 * Copyright (c) 2020 Devtron Labs
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package repository

import (
	"github.com/devtron-labs/devtron/internal/sql/models"
	"github.com/go-pg/pg"
)

type NotificationSettingsRepository interface {
	FindNSViewCount() (int, error)
	SaveNotificationSettingsConfig(notificationSettingsView *NotificationSettingsView, tx *pg.Tx) (*NotificationSettingsView, error)
	FindNotificationSettingsViewById(id int) (*NotificationSettingsView, error)
	FindNotificationSettingsViewByIds(id []*int) ([]*NotificationSettingsView, error)
	UpdateNotificationSettingsView(notificationSettingsView *NotificationSettingsView, tx *pg.Tx) (*NotificationSettingsView, error)
	SaveNotificationSetting(notificationSettings *NotificationSettings, tx *pg.Tx) (*NotificationSettings, error)
	UpdateNotificationSettings(notificationSettings *NotificationSettings, tx *pg.Tx) (*NotificationSettings, error)
	FindNotificationSettingsByViewId(viewId int) ([]NotificationSettings, error)
	SaveAllNotificationSettings(notificationSettings []NotificationSettings, tx *pg.Tx) (int, error)
	DeleteNotificationSettingsByConfigId(viewId int, tx *pg.Tx) (int, error)
	FindAll(offset int, size int) ([]*NotificationSettingsView, error)
	DeleteNotificationSettingsViewById(id int, tx *pg.Tx) (int, error)

	FindNotificationSettingDeploymentOptions(settingRequest *SearchRequest) ([]*SettingOptionDTO, error)
	FindNotificationSettingBuildOptions(settingRequest *SearchRequest) ([]*SettingOptionDTO, error)
	FetchNotificationSettingGroupBy(viewId int) ([]NotificationSettings, error)
}

type NotificationSettingsRepositoryImpl struct {
	dbConnection *pg.DB
}

func NewNotificationSettingsRepositoryImpl(dbConnection *pg.DB) *NotificationSettingsRepositoryImpl {
	return &NotificationSettingsRepositoryImpl{dbConnection: dbConnection}
}

type NotificationSettingsView struct {
	tableName struct{} `sql:"notification_settings_view" pg:",discard_unknown_columns"`
	Id        int      `sql:"id,pk"`
	Config    string   `sql:"config"`
	//ConfigName    string   `sql:"config_name"`
	//AppId         *int     `sql:"app_id"`
	//EnvironmentId *int     `sql:"env_id"`
	models.AuditLog
}

type NotificationSettingsViewWithAppEnv struct {
	Id              int    `json:"id"`
	AppId           *int   `json:"app_id"`
	EnvId           *int   `json:"env_id"`
	ConfigName      string `sql:"config_name"`
	Config          string `sql:"config"`
	AppName         string `json:"app_name"`
	EnvironmentName string `json:"env_name"`
	models.AuditLog
}

type NotificationSettings struct {
	tableName    struct{} `sql:"notification_settings"`
	Id           int      `sql:"id,pk"`
	TeamId       *int     `sql:"team_id"`
	AppId        *int     `sql:"app_id"`
	EnvId        *int     `sql:"env_id"`
	PipelineId   *int     `sql:"pipeline_id"`
	PipelineType string   `sql:"pipeline_type"`
	EventTypeId  int      `sql:"event_type_id"`
	Config       string   `sql:"config"`
	ViewId       int      `sql:"view_id"`
}

type SettingOptionDTO struct {
	//TeamId       int    `json:"-"`
	//AppId        int    `json:"-"`
	//EnvId        int    `json:"-"`
	PipelineId      int    `json:"pipelineId"`
	PipelineName    string `json:"pipelineName"`
	PipelineType    string `json:"pipelineType"`
	AppName         string `json:"appName"`
	EnvironmentName string `json:"environmentName,omitempty"`
}

func (impl *NotificationSettingsRepositoryImpl) FindNSViewCount() (int, error) {
	count, err := impl.dbConnection.Model(&NotificationSettingsView{}).Count()
	if err != nil {
		return 0, err
	}
	return count, nil
}

func (impl *NotificationSettingsRepositoryImpl) FindAll(offset int, size int) ([]*NotificationSettingsView, error) {
	var ns []*NotificationSettingsView
	err := impl.dbConnection.Model(&ns).Order("created_on desc").Offset(offset).Limit(size).Select()
	if err != nil {
		return nil, err
	}
	return ns, err
}

func (impl *NotificationSettingsRepositoryImpl) SaveNotificationSettingsConfig(notificationSettingsView *NotificationSettingsView, tx *pg.Tx) (*NotificationSettingsView, error) {
	err := tx.Insert(notificationSettingsView)
	if err != nil {
		return nil, err
	}
	return notificationSettingsView, nil
}

func (impl *NotificationSettingsRepositoryImpl) FindNotificationSettingsViewById(id int) (*NotificationSettingsView, error) {
	notificationSettingsView := &NotificationSettingsView{}
	err := impl.dbConnection.Model(notificationSettingsView).Where("id = ?", id).Select()
	if err != nil {
		return nil, err
	}
	return notificationSettingsView, nil
}

func (impl *NotificationSettingsRepositoryImpl) FindNotificationSettingsViewByIds(ids []*int) ([]*NotificationSettingsView, error) {
	var notificationSettingsView []*NotificationSettingsView
	if ids == nil || len(ids) == 0 {
		return notificationSettingsView, nil
	}
	err := impl.dbConnection.Model(&notificationSettingsView).Where("id in (?)", pg.In(ids)).Select()
	if err != nil {
		return nil, err
	}
	return notificationSettingsView, nil
}

func (impl *NotificationSettingsRepositoryImpl) UpdateNotificationSettingsView(notificationSettingsView *NotificationSettingsView, tx *pg.Tx) (*NotificationSettingsView, error) {
	err := tx.Update(notificationSettingsView)
	if err != nil {
		return nil, err
	}
	return notificationSettingsView, nil
}

func (impl *NotificationSettingsRepositoryImpl) SaveNotificationSetting(notificationSettings *NotificationSettings, tx *pg.Tx) (*NotificationSettings, error) {
	err := tx.Insert(notificationSettings)
	if err != nil {
		return nil, err
	}
	return notificationSettings, nil
}

func (impl *NotificationSettingsRepositoryImpl) SaveAllNotificationSettings(notificationSettings []NotificationSettings, tx *pg.Tx) (int, error) {
	res, err := tx.Model(&notificationSettings).Insert()
	if err != nil {
		return 0, err
	}
	return res.RowsAffected(), nil
}

func (impl *NotificationSettingsRepositoryImpl) UpdateNotificationSettings(notificationSettings *NotificationSettings, tx *pg.Tx) (*NotificationSettings, error) {
	err := tx.Update(notificationSettings)
	if err != nil {
		return nil, err
	}
	return notificationSettings, nil
}

func (impl *NotificationSettingsRepositoryImpl) FindNotificationSettingsByViewId(viewId int) ([]NotificationSettings, error) {
	var notificationSettings []NotificationSettings
	err := impl.dbConnection.Model(&notificationSettings).Where("view_id = ?", viewId).Select()
	if err != nil {
		return nil, err
	}
	return notificationSettings, nil
}

func (impl *NotificationSettingsRepositoryImpl) DeleteNotificationSettingsByConfigId(viewId int, tx *pg.Tx) (int, error) {
	var notificationSettings *NotificationSettings
	res, err := tx.Model(notificationSettings).Where("view_id = ?", viewId).Delete()
	if err != nil {
		return 0, err
	}
	return res.RowsAffected(), nil
}

func (impl *NotificationSettingsRepositoryImpl) DeleteNotificationSettingsViewById(id int, tx *pg.Tx) (int, error) {
	var notificationSettingsView *NotificationSettingsView
	res, err := tx.Model(notificationSettingsView).Where("id = ?", id).Delete()
	if err != nil {
		return 0, err
	}
	return res.RowsAffected(), nil
}

func (impl *NotificationSettingsRepositoryImpl) FindNotificationSettingDeploymentOptions(settingRequest *SearchRequest) ([]*SettingOptionDTO, error) {
	var settingOption []*SettingOptionDTO
	query := "SELECT p.id as pipeline_id,p.pipeline_name, env.environment_name, a.app_name FROM pipeline p" +
		" INNER JOIN app a on a.id=p.app_id" +
		" INNER JOIN environment env on env.id = p.environment_id"
	query = query + " WHERE p.deleted = false"

	if len(settingRequest.TeamId) > 0 {
		query = query + " AND a.team_id in (?)"
	} else if len(settingRequest.EnvId) > 0 {
		query = query + " AND p.environment_id in (?)"
	} else if len(settingRequest.AppId) > 0 {
		query = query + " AND p.app_id in (?)"
	} else if len(settingRequest.PipelineName) > 0 {
		query = query + " AND p.pipeline_name like (?)"
	}
	query = query + " GROUP BY 1,2,3,4;"

	var err error
	if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) == 0 && len(settingRequest.AppId) == 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId))
	} else if len(settingRequest.TeamId) == 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) == 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.EnvId))
	} else if len(settingRequest.TeamId) == 0 && len(settingRequest.EnvId) == 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.AppId))
	} else if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) == 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId), pg.In(settingRequest.EnvId))
	} else if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) == 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId), pg.In(settingRequest.AppId))
	} else if len(settingRequest.TeamId) == 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.EnvId), pg.In(settingRequest.AppId))
	} else if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId), pg.In(settingRequest.EnvId), pg.In(settingRequest.AppId))
	} else if len(settingRequest.PipelineName) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, settingRequest.PipelineName)
	}
	if err != nil {
		return nil, err
	}
	return settingOption, err
}

func (impl *NotificationSettingsRepositoryImpl) FindNotificationSettingBuildOptions(settingRequest *SearchRequest) ([]*SettingOptionDTO, error) {
	var settingOption []*SettingOptionDTO
	query := "SELECT cip.id as pipeline_id,cip.name as pipeline_name, a.app_name from ci_pipeline cip" +
		" INNER JOIN app a on a.id = cip.app_id" +
		" INNER JOIN team t on t.id= a.team_id"
	if len(settingRequest.EnvId) > 0 {
		query = query + " INNER JOIN ci_artifact cia on cia.pipeline_id = cip.id"
		query = query + " INNER JOIN cd_workflow wf on wf.ci_artifact_id = cia.id"
		query = query + " INNER JOIN pipeline p on p.id = wf.pipeline_id"
	}
	query = query + " WHERE cip.deleted = false"

	if len(settingRequest.TeamId) > 0 {
		query = query + " AND a.team_id in (?)"
	} else if len(settingRequest.EnvId) > 0 {
		query = query + " AND p.environment_id in (?)"
	} else if len(settingRequest.AppId) > 0 {
		query = query + " AND cip.app_id in (?)"
	} else if len(settingRequest.PipelineName) > 0 {
		query = query + " AND cip.name like (?)"
	}
	query = query + " GROUP BY 1,2,3;"
	var err error
	if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) == 0 && len(settingRequest.AppId) == 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId))
	} else if len(settingRequest.TeamId) == 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) == 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.EnvId))
	} else if len(settingRequest.TeamId) == 0 && len(settingRequest.EnvId) == 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.AppId))
	} else if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) == 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId), pg.In(settingRequest.EnvId))
	} else if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) == 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId), pg.In(settingRequest.AppId))
	} else if len(settingRequest.TeamId) == 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.EnvId), pg.In(settingRequest.AppId))
	} else if len(settingRequest.TeamId) > 0 && len(settingRequest.EnvId) > 0 && len(settingRequest.AppId) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, pg.In(settingRequest.TeamId), pg.In(settingRequest.EnvId), pg.In(settingRequest.AppId))
	} else if len(settingRequest.PipelineName) > 0 {
		_, err = impl.dbConnection.Query(&settingOption, query, settingRequest.PipelineName)
	}
	if err != nil {
		return nil, err
	}
	return settingOption, err
}

type SearchRequest struct {
	TeamId       []*int `json:"teamId" validate:"number"`
	EnvId        []*int `json:"envId" validate:"number"`
	AppId        []*int `json:"appId" validate:"number"`
	PipelineName string `json:"pipelineName"`
	UserId       int32  `json:"-"`
}

func (impl *NotificationSettingsRepositoryImpl) FetchNotificationSettingGroupBy(viewId int) ([]NotificationSettings, error) {
	var ns []NotificationSettings
	queryTemp := "select ns.team_id,ns.env_id,ns.app_id,ns.pipeline_id,ns.pipeline_type from notification_settings ns" +
		" where ns.view_id=? group by 1,2,3,4,5;"
	_, err := impl.dbConnection.Query(&ns, queryTemp, viewId)
	if err != nil {
		return nil, err
	}
	return ns, err
}