YGGverse/ggemini
1
0
Fork 0
mirror of https://github.com/YGGverse/ggemini.git synced 2026-04-01 09:35:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

return connection in result

Browse source
This commit is contained in:
yggverse 2025-02-03 01:08:31 +02:00
parent 8df7af44b5
commit 788b792167
3 changed files with 42 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

57
src/client/connection/response.rs
View file

@ -36,7 +36,7 @@ impl Response {
connection: Connection,
priority: Priority,
cancellable: Cancellable,
callback: impl FnOnce(Result<Self, Error>) + 'static,
callback: impl FnOnce(Result<Self, Error>, Connection) + 'static,
) {
from_stream_async(
Vec::with_capacity(HEADER_LEN),
@ -44,35 +44,38 @@ impl Response {
cancellable,
priority,
|result| {
callback(match result {
Ok(buffer) => match buffer.first() {
Some(byte) => match byte {
1 => match Input::from_utf8(&buffer) {
Ok(input) => Ok(Self::Input(input)),
Err(e) => Err(Error::Input(e)),
callback(
match result {
Ok(buffer) => match buffer.first() {
Some(byte) => match byte {
1 => match Input::from_utf8(&buffer) {
Ok(input) => Ok(Self::Input(input)),
Err(e) => Err(Error::Input(e)),
},
2 => match Success::from_utf8(&buffer) {
Ok(success) => Ok(Self::Success(success)),
Err(e) => Err(Error::Success(e)),
},
3 => match Redirect::from_utf8(&buffer) {
Ok(redirect) => Ok(Self::Redirect(redirect)),
Err(e) => Err(Error::Redirect(e)),
},
4 | 5 => match Failure::from_utf8(&buffer) {
Ok(failure) => Ok(Self::Failure(failure)),
Err(e) => Err(Error::Failure(e)),
},
6 => match Certificate::from_utf8(&buffer) {
Ok(certificate) => Ok(Self::Certificate(certificate)),
Err(e) => Err(Error::Certificate(e)),
},
b => Err(Error::Code(*b)),
},
2 => match Success::from_utf8(&buffer) {
Ok(success) => Ok(Self::Success(success)),
Err(e) => Err(Error::Success(e)),
},
3 => match Redirect::from_utf8(&buffer) {
Ok(redirect) => Ok(Self::Redirect(redirect)),
Err(e) => Err(Error::Redirect(e)),
},
4 | 5 => match Failure::from_utf8(&buffer) {
Ok(failure) => Ok(Self::Failure(failure)),
Err(e) => Err(Error::Failure(e)),
},
6 => match Certificate::from_utf8(&buffer) {
Ok(certificate) => Ok(Self::Certificate(certificate)),
Err(e) => Err(Error::Certificate(e)),
},
b => Err(Error::Code(*b)),
None => Err(Error::Protocol),
},
None => Err(Error::Protocol),
Err(e) => Err(e),
},
Err(e) => Err(e),
})
connection,
)
},
);
}