package main

import (
	"encoding/hex"
	"fmt"
	"net"
	"regexp"
	"time"
)

// ExtractDeviceIDFromBatchRaw 从批量原始数据中提取设备ID（序列号形如 1513343-01 或 ABC123-01）
func ExtractDeviceIDFromBatchRaw(data string) string {
	m := regexp.MustCompile(`([A-Za-z0-9]+)-\d+`).FindStringSubmatch(data)
	if len(m) == 2 {
		Logger.Printf("从原始数据提取到设备ID: %s", m[1])
		return m[1]
	}
	Logger.Printf("未能从原始数据中提取设备ID，跳过转发")
	return ""
}

// ForwardRawData 将原始数据转发到设备配置的 TCP 目标
func ForwardRawData(deviceID string, raw string) error {
	if deviceID == "" {
		Logger.Printf("转发跳过：deviceID 为空")
		return nil
	}

	dev, err := GetDevice(deviceID)
	if err != nil {
		Logger.Printf("查询设备配置失败 deviceID=%s: %v", deviceID, err)
		return err
	}
	if dev == nil {
		Logger.Printf("未找到设备配置 deviceID=%s，跳过转发", deviceID)
		return nil
	}
	if !dev.ForwardEnable {
		Logger.Printf("设备未启用转发 deviceID=%s，跳过转发", deviceID)
		return nil
	}
	if !dev.Host.Valid || dev.Host.String == "" || !dev.Port.Valid || dev.Port.Int64 <= 0 {
		Logger.Printf("设备转发目标未配置完整 deviceID=%s host=%v port=%v，跳过转发", deviceID, dev.Host, dev.Port)
		return nil
	}

	address := fmt.Sprintf("%s:%d", dev.Host.String, dev.Port.Int64)
	Logger.Printf("准备连接转发目标 deviceID=%s addr=%s", deviceID, address)

	conn, err := net.DialTimeout("tcp", address, 5*time.Second)
	if err != nil {
		Logger.Printf("连接转发目标失败 deviceID=%s addr=%s: %v", deviceID, address, err)
		return err
	}
	defer conn.Close()
	Logger.Printf("连接转发目标成功 deviceID=%s addr=%s", deviceID, address)

	if dev.RegCodeHex.Valid && dev.RegCodeHex.String != "" {
		bytes, decErr := hex.DecodeString(dev.RegCodeHex.String)
		if decErr != nil {
			Logger.Printf("注册码HEX解码失败 deviceID=%s hex=%s: %v（将继续发送原始数据）", deviceID, dev.RegCodeHex.String, decErr)
		} else {
			Logger.Printf("发送注册码 deviceID=%s bytes=%d", deviceID, len(bytes))
			if _, werr := conn.Write(bytes); werr != nil {
				Logger.Printf("发送注册码失败 deviceID=%s: %v", deviceID, werr)
				return werr
			}
			time.Sleep(1 * time.Second)
		}
	} else {
		Logger.Printf("未配置注册码 deviceID=%s，直接发送原始数据", deviceID)
	}

	Logger.Printf("发送原始数据 deviceID=%s bytes=%d", deviceID, len(raw))
	_, err = conn.Write([]byte(raw))
	if err != nil {
		Logger.Printf("发送原始数据失败 deviceID=%s: %v", deviceID, err)
		return err
	}
	Logger.Printf("完成转发 deviceID=%s", deviceID)
	return nil
}