package main

import (
	"crypto/rand"
	"crypto/rsa"
	"crypto/x509"
	"encoding/pem"
	"fmt"
	"gitea.pena/PenaSide/common/encrypt"
	"gitea.pena/PenaSide/customer/internal/models"
	"github.com/stretchr/testify/assert"
	"os"
	"testing"
)

func TestValidateTG(t *testing.T) {
	err := validateTG("7127966184:AAG1steOCH4wDvHRe9QcsXJPS4dWRyRYsqg", -1002177203276, -1002177203276)
	assert.NoError(t, err)
}

func TestValidateMail(t *testing.T) {
	err := validateMail(models.MailClientCfg{
		ApiUrl:      "https://api.smtp.bz/v1/smtp/send",
		Sender:      "noreply@mailing.pena.digital",
		ApiKey:      "P0YsjUB137upXrr1NiJefHmXVKW1hmBWlpev",
		MailAddress: "sells@pena.digital",
	})
	assert.NoError(t, err)
}

func TestValidateURLs(t *testing.T) {
	urls := []string{
		"http://10.8.0.6:59300",
		"http://10.8.0.6:59303",
		"http://10.8.0.6:3131",
		"10.8.0.6:9001",
		"10.8.0.6:9085",
		"http://10.8.0.6:7036",
		"http://10.6.0.17",
		"https://api.smtp.bz/v1/smtp/send",
		"10.8.0.6:59665",
		"https://sadmin.pena",
		"10.8.0.15:7113",
	}
	err := validateURLs(urls)
	assert.NoError(t, err)
}

func TestValidateKafka(t *testing.T) {
	err := validateKafka([]string{"localhost:9092"}, "test-topic")
	assert.NoError(t, err)
}

func TestValidateEncryptKeys(t *testing.T) {
	priveKey, pubKey, err := generateRSAKeys(2048)
	assert.NoError(t, err)
	err = validateEncryptKeys(&encrypt.Encrypt{
		PubKey:  pubKey,
		PrivKey: priveKey,
	})
	assert.NoError(t, err)
}

func generateRSAKeys(bitSize int) (privateKey string, publicKey string, err error) {
	privKey, err := rsa.GenerateKey(rand.Reader, bitSize)
	if err != nil {
		return "", "", err
	}

	privKeyPEM := pem.EncodeToMemory(&pem.Block{
		Type:  "RSA PRIVATE KEY",
		Bytes: x509.MarshalPKCS1PrivateKey(privKey),
	})

	pubKeyBytes, err := x509.MarshalPKIXPublicKey(&privKey.PublicKey)
	if err != nil {
		return "", "", err
	}
	pubKeyPEM := pem.EncodeToMemory(&pem.Block{
		Type:  "RSA PUBLIC KEY",
		Bytes: pubKeyBytes,
	})

	return string(privKeyPEM), string(pubKeyPEM), nil
}

func TestLoadConfig(t *testing.T) {
	envVars := map[string]string{
		"JWT_ISSUER":                      "pena-auth-service",
		"JWT_AUDIENCE":                    "pena",
		"JWT_PUBLIC_KEY":                  "-----BEGIN PUBLIC KEY-----\nMIGeMA0GCSqGSIb3DQEBAQUAA4GMADCBiAKBgHgnvr7O2tiApjJfid1orFnIGm69\n80fZp+Lpbjo+NC/0whMFga2Biw5b1G2Q/B2u0tpO1Fs/E8z7Lv1nYfr5jx2S8x6B\ndA4TS2kB9Kf0wn0+7wSlyikHoKhbtzwXHZl17GsyEi6wHnsqNBSauyIWhpha8i+Y\n+3GyaOY536H47qyXAgMBAAE=\n-----END PUBLIC KEY-----",
		"CLIENT_HTTP_HOST":                "0.0.0.0",
		"CLIENT_HTTP_PORT":                "8080",
		"ADMIN_HTTP_HOST":                 "0.0.0.0",
		"ADMIN_HTTP_PORT":                 "8081",
		"GRPC_HOST":                       "0.0.0.0",
		"GRPC_PORT":                       "9001",
		"GRPC_DOMEN":                      "customer-service:9000",
		"MONGO_URL":                       "mongodb://test:test@localhost:27020/",
		"MONGO_DB_NAME":                   "admin",
		"KAFKA_BROKERS":                   "localhost:9092",
		"KAFKA_TOPIC_TARIFF":              "test-topic",
		"AUTH_MICROSERVICE_URL":           "http://localhost:8000/user",
		"HUBADMIN_MICROSERVICE_URL":       "http://localhost:8001/tariff",
		"CURRENCY_MICROSERVICE_URL":       "http://cbrfworker-service:8000/change",
		"DISCOUNT_MICROSERVICE_GRPC_HOST": "localhost:9040",
		"PAYMENT_MICROSERVICE_GRPC_HOST":  "treasurer-service:9085",
		"VERIFICATION_MICROSERVICE_URL":   "http://10.8.0.8:7035/verification",
		"TEMPLATEGEN_MICROSERVICE_URL":    "10.6.0.17",
		"CODEWORD_MICROSERVICE_GRPC_HOST": "http://localhost:8000/user",
		"TRASH_LOG_HOST":                  "localhost:7113",
		"API_URL":                         "https://api.smtp.bz/v1/smtp/send",
		"MAIL_SENDER":                     "noreply@mailing.pena.digital",
		"MAIL_API_KEY":                    "P0YsjUB137upXrr1NiJefHmXVKW1hmBWlpev",
		"MAIL_ADDRESS":                    "mail@mail.com",
		"NOTIFICATION_BOT_TOKEN":          "7127966184:AAG1steOCH4wDvHRe9QcsXJPS4dWRyRYsqg",
		"NOTIFICATION_CHANNEL":            "-1002177203276",
		"NOTIFICATION_RS_PAY_CHANNEL":     "-1002177203276",
		"ADMIN_FRONT_URL":                 "https://admin-front.ru",
		"ENCRYPT_PUBLIC_KEY":              "1",
		"ENCRYPT_PRIVATE_KEY":             "2",
	}

	for key, value := range envVars {
		os.Setenv(key, value)
	}

	cfg, err := loadConfig()
	assert.Nil(t, err)

	fmt.Println(cfg)
}