import SwiftUI



struct AlbumCollectionView: View {

    @Environment(\.managedObjectContext) var managedObjectContext

    @FetchRequest(entity: Photo.entity(),

                  sortDescriptors: [

                    NSSortDescriptor(keyPath: \Photo.id, ascending: true)

                  ]

    ) var photos: FetchedResults&lt;Photo&gt;

    

    @FetchRequest(entity: Album.entity(),

                  sortDescriptors: [

                    NSSortDescriptor(keyPath: \Album.name, ascending: true)

                  ]

    ) var albums: FetchedResults&lt;Album&gt;

    

    @State private var showingImagePicker = false

    @State private var showingPopover = false

    @State private var inputImage: UIImage?

    @State private var albumNameInput: String = &#34;&#34;

    

    let columns = [

        GridItem(.adaptive(minimum: 150))

    ]

    

    var body: some View {

        ScrollView {

            LazyVGrid(columns: columns, spacing: 20) {

                ForEach(albums) { album in

                    NavigationLink(destination: AlbumView(album: album)) {

                        VStack {

                            Image(data: album.albumCoverImageData, placeholder: &#34;No Image&#34;)

                                .resizable()

                                .aspectRatio(1, contentMode: .fill)

                                .contextMenu(menuItems: {

                                    Button(action: {

                                        deleteAlbum(selectedAlbum: album)

                                    }) {

                                        Label(&#34;Remove&#34;, systemImage: &#34;trash&#34;)

                                    }

                                })

                            Text(album.name ?? &#34;&#34;)

                        }

                    }

                }

            }

            .padding()

            .navigationBarTitle(Text(&#34;Albums&#34;))

            .navigationBarItems(trailing:

                Menu {

                    Button(action: {

                        self.showingPopover = true

                    }) {

                        Label(&#34;Add Album&#34;, systemImage: &#34;photo.on.rectangle.angled&#34;)

                    }

                    Button(action: {

                        self.showingImagePicker = true

                    }) {

                        Label(&#34;Add Photo&#34;, systemImage: &#34;photo&#34;)

                    }

                } label: {

                    Image(systemName: &#34;plus&#34;)

                        .popover(isPresented: $showingPopover, arrowEdge: .trailing ,content: {

                            VStack {

                                Text(&#34;New Album&#34;)

                                TextField(&#34;Enter Album Name&#34;,text: $albumNameInput)

                                    .textFieldStyle(RoundedBorderTextFieldStyle())

                                Button(&#34;Submit&#34;){

                                    addAlbum(name: albumNameInput, albumCover: UIImage(named: &#34;Image 1&#34;)!)

                                    albumNameInput = &#34;&#34;

                                    showingPopover = false

                                }

                            }

                            .padding()

                        })

                }

            )

            .sheet(isPresented: $showingImagePicker, onDismiss: loadImage) {

                ImagePicker(image: self.$inputImage)

            }

        }

    }

    

    func addAlbum(name: String, albumCover: UIImage) {

        let newAlbum = Album(context: managedObjectContext)

        

        newAlbum.id = UUID()

        newAlbum.name = name

        newAlbum.passwordProtected = false

        newAlbum.albumCoverImageData = albumCover.jpegData(compressionQuality: 1.0)

        

        saveContext()

    }

    

    func deleteAlbum(selectedAlbum: Album) {

        for album in albums {

            if selectedAlbum == album {

                self.managedObjectContext.delete(album)

            }

        }

        saveContext()

    }

    

    func loadImage() {

        guard let inputImage = inputImage else { return }

        addPhoto(image: inputImage)

    }

    

    func addPhoto(image: UIImage) {

        let newPhoto = Photo(context: managedObjectContext)

        

        newPhoto.id = UUID()

        newPhoto.imageData = image.jpegData(compressionQuality: 1.0)

        

        saveContext()

    }

    

    func saveContext() {

        do {

            try managedObjectContext.save()

        } catch {

            print(&#34;Error saving managed object context: \(error)&#34;)

        }

    }

}