Impressia/Vernissage/Widgets/ImageRowItemAsync.swift

//
//  https://mczachurski.dev
//  Copyright © 2023 Marcin Czachurski and the repository contributors.
//  Licensed under the Apache License 2.0.
//

import SwiftUI
import Nuke
import NukeUI
import PixelfedKit
import ClientKit
import ServicesKit
import EnvironmentKit
import WidgetsKit

@MainActor
struct ImageRowItemAsync: View {
    @Environment(ApplicationState.self) var applicationState
    @Environment(Client.self) var client
    @Environment(RouterPath.self) var routerPath

    private var statusViewModel: StatusModel
    private var attachment: AttachmentModel
    private let showAvatar: Bool
    private let imageFromCache: Bool

    @Binding private var containerWidth: Double
    @Binding private var showSpoilerText: Bool
    @Binding private var clipToRectangle: Bool

    @State private var showThumbImage = false
    @State private var opacity = 1.0
    @State private var isFavourited = false

    private let onImageDownloaded: (Double, Double) -> Void

    init(statusViewModel: StatusModel,
         attachment: AttachmentModel,
         withAvatar showAvatar: Bool = true,
         containerWidth: Binding<Double>,
         clipToRectangle: Binding<Bool> = Binding.constant(false),
         showSpoilerText: Binding<Bool> = Binding.constant(true),
         onImageDownloaded: @escaping (_: Double, _: Double) -> Void) {
        self.showAvatar = showAvatar
        self.statusViewModel = statusViewModel
        self.attachment = attachment
        self.onImageDownloaded = onImageDownloaded

        self._containerWidth = containerWidth
        self._showSpoilerText = showSpoilerText
        self._clipToRectangle = clipToRectangle

        self.imageFromCache = ImagePipeline.shared.cache.containsCachedImage(for: ImageRequest(url: attachment.url))
    }

    var body: some View {
        LazyImage(url: attachment.url) { state in
            if let image = state.image {
                if self.statusViewModel.sensitive && !self.applicationState.showSensitive {
                    ZStack {
                        ContentWarning(spoilerText: self.showSpoilerText ? self.statusViewModel.spoilerText : nil) {
                            self.imageContainerView(image: image)
                                .imageContextMenu(statusModel: self.statusViewModel,
                                                  attachmentModel: self.attachment,
                                                  uiImage: state.imageResponse?.image)
                        } blurred: {
                            ZStack {
                                BlurredImage(blurhash: attachment.blurhash)
                                if self.showAvatar {
                                    ImageAvatar(displayName: self.statusViewModel.account.displayNameWithoutEmojis,
                                                avatarUrl: self.statusViewModel.account.avatar,
                                                rebloggedAccountDisplayName: self.statusViewModel.reblogStatus?.account.displayNameWithoutEmojis,
                                                rebloggedAccountAvatar: self.statusViewModel.reblogStatus?.account.avatar) { isAuthor in
                                        if isAuthor {
                                            self.routerPath.navigate(to: .userProfile(accountId: self.statusViewModel.account.id,
                                                                                      accountDisplayName: self.statusViewModel.account.displayNameWithoutEmojis,
                                                                                      accountUserName: self.statusViewModel.account.acct))
                                        } else {
                                            if let rebloggedAccountId = self.statusViewModel.reblogStatus?.account.id,
                                               let rebloggedAccountUsername = self.statusViewModel.reblogStatus?.account.acct {
                                                self.routerPath.navigate(to: .userProfile(accountId: rebloggedAccountId,
                                                                                          accountDisplayName: self.statusViewModel.reblogStatus?.account.displayNameWithoutEmojis,
                                                                                          accountUserName: rebloggedAccountUsername))
                                            }
                                        }
                                    }
                                }
                            }
                            .onTapGesture {
                                self.navigateToStatus()
                            }
                        }
                    }
                    .opacity(self.opacity)
                    .onAppear {
                        if let uiImage = state.imageResponse?.image {
                            self.recalculateSizeOfDownloadedImage(uiImage: uiImage)
                        }

                        if self.imageFromCache == false {
                            self.opacity = 0.0
                            withAnimation {
                                self.opacity = 1.0
                            }
                        }
                    }
                } else {
                    self.imageContainerView(image: image)
                        .imageContextMenu(statusModel: self.statusViewModel,
                                          attachmentModel: self.attachment,
                                          uiImage: state.imageResponse?.image)
                        .opacity(self.opacity)
                        .onAppear {
                            if let uiImage = state.imageResponse?.image {
                                self.recalculateSizeOfDownloadedImage(uiImage: uiImage)
                            }

                            if self.imageFromCache == false {
                                self.opacity = 0.0
                                withAnimation {
                                    self.opacity = 1.0
                                }
                            }
                        }
                }
            } else if state.error != nil {
                ZStack {
                    BlurredImage(blurhash: attachment.blurhash)

                    VStack(alignment: .center) {
                        Spacer()
                        Text("global.error.errorDuringImageDownload", comment: "Cannot download image")
                            .foregroundColor(.white)
                        Spacer()
                    }
                }
                .onTapGesture {
                    self.navigateToStatus()
                }
            } else {
                VStack(alignment: .center) {
                    BlurredImage(blurhash: attachment.blurhash)
                        .onTapGesture {
                            self.navigateToStatus()
                        }
                }
            }
        }
        .priority(.high)
    }

    @ViewBuilder
    private func imageContainerView(image: Image) -> some View {
        ZStack {
            self.imageView(image: image)

            if self.showAvatar {
                ImageAvatar(displayName: self.statusViewModel.account.displayNameWithoutEmojis,
                            avatarUrl: self.statusViewModel.account.avatar,
                            rebloggedAccountDisplayName: self.statusViewModel.reblogStatus?.account.displayNameWithoutEmojis,
                            rebloggedAccountAvatar: self.statusViewModel.reblogStatus?.account.avatar) { isAuthor in
                    if isAuthor {
                        self.routerPath.navigate(to: .userProfile(accountId: self.statusViewModel.account.id,
                                                                  accountDisplayName: self.statusViewModel.account.displayNameWithoutEmojis,
                                                                  accountUserName: self.statusViewModel.account.acct))
                    } else {
                        if let rebloggedAccountId = self.statusViewModel.reblogStatus?.account.id,
                           let rebloggedAccountUsername = self.statusViewModel.reblogStatus?.account.acct {
                            self.routerPath.navigate(to: .userProfile(accountId: rebloggedAccountId,
                                                                      accountDisplayName: self.statusViewModel.reblogStatus?.account.displayNameWithoutEmojis,
                                                                      accountUserName: rebloggedAccountUsername))
                        }
                    }
                }
            }

            ImageAlternativeText(text: self.attachment.description) { text in
                self.routerPath.presentedAlert = .alternativeText(text: text)
            }

            ImageFavourite(isFavourited: $isFavourited)
            FavouriteTouch(showFavouriteAnimation: $showThumbImage)
        }
    }

    @ViewBuilder
    private func imageView(image: Image) -> some View {
        image
            .resizable()
            .if(self.clipToRectangle == true) {
                $0
                    .aspectRatio(contentMode: .fill)
                    .frame(width: self.containerWidth, height: self.containerWidth)
                    .clipped()
                    // Fix issue with clickable content area outside of the visible image: https://developer.apple.com/forums/thread/123717.
                    .contentShape(Rectangle())
            }
            .if(self.clipToRectangle == false) {
                $0.aspectRatio(contentMode: .fit)
            }
            .onTapGesture(count: 2) {
                Task {
                    // Update favourite in Pixelfed server.
                    try? await self.client.statuses?.favourite(statusId: self.statusViewModel.id)
                }

                // Run adnimation and haptic feedback.
                self.showThumbImage = true
                HapticService.shared.fireHaptic(of: .buttonPress)
                
                // Mark that user performed specific action.
                TimelineDoubleTapTip().invalidate(reason: .actionPerformed)

                // Mark favourite booleans used to show star in the timeline view.
                self.statusViewModel.favourited = true
                withAnimation(.default.delay(2.0)) {
                    self.isFavourited = true
                }
            }
            .onTapGesture {
                self.navigateToStatus()
            }
            .onAppear {
                self.isFavourited = self.statusViewModel.favourited
            }
    }

    private func navigateToStatus() {
        self.routerPath.navigate(to: .status(
            id: statusViewModel.id,
            blurhash: statusViewModel.mediaAttachments.first?.blurhash,
            highestImageUrl: statusViewModel.mediaAttachments.getHighestImage()?.url,
            metaImageWidth: statusViewModel.getImageWidth(),
            metaImageHeight: statusViewModel.getImageHeight()
        ))
    }

    private func recalculateSizeOfDownloadedImage(uiImage: UIImage) {
        ImageSizeService.shared.save(for: attachment.url,
                                     width: uiImage.size.width,
                                     height: uiImage.size.height)

        let size = ImageSizeService.shared.calculate(for: attachment.url, andContainerWidth: UIScreen.main.bounds.size.width)
        self.onImageDownloaded(size.width, size.height)
    }
}