omnivore/apple/OmnivoreKit/Sources/App/Views/Highlights/NotebookView.swift

#if os(iOS)
  import CoreData
  import MarkdownUI
  import Models
  import Services
  import SwiftUI
  import Views

  struct NotebookView: View {
    @EnvironmentObject var dataService: DataService
    @Environment(\.presentationMode) private var presentationMode
    @StateObject var viewModel = NotebookViewModel()

    @State var showAnnotationModal = false
    @State var errorAlertMessage: String?
    @State var showErrorAlertMessage = false
    @State var noteAnnotation = ""

    let itemObjectID: NSManagedObjectID
    @Binding var hasHighlightMutations: Bool
    @State var setLabelsHighlight: Highlight?
    @State var showShareView: Bool = false

    var emptyView: some View {
      Text(LocalText.highlightCardNoHighlightsOnPage)
        .multilineTextAlignment(.leading)
        .padding(16)
    }

    var innerBody: some View {
      listView
        .navigationTitle("Notebook")
        .listStyle(PlainListStyle())
      #if os(iOS)
        .navigationBarTitleDisplayMode(.inline)
        .toolbar {
          ToolbarItem(placement: .navigationBarLeading) {
            dismissButton
          }
          ToolbarItem(placement: .navigationBarTrailing) {
            actionsMenu
          }
        }.formSheet(isPresented: $showShareView) {
          ShareSheet(activityItems: [viewModel.highlightsAsMarkdown()])
        }
      #else
        .toolbar {
          ToolbarItemGroup {
            dismissButton
          }
        }
      #endif
    }

    var noteSection: some View {
      HStack {
        // let isEmpty = viewModel.noteItem.annotation.isEmpty
        Spacer(minLength: 6)

        if let note = viewModel.noteItem, let annotation = note.annotation, !annotation.isEmpty {
          Markdown(annotation)
            .lineSpacing(6)
            .accentColor(.appGraySolid)
            .foregroundColor(.appGrayTextContrast)
            .font(.appSubheadline)
            .padding(12)
            .frame(maxWidth: .infinity, alignment: .leading)
            .background(Color.appButtonBackground)
            .cornerRadius(8)

        } else {
          Text("Add Notes...")
            .lineSpacing(6)
            .accentColor(.appGraySolid)
            .foregroundColor(.appGrayText)
            .font(.appSubheadline)
            .padding(12)
            .frame(maxWidth: .infinity, alignment: .leading)
            .background(Color.appButtonBackground)
            .cornerRadius(8)
        }
      }
      .onTapGesture {
        noteAnnotation = viewModel.noteItem?.annotation ?? ""
        showAnnotationModal = true
      }
    }

    var listView: some View {
      List {
        Section("Article Notes") {
          noteSection
            .listRowSeparator(.hidden, edges: .bottom)
        }
        Section("Highlights") {
          if viewModel.highlightItems.count > 0 {
            ForEach(Array(viewModel.highlightItems.enumerated()), id: \.offset) { idx, highlightParams in
              HighlightsListCard(
                viewModel: self.viewModel,
                highlightParams: highlightParams,
                hasHighlightMutations: $hasHighlightMutations,
                onSaveAnnotation: {
                  viewModel.updateAnnotation(
                    highlightID: highlightParams.highlightID,
                    annotation: $0,
                    dataService: dataService
                  )
                },
                onDeleteHighlight: {
                  hasHighlightMutations = true
                  viewModel.deleteHighlight(
                    highlightID: highlightParams.highlightID,
                    dataService: dataService
                  )
                },
                onSetLabels: { highlightID in
                  setLabelsHighlight = Highlight.lookup(byID: highlightID, inContext: dataService.viewContext)
                }
              )
              .listRowSeparator(.hidden, edges: idx == 0 ? .bottom : .all)
            }
          } else {
            emptyView
              .listRowSeparator(.hidden, edges: .bottom)
          }
        }
        Spacer(minLength: 120)
          .listRowSeparator(.hidden, edges: .all)
      }.sheet(item: $setLabelsHighlight) { highlight in
        ApplyLabelsView(mode: .highlight(highlight), isSearchFocused: false, onSave: { selectedLabels in
          hasHighlightMutations = true

          viewModel.setLabelsForHighlight(highlightID: highlight.unwrappedID,
                                          labels: selectedLabels,
                                          dataService: dataService)
        })
      }.sheet(isPresented: $showAnnotationModal) {
        HighlightAnnotationSheet(
          annotation: $noteAnnotation,
          onSave: {
            viewModel.updateNoteAnnotation(
              annotation: noteAnnotation,
              dataService: dataService
            )
            showAnnotationModal = false
            hasHighlightMutations = true
          },
          onCancel: {
            showAnnotationModal = false
          },
          errorAlertMessage: $errorAlertMessage,
          showErrorAlertMessage: $showErrorAlertMessage
        )
      }
    }

    var dismissButton: some View {
      Button(
        action: { presentationMode.wrappedValue.dismiss() },
        label: { Text(LocalText.genericClose) }
      )
    }

    var actionsMenu: some View {
      Menu(
        content: {
          Button(
            action: { showShareView = true },
            label: { Label(LocalText.exportGeneric, systemImage: "square.and.arrow.up") }
          )
        },
        label: {
          Image(systemName: "ellipsis")
            .foregroundColor(.appGrayTextContrast)
        }
      )
    }

    var body: some View {
      Group {
        #if os(iOS)
          NavigationView {
            innerBody
          }
        #elseif os(macOS)
          innerBody
            .frame(minWidth: 400, minHeight: 400)
        #endif
      }
      .task {
        viewModel.load(itemObjectID: itemObjectID, dataService: dataService)
      }
    }
  }
#endif