public protocol PreviewStoryViewProtocol: AnyObject {
    var endFrame: CGRect { get }
    var startFrame: CGRect { get }
}
public class PreviewStoryView: UIView, PreviewStoryViewProtocol {
    public var startFrame: CGRect {
        return convert(bounds, to: nil)
    }
    public var endFrame: CGRect {
        return convert(bounds, to: nil)
    }
}

class StoriesNavigationController: UINavigationController {
    // MARK: - Private properties
    private var previewFrame: PreviewStoryViewProtocol?
    // MARK: - Setup
    func setup(viewControllers: [UIViewController], previewFrame: PreviewStoryViewProtocol?) {
        self.previewFrame = previewFrame
        self.viewControllers = viewControllers
    }
    // MARK: - Lifecycle
    convenience init() {
        self.init(nibName: nil, bundle: nil)
        setupUI()
    }
}
extension StoriesNavigationController {
    private func setupUI() {
        setNavigationBarHidden(true, animated: false)
        modalPresentationStyle = .custom
    }
}

extension StoriesNavigationController: UIViewControllerTransitioningDelegate {
    public func animationController(
        forPresented presented: UIViewController,
        presenting: UIViewController,
        source: UIViewController) -> UIViewControllerAnimatedTransitioning?
    {
        guard let startFrame = previewFrame?.startFrame else { return nil }
        return StoriesNavigationPresentAnimator(startFrame: startFrame)
    }
    public func animationController(
        forDismissed dismissed: UIViewController) -> UIViewControllerAnimatedTransitioning?
    {
        guard let endFrame = previewFrame?.endFrame else { return nil }
        return StoriesNavigationDismissAnimator(endFrame: endFrame)
    }
}

class StoriesNavigationAnimator: NSObject, UIViewControllerAnimatedTransitioning {
    private enum Spec {
        static let animationDuration: TimeInterval = 0.3
    }
    private let startFrame: CGRect
    init(startFrame: CGRect) {
        self.startFrame = startFrame
    }
    func transitionDuration(using transitionContext: UIViewControllerContextTransitioning?) -> TimeInterval {
        return Spec.animationDuration
    }
    func animateTransition(using transitionContext: UIViewControllerContextTransitioning) {
        // 1
        guard let toViewController = transitionContext.viewController(forKey: .to),
              let snapshot = toViewController.view.snapshotView(afterScreenUpdates: true)
        else {
            return
        }
        // 2
        let containerView = transitionContext.containerView
        // 3
        containerView.addSubview(toViewController.view)
        toViewController.view.isHidden = true
        // 4
        snapshot.frame = startFrame
        snapshot.alpha = 0.0
        containerView.addSubview(snapshot)
        UIView.animate(withDuration: Spec.animationDuration, animations: {
            // 5
            snapshot.frame = (transitionContext.finalFrame(for: toViewController))
            snapshot.alpha = 1.0
        }, completion: { _ in
            // 6
            toViewController.view.isHidden = false
            snapshot.removeFromSuperview()
            // 7
            if transitionContext.transitionWasCancelled {
                toViewController.view.removeFromSuperview()
            }
            // 8
            transitionContext.completeTransition(!transitionContext.transitionWasCancelled)
        })
    }
}

class StoriesNavigationDismissAnimator: NSObject, UIViewControllerAnimatedTransitioning {
    private enum Spec {
        static let animationDuration: TimeInterval = 0.3
    }
    private let endFrame: CGRect
    init(endFrame: CGRect) {
        self.endFrame = endFrame
    }
    func transitionDuration(using transitionContext: UIViewControllerContextTransitioning?) -> TimeInterval {
        return Spec.animationDuration
    }
    func animateTransition(using transitionContext: UIViewControllerContextTransitioning) {
        guard let fromViewController = transitionContext.viewController(forKey: .from),
              let snapshot = fromViewController.view.snapshotView(afterScreenUpdates: true)
        else {
            return
        }
        let containerView = transitionContext.containerView
        containerView.addSubview(snapshot)
        fromViewController.view.isHidden = true
        UIView.animate(withDuration: Spec.animationDuration, delay: 0, options: .curveEaseOut, animations: {
            snapshot.frame = self.endFrame
            snapshot.alpha = 0
        }, completion: { _ in
            fromViewController.view.isHidden = false
            snapshot.removeFromSuperview()
            transitionContext.completeTransition(!transitionContext.transitionWasCancelled)
        })
    }
}

class StoryBaseViewController: UIViewController {
    // MARK: - Constants
    private enum Spec {
        enum CloseImage {
            static let size: CGSize = CGSize(width: 40, height: 40)
            static var original: CGPoint = CGPoint(x: 24, y: 50)
        }
    }
    // MARK: - UI components
    private lazy var closeButton: UIButton = {
        let button = UIButton(type: .custom)
        button.setImage(#imageLiteral(resourceName: "closeImage"), for: .normal)
        button.addTarget(self, action: #selector(closeButtonAction(sender:)), for: .touchUpInside)
        button.frame = CGRect(origin: Spec.CloseImage.original, size: Spec.CloseImage.size)
        return button
    }()
    // MARK: - Lifecycle
    public override func loadView() {
        super.loadView()
        view.addSubview(closeButton)
    }
    @objc
    private func closeButtonAction(sender: UIButton!) {
        dismiss(animated: true, completion: nil)
    }
}

class ViewController: UIViewController {
    // MARK: - UI components
    private lazy var previewView: PreviewStoryView = {
        let preview = PreviewStoryView()
        preview.frame.size = CGSize(width: 200, height: 200)
        preview.backgroundColor = .black
        preview.layer.cornerRadius = 10
        preview.center = view.center
        return preview
    }()
    private lazy var showButton: UIButton = {
        let button = UIButton()
        button.setTitle("Show", for: .normal)
        button.addTarget(self, action: #selector(handleButtonAction), for: .touchUpInside)
        button.frame = CGRect(origin: CGPoint(x: 0, y: 0), size: CGSize(width: 200, height: 200))
        return button
    }()
    // MARK: - Lifecycle
    override func viewDidLoad() {
        super.viewDidLoad()
        setupUI()
    }
}
extension ViewController {
    private func setupUI() {
        view.backgroundColor = .darkGray
        view.addSubview(previewView)
        previewView.addSubview(showButton)
        let panGesture = UIPanGestureRecognizer(target: self, action: #selector(handlePanGesture(gesture:)))
        previewView.addGestureRecognizer(panGesture)
    }
}
extension ViewController {
    @objc
    func handleButtonAction(sender: UIButton!) {
        var storyViewControllers: [UIViewController] {
            let vc1 = StoryBaseViewController()
            vc1.view.backgroundColor = .red
            let vc2 = StoryBaseViewController()
            vc2.view.backgroundColor = .green
            let vc3 = StoryBaseViewController()
            vc3.view.backgroundColor = .blue
            return [vc1, vc2, vc3]
        }
        let storiesVC = StoriesNavigationController()
        storiesVC.setup(viewControllers: storyViewControllers, previewFrame: previewView)
        present(storiesVC, animated: true, completion: nil)
    }
    @objc
    func handlePanGesture(gesture: UIPanGestureRecognizer) {
        let stateIsValidate = gesture.state == .began || gesture.state == .changed
        if let gestureView = gesture.view, stateIsValidate {
            let translation = gesture.translation(in: self.view)
            let newXPosition = gestureView.center.x + translation.x
            let newYPosition = gestureView.center.y + translation.y
            gestureView.center = CGPoint(x: newXPosition, y: newYPosition)
            gesture.setTranslation(.zero, in: self.view)
        }
    }
}