package chart

import (
	"context"
	"encoding/json"
	"github.com/samber/lo"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/interanl/pkg/db/transaction"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/interanl/pkg/domain"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/pkg/contextdata"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/pkg/xerr"

	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/api/internal/svc"
	"gitlab.fjmaimaimai.com/allied-creation/sumifcc-bchart/cmd/chart-server/api/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type UpdateChartLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewUpdateChartLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateChartLogic {
	return &UpdateChartLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UpdateChartLogic) UpdateChart(req *types.ChartUpdateRequest) (resp *types.ChartUpdateResponse, err error) {
	var (
		conn         = l.svcCtx.DefaultDBConn()
		chart        *domain.Chart
		chartSetting *domain.ChartSetting
		tenantId     = contextdata.GetTenantFromCtx(l.ctx)
	)

	if chart, err = l.svcCtx.ChartRepository.FindOne(l.ctx, conn, req.Id); err != nil || chart.TenantId != tenantId {
		return nil, xerr.NewErrMsgErr("图表不存在", err)
	}
	if chartSetting, err = l.svcCtx.ChartSettingRepository.FindOne(l.ctx, conn, req.Id); err != nil {
		return nil, xerr.NewErrMsgErr("图表配置不存在", err)
	}
	oldDataSource := chartSetting.DataSourceIds

	if err = transaction.UseTrans(l.ctx, l.svcCtx.DB, func(ctx context.Context, conn transaction.Conn) error {
		if len(req.Cover) > 0 && chart.Cover != req.Cover {
			chart.Cover = req.Cover
		}
		if chart, err = l.svcCtx.ChartRepository.UpdateWithVersion(l.ctx, conn, chart); err != nil {
			return err
		}
		chartProperty := types.NewProperty(req.ChartProperty)
		chartSetting.Title = chartProperty.Title
		chartSetting.TableAbility = chartProperty.TableAbility
		chartSetting.Series = chartProperty.Series
		chartSetting.Other = chartProperty.Other
		chartSetting.DataSourceIds = chartProperty.GetAllDataSourceId()
		if chartSetting, err = l.svcCtx.ChartSettingRepository.UpdateWithVersion(l.ctx, conn, chartSetting); err != nil {
			return err
		}
		//对比更新前后数据源
		left, right := lo.Difference(chartSetting.DataSourceIds, oldDataSource)
		//同步
		if len(left) > 0 {
			for _, sourceId := range left {
				pusher := &types.SyncTableDataPusher{
					CompanyId: tenantId,
					ObjectId:  int(sourceId),
				}
				mBytes, _ := json.Marshal(pusher)
				_, _ = l.svcCtx.Redis.LpushCtx(l.ctx, l.svcCtx.Config.Name+":table_data", string(mBytes))
			}
		}
		//删除
		if len(right) > 0 {
			for _, sourceId := range right {
				//验证其他图表是否使用
				used, err := l.svcCtx.ChartSettingRepository.CheckUseDataSource(l.ctx, conn, int(sourceId))
				if err == nil && !used { //未使用，删除
					err = l.svcCtx.ObjectTableDataRepository.DropTable(l.ctx, conn, int(sourceId))
					if err != nil {
						return err
					}
				}
			}
		}
		return nil
	}, true); err != nil {
		return nil, xerr.NewErrMsgErr("创建失败", err)
	}
	return
}