package mqtt

import (
	"bytes"
	"crypto/rand"
	"crypto/rsa"
	"crypto/x509"
	"encoding/base64"
	"encoding/json"
	"encoding/pem"
	"fmt"
	mqtt "github.com/eclipse/paho.mqtt.golang"
	"github.com/google/uuid"
	"go_mqtt/models"
	"io/ioutil"
	"log"
	"os"
	"strings"
	"time"
)

// RSA_Decrypts RSA解密支持分段解密
func RSA_Decrypts(cipherText []byte, path string) []byte {
	//打开文件
	var bytesDecrypt []byte
	//file, err := os.Open(path)
	//if err != nil {
	//	fmt.Println(err)
	//}
	//
	////获取文件内容
	//info, _ := file.Stat()
	//buf := make([]byte, info.Size())
	//file.Read(buf)

	// 读取私钥文件
	keyData, err := ioutil.ReadFile(path)
	if err != nil {
		return nil
	}

	//pem解码
	block, _ := pem.Decode(keyData)
	//X509解码
	privateKey, err := x509.ParsePKCS8PrivateKey(block.Bytes)
	if err != nil {
		fmt.Println(err.Error())
		os.Exit(0)
	}
	p := privateKey.(*rsa.PrivateKey)
	keySize := p.Size()
	srcSize := len(cipherText)
	log.Println("密钥长度", keySize, "密文长度", srcSize)
	var offSet = 0
	var buffer = bytes.Buffer{}
	for offSet < srcSize {
		endIndex := offSet + keySize
		if endIndex > srcSize {
			endIndex = srcSize
		}
		bytesOnce, err := rsa.DecryptPKCS1v15(rand.Reader, p, cipherText[offSet:endIndex])
		if err != nil {
			return nil
		}
		buffer.Write(bytesOnce)
		offSet = endIndex
	}
	bytesDecrypt = buffer.Bytes()
	return bytesDecrypt
}

// RSA_DecryptsOne RSA解密支持分段解密
func RSA_DecryptsOne(cipherText []byte, privateKeyStr string) []byte {
	//打开文件
	var bytesDecrypt []byte
	// 读取私钥文件转成字节数组
	keyData := []byte(privateKeyStr)

	//pem解码
	block, _ := pem.Decode(keyData)
	//X509解码
	privateKey, err := x509.ParsePKCS8PrivateKey(block.Bytes)
	if err != nil {
		fmt.Println(err.Error())
		os.Exit(0)
	}
	p := privateKey.(*rsa.PrivateKey)
	keySize := p.Size()
	srcSize := len(cipherText)
	log.Println("密钥长度", keySize, "密文长度", srcSize)
	var offSet = 0
	var buffer = bytes.Buffer{}
	for offSet < srcSize {
		endIndex := offSet + keySize
		if endIndex > srcSize {
			endIndex = srcSize
		}
		bytesOnce, err := rsa.DecryptPKCS1v15(rand.Reader, p, cipherText[offSet:endIndex])
		if err != nil {
			return nil
		}
		buffer.Write(bytesOnce)
		offSet = endIndex
	}
	bytesDecrypt = buffer.Bytes()
	return bytesDecrypt
}

// RsaEncryptBlock 公钥加密-分段
func RsaEncryptBlock(src []byte, path string) (bytesEncrypt []byte, err error) {
	//打开文件
	//file, err := os.Open(path)
	//if err != nil {
	//	return
	//}
	//
	////读取文件的内容
	//info, _ := file.Stat()
	//buf := make([]byte, info.Size())
	//file.Read(buf)

	// 读取公钥文件
	keyData, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, err
	}

	//pem解码
	block, _ := pem.Decode(keyData)
	//x509解码
	publicKeyInterface, err := x509.ParsePKIXPublicKey(block.Bytes)
	if err != nil {
		fmt.Println(err)
	}
	//类型断言
	publicKey := publicKeyInterface.(*rsa.PublicKey)
	keySize, srcSize := publicKey.Size(), len(src)
	log.Println("密钥长度", keySize, "明文长度", srcSize)
	offSet, once := 0, keySize-11
	buffer := bytes.Buffer{}
	for offSet < srcSize {
		endIndex := offSet + once
		if endIndex > srcSize {
			endIndex = srcSize
		}
		// 加密一部分
		bytesOnce, err := rsa.EncryptPKCS1v15(rand.Reader, publicKey, src[offSet:endIndex])
		if err != nil {
			return nil, err
		}
		buffer.Write(bytesOnce)
		offSet = endIndex
	}
	bytesEncrypt = buffer.Bytes()
	return
}

func timing(client mqtt.Client) {
	//定时器，10秒钟执行一次
	ticker := time.NewTicker(10 * time.Second)
	for {
		time := <-ticker.C
		fmt.Println("定时器====>", time.String())
		if !client.IsConnectionOpen() {
			client.Connect()
		} else {
			fmt.Println("client has connect")
		}
	}
}

var messagePubHandler mqtt.MessageHandler = func(client mqtt.Client, msg mqtt.Message) {
	fmt.Printf("Received message: %s from topic: %s\n", msg.Payload(), msg.Topic())

	message := fmt.Sprintf("%s", msg.Payload())
	topic := fmt.Sprintf("%s", msg.Topic())
	fmt.Println(message)
	fmt.Println(topic)

	//privatePath := "/Users/edao/GolandProjects/go_mqtt/privateKey.pem"
	privateKeyPEM := `-----BEGIN PRIVATE KEY-----
MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQCYwvInDUICXbmWcOAR5hm86mz7WlKL3dDf12MSeIY2jm5QpUSSoMhrJOWLbOt5fXQepPaNO0M30A+3C4SFZrM/9WA5ehazO1u1m1LAkYzoSDRH3MMJsrJC3lCGDeOROteu+safuP9k/npU1YQu/+Ll2xEJNxyvUx4jLGM4LamiMI6ytM3gdnOAGP4YRQo9Etwo6I986yg/seCQC5rza9M4iBamoin7U8h9yOMKrM6xK/k9CcY/vn5+Uhe3Pvk4qj/2Ff3OXkkc1wfdILqdLwLOKL0Tb3ZciwG0p1CKO80yf5hyYoWjqZk5Rcd07nTo2gqTFGfLl9sqI9/+ipMDtnHlAgMBAAECggEATJd5yCC6lusdMRO5FOBUyUaUi9X2i1AU+RZKAynQySvSnbavUgExW58tRCHBUrGW9gJp59ft1N8J8hHhSO18NDY4H7laBlVdnwmYjRqtFo2VQO6sD4G8JRDION5f2iIxn/b2fYDI9H8vILfJRbNgtTSILyGlzTYUZzhLKxCh+8IsN96Nic8wa5COd1vZZmdhf2y8TG8clFWmozaScNSAATx7y+8XLVWjjWiIRZ6xQvx0uQPUParc9KihXXTKR2pA22yPIdz+U4MGD4kC0eczlcFKZ/dYv9e7OIGgnJfT0idSCu7nYb1pxJ1LxD9fS6IScNTF5dSe0OIL98e+XdyoAQKBgQDRep+5cW4iAKrEMH+djmcXAkoMiYtNVtnu0efLE8dP6vjYytQi368X9SdcASbfrQ31eEZmr/xQnlUF8oyHGkI38YS8dpAHzQcrkP3BljbbzB/3gJZaUdghGsDrK0xAJIzzmFKQpeKnGtr23vxUgaGrNsCYvQ0eQ7+5056KXS4r5QKBgQC6r8xtRSaje6L4WIydjWvYywsmRO0Of0aJLMDA/Wt2MWhHfh7ba9oI1cKGN80ap7xB2a9lQLgpv+C53wNtE5SpvjxsikAj96nUMMhGy9ojXrUith6HQhiINETz6Shnznd+AyrXP6KI/RpfA5nkDB5nrJxODwtYLP467IL7Cv7OAQKBgQCl4KxKdH/5fP28jYsAgJsxpSZt9xzQCU5Zxu396ZOSvUaApVyGoQpNtluMh3z48lhzYOKevgzW6gn5w69z7F8zXZT2iAxVoQ1kelP2z7RxKJrHqpNkwhqbXEwX7RlcUZUr8BqxYCqymJl7k+fMIzqaEalBSbLxnEReKi0I8/Bz4QKBgHK4b0ZCtVDHPEmimJ6E9l4dv/c/afF7swu+zaCK2ouiJvOwBCRQbYb6XPR/u/GCXASXUdpF4CX/vIhcDE3uN2/r8FO+zVWM7vbvF1OyF5WesG7pPW9e5ZZlkG3WvLa1wOZV6fCmMSo/ZwI2Q05JSDHrd43cXttLotrw1jiQ9C4BAoGBAKi4SOoOVQ5J5HQCDkBwPbG1AOLHFinzfoDl26GF/8Hy7fmmd1JiRTFldQp/A9VTAABz3sVYmMB92HSIaJhuDMoYJNI2Cf/cZifsv7vUL8cbLn+lPsKsebiuB0m0g4P2qLwLfegfNGEgA7lA5HIz3SELqbdp3iuqJeQl1fsJqD74
-----END PRIVATE KEY-----`

	if topic == "app_push" {
		fmt.Println("珠海电厂APP收到mqtt消息")
		decodeByte, err := base64.StdEncoding.DecodeString(message)
		if err != nil {
			fmt.Println(err.Error())
		}
		//decrptCode := RSA_Decrypts(decodeByte, privatePath)

		decrptCode := RSA_DecryptsOne(decodeByte, privateKeyPEM)

		fmt.Println("解密后的字符串：")
		fmt.Println(string(decrptCode))
		fmt.Println("-----user--------")
		var user models.User
		json.Unmarshal(decrptCode, &user)
		fmt.Println(user)

		if models.GetUser(&user) {
			models.UpdateUser(&user)
		} else {
			models.SaveUser(&user)
		}

	} else if topic == "app_push_dyw" {
		fmt.Println("大亚湾电厂APP收到mqtt消息")
		decodeByte, err := base64.StdEncoding.DecodeString(message)
		if err != nil {
			fmt.Println(err.Error())
		}
		decrptCode := RSA_DecryptsOne(decodeByte, privateKeyPEM)
		fmt.Println("解密后的字符串：")
		fmt.Println(string(decrptCode))
		fmt.Println("-----userdyw--------")
		var user models.Userdyw
		json.Unmarshal(decrptCode, &user)
		fmt.Println(user)

		if models.GetUserdyw(&user) {
			models.UpdateUserdyw(&user)
		} else {
			models.SaveUserdyw(&user)
		}

	} else if topic == "app_push_yf" {
		fmt.Println("云浮电厂APP收到mqtt消息")
		decodeByte, err := base64.StdEncoding.DecodeString(message)
		if err != nil {
			fmt.Println(err.Error())
		}
		decrptCode := RSA_DecryptsOne(decodeByte, privateKeyPEM)
		fmt.Println("解密后的字符串：")
		fmt.Println(string(decrptCode))
		fmt.Println("-----userdyw--------")
		var user models.Useryf
		json.Unmarshal(decrptCode, &user)
		fmt.Println(user)

		if models.GetUseryf(&user) {
			models.UpdateUseryf(&user)
		} else {
			models.SaveUseryf(&user)
		}

	} else {
		result := strings.Split(message, " ")
		fmt.Println(result[0])
		fmt.Println(result[1])

		now := time.Now()
		fmt.Println("---------now--------")
		fmt.Println(now)
		temperature := new(models.Temperature)
		temperature.CreateDate = now
		temperature.DataDate = now.Format("2006-01-02")
		temperature.DataHour = now.Format("2006-01-02 15")
		temperature.DataMinute = now.Format("2006-01-02 15:04")
		temperature.Humidity = result[0]
		temperature.Temperature = result[1]
		temperature.Topic = topic
		if topic == "WifiSHT/7C87CE9CA4E6/SHT20" {
			temperature.LocationDesc = "广东省珠海市高新区唐家湾镇东岸村水风三街28号501"
		}
		if topic == "WifiSHT/7C87CE9F5CBF/SHT20" {
			temperature.LocationDesc = "广东省珠海市金湾区三灶镇百川路1号1栋1单元1508房"
		}
		if topic == "WifiSHT/4CEBD686B6AA/SHT20" {
			temperature.LocationDesc = "广西壮族自治区崇左市天等县天等镇荣华村弄在屯113号"
		}
		fmt.Println("测温传感器地点:", temperature.LocationDesc)
		models.SaveTemperature(temperature)
	}

	//uuid := uuid.New()
	//models.SaveProduct()
	//models.SaveUser(uuid.String())
	//models.SaveOrder()

	//var users []models.User
	//var orders []models.Order
	//var user models.User
	//mydb.DB.Where("name = ?", "wenfei").First(&user)
	//fmt.Println(user)

	// 查询多条记录
	//mydb.DB.Find(&users) // 查询所有产品信息
	//mydb.DB.Order("id desc").Limit(2).Offset(0).Find(&users) // 按价格降序排序，取前 10 条记录
	//fmt.Println("users:", users)

	// 查询多条记录
	//mydb.DB.Find(&users) // 查询所有产品信息
	//mydb.DB.Where("id > ? and id < ?", 60,65).Find(&orders)
	//fmt.Println("orders:", orders)

	// 原生 SQL 查询
	//var products []models.Product
	//mydb.DB.Raw("SELECT * FROM product WHERE price > ?", 1000).Scan(&products) // 查询价格大于 1000 的产品信息
	//fmt.Println("Products:", products)

}

var connectHandler mqtt.OnConnectHandler = func(client mqtt.Client) {
	fmt.Println("Connected")
	subTemperature(client)
	subAppPush(client)
	subAppPushDyw(client)
}

var connectLostHandler mqtt.ConnectionLostHandler = func(client mqtt.Client, err error) {
	fmt.Printf("Connect lost: %v", err)
}

func sub(client mqtt.Client) {
	topic := "gomqtt01"
	token := client.Subscribe(topic, 1, nil)
	token.Wait()
	fmt.Printf("Subscribed to topic: %s", topic)
}

func subTemperature(client mqtt.Client) {
	topic := "WifiSHT/+/SHT20"
	token := client.Subscribe(topic, 2, nil)
	token.Wait()
	fmt.Printf("Subscribed to topic: %s", topic)
}

func subAppPush(client mqtt.Client) {
	topic := "app_push"
	token := client.Subscribe(topic, 2, nil)
	token.Wait()
	fmt.Printf("Subscribed to topic: %s", topic)
}
func subAppPushDyw(client mqtt.Client) {
	topic := "app_push_dyw"
	token := client.Subscribe(topic, 2, nil)
	token.Wait()
	fmt.Printf("Subscribed to topic: %s", topic)
}

func publish(client mqtt.Client) {
	num := 999999000000000000
	for i := 0; i < num; i++ {
		text := fmt.Sprintf("Message %d", i)
		token := client.Publish("gomqtt01", 0, false, text)
		token.Wait()
		fmt.Println(i)
		time.Sleep(6 * time.Second)
	}
}

func main() {

	uuid := uuid.New()
	fmt.Println("Generated UUID:", uuid)

	var broker = "47.242.184.139"
	var port = 1883
	opts := mqtt.NewClientOptions()
	opts.AddBroker(fmt.Sprintf("tcp://%s:%d", broker, port))
	opts.SetClientID(uuid.String())
	opts.SetUsername("admin")
	opts.SetPassword("publish452131wW452131wW$")
	opts.SetDefaultPublishHandler(messagePubHandler)
	opts.OnConnect = connectHandler
	opts.OnConnectionLost = connectLostHandler
	client := mqtt.NewClient(opts)
	if token := client.Connect(); token.Wait() && token.Error() != nil {
		fmt.Println(token.Error())
	}

	//client.IsConnectionOpen()

	go timing(client)

	//sub(client)
	subTemperature(client)
	subAppPush(client)
	subAppPushDyw(client)
	//publish(client)

	for true {
		time.Sleep(5 * time.Second)
	}

	//client.Disconnect(1000)

}