philips-software/cogito

import Foundation
import ReSwift
import SwiftyJSON

struct ValueStoreService: TelepathService {
    let store: Store<AppState>

    init(store: Store<AppState>) {
        self.store = store
    }

    func addKeyValuePair(key: String, request: JsonRpcRequest, channel: TelepathChannel) {
        if let value = request.params["value"].string {
            store.dispatch(ValueStoreActions.Add(key: key, value: value))
            store.dispatch(TelepathActions.Send(id: request.id, result: "success", on: channel))
        } else {
            store.dispatch(TelepathActions.Send(
                id: request.id,
                error: ValueStoreError.valueNotFound,
                on: channel))
        }
    }

    func getValueForKey(key: String, request: JsonRpcRequest, channel: TelepathChannel) {
        if let value = store.state.valueStore.valueForKey(key: key) {
            store.dispatch(TelepathActions.Send(id: request.id, result: value, on: channel))
        } else {
            store.dispatch(TelepathActions.Send(id: request.id, result: JSON.null, on: channel))
        }
    }

    func deleteKey(key: String, request: JsonRpcRequest, channel: TelepathChannel) {
        if store.state.valueStore.store[key] != nil {
            store.dispatch(ValueStoreActions.Delete(key: key))
            store.dispatch(TelepathActions.Send(id: request.id, result: "success", on: channel))
        } else {
            store.dispatch(TelepathActions.Send(id: request.id,
                                                error: ValueStoreError.noKeyInStore,
                                                on: channel))
        }
    }

    func onRequest(_ request: JsonRpcRequest, on channel: TelepathChannel) {
        guard ["addKeyValuePair", "getValueForKey", "deleteKey"].contains(request.method) else { return }
        if let key = request.params["key"].string {
            switch request.method {
            case "addKeyValuePair":
                addKeyValuePair(key: key, request: request, channel: channel)
            case "getValueForKey":
                getValueForKey(key: key, request: request, channel: channel)
            case "deleteKey":
                deleteKey(key: key, request: request, channel: channel)
            default:
                break
            }
        } else {
            store.dispatch(TelepathActions.Send(id: request.id, error: ValueStoreError.keyNotFound, on: channel))
        }

    }
}