pexavc / granite Goto Github PK

struct AskResponse: GraniteReducer {
    typealias Center = SandService.Center
    
    @Payload var response: SandGPT.Response?
    
    @Event var processDocument: AskDocumentResponse.Reducer
    
    func reduce(state: inout Center.State) {
        guard response?.isComplete == false else {
            print("[SandService] AskIsComplete")
            
            if response?.isStream == false {
                if let payload = response {
                    processDocument.send(payload) // <-------- Becomes priority stateful update (1)
                } else {
                    state.response = response?.data ?? ""
                }
            }
            
            state.isResponding = false // <---------- This won't be final (2)
            return
        }
        print("[SandService] test")
        
        state.response = response?.data ?? ""
    }
}

struct AskDocumentResponse: GraniteReducer {
    typealias Center = SandService.Center
    
    @Relay var environment: EnvironmentService
    
    @Payload var response: SandGPT.Response?
    
    func reduce(state: inout Center.State) {
        guard let response = response?.data else { return }
        let doc = Document(parsing: response, source: nil)
        let lineCount = Int(doc.range?.upperBound.line ?? 0)
        
        environment
            .center
            .responseWindowSize
            .send(
                EnvironmentService
                    .ResponseWindowSizeUpdated
                    .Meta(lineCount: lineCount))
        
        state.response = response
        state.isResponding = false // <------------- this is final (3)
    }
    
    var thread: DispatchQueue? {
        .global(qos: .utility)
    }
}