import CustomDump
import SwiftUI

/// A data type that describes the state of a confirmation dialog that can be shown to the user. The
/// `Action` generic is the type of actions that can be sent from tapping on a button in the sheet.
///
/// This type can be used in your application's state in order to control the presentation or
/// dismissal of dialogs. It is preferable to use this API instead of the default SwiftUI API for
/// dialogs because SwiftUI uses 2-way bindings in order to control the showing and dismissal of
/// dialogs, and that does not play nicely with the Composable Architecture. The library requires
/// that all state mutations happen by sending an action so that a reducer can handle that logic,
/// which greatly simplifies how data flows through your application, and gives you instant
/// testability on all parts of your application.
///
/// To use this API, you model all the dialog actions in your domain's action enum:
///
/// ```swift
/// enum AppAction: Equatable {
///   case cancelTapped
///   case deleteTapped
///   case favoriteTapped
///   case infoTapped
///
///   // Your other actions
/// }
/// ```
///
/// And you model the state for showing the dialog in your domain's state, and it can start off in a
/// `nil` state:
///
/// ```swift
/// struct AppState: Equatable {
///   var confirmationDialog: ConfirmationDialogState<AppAction>?
///
///   // Your other state
/// }
/// ```
///
/// Then, in the reducer you can construct an `ConfirmationDialogState` value to represent the
/// dialog you want to show to the user:
///
/// ```swift
/// let appReducer = Reducer<AppState, AppAction, AppEnvironment> { state, action, env in
///   switch action
///     case .cancelTapped:
///       state.confirmationDialog = nil
///       return .none
///
///     case .deleteTapped:
///       state.confirmationDialog = nil
///       // Do deletion logic...
///
///     case .favoriteTapped:
///       state.confirmationDialog = nil
///       // Do favoriting logic
///
///     case .infoTapped:
///       state.confirmationDialog = .init(
///         title: "What would you like to do?",
///         buttons: [
///           .default(TextState("Favorite"), action: .send(.favoriteTapped)),
///           .destructive(TextState("Delete"), action: .send(.deleteTapped)),
///           .cancel(),
///         ]
///       )
///     return .none
///   }
/// }
/// ```
///
/// And then, in your view you can use the `confirmationDialog(_:dismiss:)` method on `View` in
/// order to present the dialog in a way that works best with the Composable Architecture:
///
/// ```swift
/// Button("Info") { viewStore.send(.infoTapped) }
///   .confirmationDialog(
///     self.store.scope(state: \.confirmationDialog),
///     dismiss: .cancelTapped
///   )
/// ```
///
/// This makes your reducer in complete control of when the dialog is shown or dismissed, and makes
/// it so that any choice made in the dialog is automatically fed back into the reducer so that you
/// can handle its logic.
///
/// Even better, you can instantly write tests that your dialog behavior works as expected:
///
/// ```swift
/// let store = TestStore(
///   initialState: AppState(),
///   reducer: appReducer,
///   environment: .mock
/// )
///
/// store.send(.infoTapped) {
///   $0.confirmationDialog = .init(
///     title: "What would you like to do?",
///     buttons: [
///       .default(TextState("Favorite"), send: .favoriteTapped),
///       .destructive(TextState("Delete"), send: .deleteTapped),
///       .cancel(),
///     ]
///   )
/// }
/// store.send(.favoriteTapped) {
///   $0.confirmationDialog = nil
///   // Also verify that favoriting logic executed correctly
/// }
/// ```
///
@available(iOS 13, *)
@available(macOS 12, *)
@available(tvOS 13, *)
@available(watchOS 6, *)
public struct ConfirmationDialogState<Action> {
  public let id = UUID()
  public var buttons: [Button]
  public var message: TextState?
  public var title: TextState
  public var titleVisibility: Visibility

  @available(iOS 15, *)
  @available(macOS 12, *)
  @available(tvOS 15, *)
  @available(watchOS 8, *)
  public init(
    title: TextState,
    titleVisibility: Visibility,
    message: TextState? = nil,
    buttons: [Button] = []
  ) {
    self.buttons = buttons
    self.message = message
    self.title = title
    self.titleVisibility = titleVisibility
  }

  public init(
    title: TextState,
    message: TextState? = nil,
    buttons: [Button] = []
  ) {
    self.buttons = buttons
    self.message = message
    self.title = title
    self.titleVisibility = .automatic
  }

  public typealias Button = AlertState<Action>.Button

  public enum Visibility {
    case automatic
    case hidden
    case visible

    #if compiler(>=5.5)
      @available(iOS 15, macOS 12, tvOS 15, watchOS 8, *)
      var toSwiftUI: SwiftUI.Visibility {
        switch self {
        case .automatic:
          return .automatic
        case .hidden:
          return .hidden
        case .visible:
          return .visible
        }
      }
    #endif
  }
}

@available(iOS 13, *)
@available(macOS 12, *)
@available(tvOS 13, *)
@available(watchOS 6, *)
extension ConfirmationDialogState: CustomDumpReflectable {
  public var customDumpMirror: Mirror {
    Mirror(
      self,
      children: [
        "title": self.title,
        "message": self.message as Any,
        "buttons": self.buttons,
      ],
      displayStyle: .struct
    )
  }
}

@available(iOS 13, *)
@available(macOS 12, *)
@available(tvOS 13, *)
@available(watchOS 6, *)
extension ConfirmationDialogState: Equatable where Action: Equatable {
  public static func == (lhs: Self, rhs: Self) -> Bool {
    lhs.title == rhs.title
      && lhs.message == rhs.message
      && lhs.buttons == rhs.buttons
  }
}

@available(iOS 13, *)
@available(macOS 12, *)
@available(tvOS 13, *)
@available(watchOS 6, *)
extension ConfirmationDialogState: Hashable where Action: Hashable {
  public func hash(into hasher: inout Hasher) {
    hasher.combine(self.title)
    hasher.combine(self.message)
    hasher.combine(self.buttons)
  }
}

@available(iOS 13, *)
@available(macOS 12, *)
@available(tvOS 13, *)
@available(watchOS 6, *)
extension ConfirmationDialogState: Identifiable {}

extension View {
  /// Displays a dialog when the store's state becomes non-`nil`, and dismisses it when it becomes
  /// `nil`.
  ///
  /// - Parameters:
  ///   - store: A store that describes if the dialog is shown or dismissed.
  ///   - dismissal: An action to send when the dialog is dismissed through non-user actions, such
  ///     as when a dialog is automatically dismissed by the system. Use this action to `nil` out
  ///     the associated dialog state.
  @available(iOS 13, *)
  @available(macOS 12, *)
  @available(tvOS 13, *)
  @available(watchOS 6, *)
  @ViewBuilder public func confirmationDialog<Action>(
    _ store: Store<ConfirmationDialogState<Action>?, Action>,
    dismiss: Action
  ) -> some View {
    #if compiler(>=5.5)
      if #available(iOS 15, tvOS 15, watchOS 8, *) {
        self.modifier(
          NewConfirmationDialogModifier(
            viewStore: ViewStore(store, removeDuplicates: { $0?.id == $1?.id }),
            dismiss: dismiss
          )
        )
      } else {
        #if !os(macOS)
          self.modifier(
            OldConfirmationDialogModifier(
              viewStore: ViewStore(store, removeDuplicates: { $0?.id == $1?.id }),
              dismiss: dismiss
            )
          )
        #endif
      }
    #elseif !os(macOS)
      self.modifier(
        OldConfirmationDialogModifier(
          viewStore: ViewStore(store, removeDuplicates: { $0?.id == $1?.id }),
          dismiss: dismiss
        )
      )
    #endif
  }
}

#if compiler(>=5.5)
  // NB: Workaround for iOS 14 runtime crashes during iOS 15 availability checks.
  @available(iOS 15, macOS 12, tvOS 15, watchOS 8, *)
  private struct NewConfirmationDialogModifier<Action>: ViewModifier {
    @ObservedObject var viewStore: ViewStore<ConfirmationDialogState<Action>?, Action>
    let dismiss: Action

    func body(content: Content) -> some View {
      content.confirmationDialog(
        (viewStore.state?.title).map { Text($0) } ?? Text(""),
        isPresented: viewStore.binding(send: dismiss).isPresent(),
        titleVisibility: viewStore.state?.titleVisibility.toSwiftUI ?? .automatic,
        presenting: viewStore.state,
        actions: { $0.toSwiftUIActions(send: viewStore.send) },
        message: { $0.message.map { Text($0) } }
      )
    }
  }
#endif

@available(iOS 13, *)
@available(macOS 12, *)
@available(tvOS 13, *)
@available(watchOS 6, *)
private struct OldConfirmationDialogModifier<Action>: ViewModifier {
  @ObservedObject var viewStore: ViewStore<ConfirmationDialogState<Action>?, Action>
  let dismiss: Action

  func body(content: Content) -> some View {
    #if !os(macOS)
      return content.actionSheet(item: viewStore.binding(send: dismiss)) { state in
        state.toSwiftUIActionSheet(send: viewStore.send)
      }
    #else
      return EmptyView()
    #endif
  }
}

@available(iOS 13, *)
@available(macOS 12, *)
@available(tvOS 13, *)
@available(watchOS 6, *)
extension ConfirmationDialogState {
  #if compiler(>=5.5)
    @available(iOS 15, macOS 12, tvOS 15, watchOS 8, *)
    @ViewBuilder
    fileprivate func toSwiftUIActions(send: @escaping (Action) -> Void) -> some View {
      ForEach(self.buttons.indices, id: \.self) {
        self.buttons[$0].toSwiftUIButton(send: send)
      }
    }
  #endif

  @available(macOS, unavailable)
  fileprivate func toSwiftUIActionSheet(send: @escaping (Action) -> Void) -> SwiftUI.ActionSheet {
    SwiftUI.ActionSheet(
      title: Text(self.title),
      message: self.message.map { Text($0) },
      buttons: self.buttons.map {
        $0.toSwiftUIAlertButton(send: send)
      }
    )
  }
}