/*
 * Copyright (C) @2021 Webank Group Holding Limited
 * <p>
 * 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
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * 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 ctrl

import (
	"github.com/webankfintech/dockin-opserver/internal/cache/keys"
	"github.com/webankfintech/dockin-opserver/internal/cache/redis"
	"github.com/webankfintech/dockin-opserver/internal/log"
)

type AllowCmd struct {
	redisClient *redis.RedisClient
}

func NewAllowCmd() *AllowCmd {
	return &AllowCmd{}
}

func (b *AllowCmd) LoadFromCache() (raw, common []string) {
	raw, err := b.redisClient.SMembers(keys.GetRawCmdRedisKey())
	if err != nil {
		log.Logger.Warnf(err.Error())
	}

	common, err = b.redisClient.SMembers(keys.GetCommonCmdRedisKey())
	if err != nil {
		log.Logger.Warnf(err.Error())
	}
	return raw, common
}

func (b *AllowCmd) AddRaw(rawCmd []string) error {
	if err := b.redisClient.SAdd(keys.GetRawCmdRedisKey(), rawCmd); err != nil {
		log.Logger.Warnf(err.Error())
		return err
	}

	return nil
}

func (b *AllowCmd) RemoveRaw(rawCmd []string) error {
	if err := b.redisClient.SRem(keys.GetRawCmdRedisKey(), rawCmd); err != nil {
		log.Logger.Warnf(err.Error())
		return err
	}

	return nil
}

func (b *AllowCmd) AddCommon(cmd []string) error {
	if err := b.redisClient.SAdd(keys.GetCommonCmdRedisKey(), cmd); err != nil {
		log.Logger.Warnf(err.Error())
		return err
	}

	return nil
}

func (b *AllowCmd) RemoveCommon(cmd []string) error {
	if err := b.redisClient.SRem(keys.GetCommonCmdRedisKey(), cmd); err != nil {
		log.Logger.Warnf(err.Error())
		return err
	}

	return nil
}
