view Simoleon/UI/FavoriteButton.swift @ 156:84137052813d

Refactor code
author Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
date Sat, 28 Aug 2021 11:15:25 +0100
parents Simoleon/Helpers/FavoriteButton.swift@081f0857af51
children 82bd84c5973c
line wrap: on
line source

//
//  FavoriteButton.swift
//  Simoleon
//
//  Created by Dennis Concepción Martín on 19/07/2021.
//

import SwiftUI

struct FavoriteButton: View {
    @State var currencyPair: CurrencyPairModel
    @Environment(\.managedObjectContext) private var viewContext
    @FetchRequest(sortDescriptors: []) private var favorites: FetchedResults<Favorite>
    @State private var starSymbol = "star"
    
    var body: some View {
        Button(action: favoriteAction) {
            RoundedRectangle(cornerRadius: 15)
                .foregroundColor(Color(.secondarySystemBackground))
                .frame(width: 60, height: 60)
                .overlay(
                    Image(systemName: generateStar())
                        .font(.system(size: 28))
                        .foregroundColor(Color(.systemYellow))
                )
        }
        .accessibilityIdentifier("AddToFavorites")
    }
    
    /*
     If currency pair is favorite:
     * Button action is to remove from favorites
     else:
     * Button action is to add to favorites
     */
    private func favoriteAction() {
        let favoriteCurrencyPairs = favorites.map { $0.currencyPair }
        let currencyPair = "(\(currencyPair.baseSymbol)/\(currencyPair.quoteSymbol)"
        if favoriteCurrencyPairs.contains(currencyPair) {
            removeFromFavorites()
        } else {
            addToFavorites()
        }
        
        let haptics = Haptics()
        haptics.simpleSuccess()
    }
    
    /*
     if currency pair is favorite:
     * Return "star.fill" symbol
     else:
     * Return "star"
     */
    private func generateStar() -> String {
        let favoriteCurrencyPairs = favorites.map { $0.currencyPair }
        let currencyPair = "(\(currencyPair.baseSymbol)/\(currencyPair.quoteSymbol)"
        if favoriteCurrencyPairs.contains(currencyPair) {
            return "star.fill"
        } else {
            return "star"
        }
    }
    
    /*
     * Get first favorite core data object that matches the specified currency pair
     * Delete it
     */
    private func removeFromFavorites() {
        let currencyPair = "(\(currencyPair.baseSymbol)/\(currencyPair.quoteSymbol)"
        withAnimation {
            let favoriteObject = favorites.first(where: { $0.currencyPair == currencyPair })
            viewContext.delete(favoriteObject ?? Favorite())

            do {
                try viewContext.save()
            } catch {
                let nsError = error as NSError
                fatalError("Unresolved error \(nsError), \(nsError.userInfo)")
            }
        }
    }
    
    /*
     * Create a favorite core data object
     * Save it
     */
    private func addToFavorites() {
        let currencyPair = "(\(currencyPair.baseSymbol)/\(currencyPair.quoteSymbol)"
        withAnimation {
            let favorite = Favorite(context: viewContext)
            favorite.currencyPair = currencyPair

            do {
                try viewContext.save()
            } catch {
                let nsError = error as NSError
                fatalError("Unresolved error \(nsError), \(nsError.userInfo)")
            }
        }
    }
}

struct FavoriteButton_Previews: PreviewProvider {
    static var previews: some View {
        FavoriteButton(currencyPair: CurrencyPairModel(baseSymbol: "USD", quoteSymbol: "EUR"))
    }
}