import UIKit

class DraggableImageView: UIImageView {
    override init(frame: CGRect) {
        super.init(frame: frame)
        setup()
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        setup()
    }
    
    private func setup() {
        isUserInteractionEnabled = true
        let panGesture = UIPanGestureRecognizer(target: self, action: #selector(handlePan(_:)))
        addGestureRecognizer(panGesture)
    }
    
    @objc private func handlePan(_ recognizer: UIPanGestureRecognizer) {
        let translation = recognizer.translation(in: self.superview)
        self.center = CGPoint(x: self.center.x + translation.x, 
                             y: self.center.y + translation.y)
        recognizer.setTranslation(.zero, in: self.superview)
    }
}

class DraggableImageView: UIImageView {
    private var initialLocation = CGPoint.zero
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        isUserInteractionEnabled = true
    }
    
    required init?(coder: NSCoder) {
        super.init(coder: coder)
        isUserInteractionEnabled = true
    }
    
    override func touchesBegan(_ touches: Set<UITouch>, with event: UIEvent?) {
        super.touchesBegan(touches, with: event)
        initialLocation = touches.first?.location(in: self.superview) ?? .zero
    }
    
    override func touchesMoved(_ touches: Set<UITouch>, with event: UIEvent?) {
        super.touchesMoved(touches, with: event)
        guard let touch = touches.first else { return }
        let location = touch.location(in: self.superview)
        
        UIView.animate(withDuration: 0.1) {
            self.center = CGPoint(
                x: self.center.x + (location.x - self.initialLocation.x),
                y: self.center.y + (location.y - self.initialLocation.y)
            )
        }
        
        initialLocation = location
    }
}

@objc private func handlePan(_ recognizer: UIPanGestureRecognizer) {
    guard let superview = self.superview else { return }
    let translation = recognizer.translation(in: superview)
    
    var newCenter = CGPoint(x: center.x + translation.x, y: center.y + translation.y)
    
    // 限制在父视图范围内
    let halfWidth = bounds.width / 2
    let halfHeight = bounds.height / 2
    
    newCenter.x = max(halfWidth, min(newCenter.x, superview.bounds.width - halfWidth))
    newCenter.y = max(halfHeight, min(newCenter.y, superview.bounds.height - halfHeight))
    
    center = newCenter
    recognizer.setTranslation(.zero, in: superview)
}

@objc private func handlePan(_ recognizer: UIPanGestureRecognizer) {
    guard let superview = self.superview else { return }
    let translation = recognizer.translation(in: superview)
    
    let newCenter = CGPoint(x: center.x + translation.x, y: center.y + translation.y)
    
    // 计算弹性效果
    let damping: CGFloat = 0.5
    let velocity = recognizer.velocity(in: superview)
    
    UIView.animate(withDuration: 0.5, 
                   delay: 0, 
                   usingSpringWithDamping: damping, 
                   initialSpringVelocity: abs(velocity.x + velocity.y) / 1000, 
                   options: [.allowUserInteraction, .beginFromCurrentState], 
                   animations: {
        self.center = newCenter
    })
    
    recognizer.setTranslation(.zero, in: superview)
}