import Foundation
import SwiftUI
import UnityFramework
import PolySpatialRealityKit
import UIKit

// 1. 定义一个专门用于调试的 UIView
class MuseDebugView: UIView {
    
    // 回调函数，把数据传回给 SwiftUI 显示
    var onUpdate: ((String) -> Void)?
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        setupGesture()
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        setupGesture()
    }
    
    private func setupGesture() {
        let hover = UIHoverGestureRecognizer(target: self, action: #selector(handleHover(_:)))
        
        //允许 笔(pencil) 和 手(direct) 触发，方便测试
        hover.allowedTouchTypes = [
            NSNumber(value: UITouch.TouchType.pencil.rawValue), // Muse
            //NSNumber(value: UITouch.TouchType.direct.rawValue)  // 手指，可以删除
        ]
        
        self.addGestureRecognizer(hover)
        
        //设置一个明显红色
        self.backgroundColor = UIColor.red.withAlphaComponent(0.2)
        self.isUserInteractionEnabled = true
    }
    

        @objc func handleHover(_ gesture: UIHoverGestureRecognizer) {
            
            if gesture.state == .ended || gesture.state == .cancelled || gesture.state == .failed {
                        return
                    }
            //获取原始数据
            let location = gesture.location(in: self)
            let alt = Float(gesture.altitudeAngle)
            let azi = Float(gesture.azimuthAngle(in: self)) //别忘了(in: self)
            
            //let roll: Float = 0.0
            let rawRoll = gesture.rollAngle
            let roll = Float(rawRoll)
            
            let rollDeg = rawRoll * 180 / .pi
            
            //假设窗口大小约 30cm，UI 坐标约 300pt -> 比例约 0.001
            //注意：VisionOS 窗口原点在左上角，Y轴向下；Unity Y轴向上。
            let unityX = Float(location.x) * 0.001
            let unityY = Float(location.y) * 0.001
            
            //构建 Log 字符串 (方便调试)
            let log = "P:(\(String(format: "%.2f", unityX)), \(String(format: "%.2f", unityY))) A:\(String(format: "%.2f", alt)) R:\(String(format: "%.2f", rollDeg))"
            //print("MuseSending: \(log)") //保持 Xcode 端也能看到
            onUpdate?(log) // 更新 Swift UI 上的文字
            
            //发送给 Unity
            //最后一个参数 1.0 代表“正在追踪”
            SendMusePoseToUnity(
                x: unityX,
                y: unityY,
                z: 0,
                alt: alt,
                azi: azi,
                roll: roll,
                status: 1.0
            )
        }
}

//连接 UIView 和 SwiftUI
struct MuseDebugViewWrapper: UIViewRepresentable {
    @Binding var logText: String
    
    func makeUIView(context: Context) -> MuseDebugView {
        let view = MuseDebugView()
        //绑定回调
        view.onUpdate = { newLog in
            //必须在主线程更新 UI
            DispatchQueue.main.async {
                self.logText = newLog
            }
        }
        return view
    }
    
    func updateUIView(_ uiView: MuseDebugView, context: Context) {
        // 不需要反向更新
    }
}

@available(visionOS 1.0, *)
struct MuseTrackerView: View {
    @State private var logText: String = "Waiting for input...\nPlease try hovering and rotating your stylus here"
    
    //@State private var dragOffset: CGSize = .zero
    
    var body: some View {
        VStack(spacing: 15) {
            ZStack {
                //磨砂玻璃 + 阴影
                RoundedRectangle(cornerRadius: 30, style: .continuous)
                    .fill(.regularMaterial)
                    .shadow(color: .black.opacity(0.9), radius: 10, y: 5)
                
                //红色检测区
                MuseDebugViewWrapper(logText: $logText)
                    .opacity(1.0)
                    .clipShape(RoundedRectangle(cornerRadius: 30))
                
                //内容
                VStack(spacing: 12) {
                    Text("Stylus Input Area")
                        .font(.headline)
                        .padding(.top, 25)
                        .foregroundColor(.primary)
                    
                    Divider().padding(.horizontal)
                    
                    Text(logText)
                        .font(.system(size: 13, design: .monospaced))
                        .padding()
                        .background(Color.black.opacity(0.5))
                        .cornerRadius(12)
                        .padding(.horizontal)
                        .frame(maxHeight: .infinity)
                    
                    Spacer()
                }
            }
            .frame(width: 300, height: 300)
        }
    }
}