1 files changed, 370 insertions, 0 deletions
diff --git a/helix-lsp/src/jsonrpc.rs b/helix-lsp/src/jsonrpc.rs
new file mode 100644
index 00000000..b9b3fd2c
--- /dev/null
+++ b/helix-lsp/src/jsonrpc.rs
@@ -0,0 +1,370 @@
+//! An implementation of the JSONRPC 2.0 spec types
+
+// Upstream implementation: https://github.com/paritytech/jsonrpc/tree/38af3c9439aa75481805edf6c05c6622a5ab1e70/core/src/types
+// Changes from upstream:
+// * unused functions (almost all non-trait-implementation functions) have been removed
+// * `#[serde(deny_unknown_fields)]` annotations have been removed on response types
+//   for compatibility with non-strict language server implementations like Ruby Sorbet
+//   (see https://github.com/helix-editor/helix/issues/2786)
+// * some variable names have been lengthened for readability
+
+use serde::de::{self, DeserializeOwned, Visitor};
+use serde::{Deserialize, Serialize};
+use serde_json::Value;
+
+// https://www.jsonrpc.org/specification#error_object
+#[derive(Debug, PartialEq, Clone)]
+pub enum ErrorCode {
+    ParseError,
+    InvalidRequest,
+    MethodNotFound,
+    InvalidParams,
+    InternalError,
+    ServerError(i64),
+}
+
+impl ErrorCode {
+    pub fn code(&self) -> i64 {
+        match *self {
+            ErrorCode::ParseError => -32700,
+            ErrorCode::InvalidRequest => -32600,
+            ErrorCode::MethodNotFound => -32601,
+            ErrorCode::InvalidParams => -32602,
+            ErrorCode::InternalError => -32603,
+            ErrorCode::ServerError(code) => code,
+        }
+    }
+}
+
+impl From<i64> for ErrorCode {
+    fn from(code: i64) -> Self {
+        match code {
+            -32700 => ErrorCode::ParseError,
+            -32600 => ErrorCode::InvalidRequest,
+            -32601 => ErrorCode::MethodNotFound,
+            -32602 => ErrorCode::InvalidParams,
+            -32603 => ErrorCode::InternalError,
+            code => ErrorCode::ServerError(code),
+        }
+    }
+}
+
+impl<'de> Deserialize<'de> for ErrorCode {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: serde::Deserializer<'de>,
+    {
+        let code: i64 = Deserialize::deserialize(deserializer)?;
+        Ok(ErrorCode::from(code))
+    }
+}
+
+impl Serialize for ErrorCode {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: serde::Serializer,
+    {
+        serializer.serialize_i64(self.code())
+    }
+}
+
+#[derive(Debug, PartialEq, Clone, Serialize, Deserialize)]
+pub struct Error {
+    pub code: ErrorCode,
+    pub message: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub data: Option<Value>,
+}
+
+impl Error {
+    pub fn invalid_params<M>(message: M) -> Self
+    where
+        M: Into<String>,
+    {
+        Error {
+            code: ErrorCode::InvalidParams,
+            message: message.into(),
+            data: None,
+        }
+    }
+}
+
+impl std::fmt::Display for Error {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{:?}: {}", self.code, self.message)
+    }
+}
+
+impl std::error::Error for Error {}
+
+// https://www.jsonrpc.org/specification#request_object
+
+/// Request ID
+#[derive(Debug, PartialEq, Clone, Hash, Eq, Deserialize, Serialize)]
+#[serde(untagged)]
+pub enum Id {
+    Null,
+    Num(u64),
+    Str(String),
+}
+
+/// Protocol Version
+#[derive(Debug, PartialEq, Clone, Copy, Hash, Eq)]
+pub enum Version {
+    V2,
+}
+
+impl Serialize for Version {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: serde::Serializer,
+    {
+        match *self {
+            Version::V2 => serializer.serialize_str("2.0"),
+        }
+    }
+}
+
+struct VersionVisitor;
+
+impl<'v> Visitor<'v> for VersionVisitor {
+    type Value = Version;
+
+    fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
+        formatter.write_str("a string")
+    }
+    fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
+    where
+        E: de::Error,
+    {
+        match value {
+            "2.0" => Ok(Version::V2),
+            _ => Err(de::Error::custom("invalid version")),
+        }
+    }
+}
+
+impl<'de> Deserialize<'de> for Version {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: serde::Deserializer<'de>,
+    {
+        deserializer.deserialize_identifier(VersionVisitor)
+    }
+}
+
+#[derive(Debug, PartialEq, Clone, Serialize, Deserialize)]
+#[serde(untagged)]
+pub enum Params {
+    None,
+    Array(Vec<Value>),
+    Map(serde_json::Map<String, Value>),
+}
+
+impl Params {
+    pub fn parse<D>(self) -> Result<D, Error>
+    where
+        D: DeserializeOwned,
+    {
+        let value: Value = self.into();
+        serde_json::from_value(value)
+            .map_err(|err| Error::invalid_params(format!("Invalid params: {}.", err)))
+    }
+}
+
+impl From<Params> for Value {
+    fn from(params: Params) -> Value {
+        match params {
+            Params::Array(vec) => Value::Array(vec),
+            Params::Map(map) => Value::Object(map),
+            Params::None => Value::Null,
+        }
+    }
+}
+
+#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
+#[serde(deny_unknown_fields)]
+pub struct MethodCall {
+    pub jsonrpc: Option<Version>,
+    pub method: String,
+    #[serde(default = "default_params")]
+    pub params: Params,
+    pub id: Id,
+}
+
+#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
+#[serde(deny_unknown_fields)]
+pub struct Notification {
+    pub jsonrpc: Option<Version>,
+    pub method: String,
+    #[serde(default = "default_params")]
+    pub params: Params,
+}
+
+#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
+#[serde(deny_unknown_fields)]
+#[serde(untagged)]
+pub enum Call {
+    MethodCall(MethodCall),
+    Notification(Notification),
+    Invalid {
+        // We can attempt to salvage the id out of the invalid request
+        // for better debugging
+        #[serde(default = "default_id")]
+        id: Id,
+    },
+}
+
+fn default_params() -> Params {
+    Params::None
+}
+
+fn default_id() -> Id {
+    Id::Null
+}
+
+impl From<MethodCall> for Call {
+    fn from(method_call: MethodCall) -> Self {
+        Call::MethodCall(method_call)
+    }
+}
+
+impl From<Notification> for Call {
+    fn from(notification: Notification) -> Self {
+        Call::Notification(notification)
+    }
+}
+
+#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
+#[serde(deny_unknown_fields)]
+#[serde(untagged)]
+pub enum Request {
+    Single(Call),
+    Batch(Vec<Call>),
+}
+
+// https://www.jsonrpc.org/specification#response_object
+
+#[derive(Debug, PartialEq, Clone, Serialize, Deserialize)]
+pub struct Success {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub jsonrpc: Option<Version>,
+    pub result: Value,
+    pub id: Id,
+}
+
+#[derive(Debug, PartialEq, Clone, Deserialize, Serialize)]
+pub struct Failure {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub jsonrpc: Option<Version>,
+    pub error: Error,
+    pub id: Id,
+}
+
+// Note that failure comes first because we're not using
+// #[serde(deny_unknown_field)]: we want a request that contains
+// both `result` and `error` to be a `Failure`.
+#[derive(Debug, PartialEq, Clone, Deserialize, Serialize)]
+#[serde(untagged)]
+pub enum Output {
+    Failure(Failure),
+    Success(Success),
+}
+
+impl From<Output> for Result<Value, Error> {
+    fn from(output: Output) -> Self {
+        match output {
+            Output::Success(success) => Ok(success.result),
+            Output::Failure(failure) => Err(failure.error),
+        }
+    }
+}
+
+#[derive(Clone, Debug, PartialEq, Deserialize, Serialize)]
+#[serde(untagged)]
+pub enum Response {
+    Single(Output),
+    Batch(Vec<Output>),
+}
+
+impl From<Failure> for Response {
+    fn from(failure: Failure) -> Self {
+        Response::Single(Output::Failure(failure))
+    }
+}
+
+impl From<Success> for Response {
+    fn from(success: Success) -> Self {
+        Response::Single(Output::Success(success))
+    }
+}
+
+#[test]
+fn method_call_serialize() {
+    use serde_json;
+
+    let m = MethodCall {
+        jsonrpc: Some(Version::V2),
+        method: "update".to_owned(),
+        params: Params::Array(vec![Value::from(1), Value::from(2)]),
+        id: Id::Num(1),
+    };
+
+    let serialized = serde_json::to_string(&m).unwrap();
+    assert_eq!(
+        serialized,
+        r#"{"jsonrpc":"2.0","method":"update","params":[1,2],"id":1}"#
+    );
+}
+
+#[test]
+fn notification_serialize() {
+    use serde_json;
+
+    let n = Notification {
+        jsonrpc: Some(Version::V2),
+        method: "update".to_owned(),
+        params: Params::Array(vec![Value::from(1), Value::from(2)]),
+    };
+
+    let serialized = serde_json::to_string(&n).unwrap();
+    assert_eq!(
+        serialized,
+        r#"{"jsonrpc":"2.0","method":"update","params":[1,2]}"#
+    );
+}
+
+#[test]
+fn success_output_deserialize() {
+    use serde_json;
+
+    let dso = r#"{"jsonrpc":"2.0","result":1,"id":1}"#;
+
+    let deserialized: Output = serde_json::from_str(dso).unwrap();
+    assert_eq!(
+        deserialized,
+        Output::Success(Success {
+            jsonrpc: Some(Version::V2),
+            result: Value::from(1),
+            id: Id::Num(1)
+        })
+    );
+}
+
+#[test]
+fn success_output_deserialize_with_extra_fields() {
+    use serde_json;
+
+    // https://github.com/helix-editor/helix/issues/2786
+    let dso = r#"{"jsonrpc":"2.0","result":1,"id":1,"requestMethod":"initialize"}"#;
+
+    let deserialized: Output = serde_json::from_str(dso).unwrap();
+    assert_eq!(
+        deserialized,
+        Output::Success(Success {
+            jsonrpc: Some(Version::V2),
+            result: Value::from(1),
+            id: Id::Num(1)
+        })
+    );
+}