use anyhow::Result;
use serde::Deserialize;
use serde::Serialize;
use std::path::PathBuf;
use std::sync::Arc;
#[cfg(feature = "async_runtime")]
use crate::async_runtime::FutureExt;
#[cfg(feature = "async_runtime")]
use crate::async_runtime::LocalBoxFuture;
use crate::configuration::ConfigKeyMap;
use crate::configuration::ConfigKeyValue;
use crate::configuration::ConfigurationDiagnostic;
use crate::configuration::GlobalConfiguration;
use crate::configuration::ResolveConfigurationResult;
use crate::plugins::PluginInfo;
use super::FileMatchingInfo;
pub trait CancellationToken: Send + Sync {
fn is_cancelled(&self) -> bool;
#[cfg(feature = "async_runtime")]
fn wait_cancellation(&self) -> LocalBoxFuture<'static, ()>;
}
#[cfg(feature = "async_runtime")]
impl CancellationToken for tokio_util::sync::CancellationToken {
fn is_cancelled(&self) -> bool {
self.is_cancelled()
}
fn wait_cancellation(&self) -> LocalBoxFuture<'static, ()> {
let token = self.clone();
async move { token.cancelled().await }.boxed_local()
}
}
pub struct NullCancellationToken;
impl CancellationToken for NullCancellationToken {
fn is_cancelled(&self) -> bool {
false
}
#[cfg(feature = "async_runtime")]
fn wait_cancellation(&self) -> LocalBoxFuture<'static, ()> {
Box::pin(std::future::pending())
}
}
pub type FormatRange = Option<std::ops::Range<usize>>;
#[derive(Debug)]
pub struct CriticalFormatError(pub anyhow::Error);
impl std::fmt::Display for CriticalFormatError {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
self.0.fmt(f)
}
}
impl std::error::Error for CriticalFormatError {
fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
self.0.source()
}
}
#[cfg(feature = "process")]
pub struct HostFormatRequest {
pub file_path: PathBuf,
pub file_text: String,
pub range: FormatRange,
pub override_config: ConfigKeyMap,
pub token: Arc<dyn CancellationToken>,
}
pub type FormatResult = Result<Option<String>>;
#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
pub struct FormatConfigId(u32);
impl std::fmt::Display for FormatConfigId {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "${}", self.0)
}
}
impl FormatConfigId {
pub fn from_raw(raw: u32) -> FormatConfigId {
FormatConfigId(raw)
}
pub fn uninitialized() -> FormatConfigId {
FormatConfigId(0)
}
pub fn as_raw(&self) -> u32 {
self.0
}
}
pub struct FormatRequest<TConfiguration> {
pub file_path: PathBuf,
pub file_text: String,
pub config_id: FormatConfigId,
pub config: Arc<TConfiguration>,
pub range: FormatRange,
pub token: Arc<dyn CancellationToken>,
}
#[derive(Clone, PartialEq, Eq, Debug, Serialize, Deserialize)]
#[serde(untagged)]
pub enum ConfigChangePathItem {
String(String),
Number(usize),
}
impl From<String> for ConfigChangePathItem {
fn from(value: String) -> Self {
Self::String(value)
}
}
impl From<usize> for ConfigChangePathItem {
fn from(value: usize) -> Self {
Self::Number(value)
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ConfigChange {
pub path: Vec<ConfigChangePathItem>,
#[serde(flatten)]
pub kind: ConfigChangeKind,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "kind", content = "value")]
pub enum ConfigChangeKind {
Add(ConfigKeyValue),
Set(ConfigKeyValue),
Remove,
}
#[derive(Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct PluginResolveConfigurationResult<T>
where
T: Clone + Serialize,
{
pub file_matching: FileMatchingInfo,
pub diagnostics: Vec<ConfigurationDiagnostic>,
pub config: T,
}
#[cfg(feature = "process")]
#[crate::async_runtime::async_trait(?Send)]
pub trait AsyncPluginHandler: Send + Sync + 'static {
type Configuration: Serialize + Clone + Send + Sync;
fn plugin_info(&self) -> PluginInfo;
fn license_text(&self) -> String;
async fn resolve_config(&self, config: ConfigKeyMap, global_config: GlobalConfiguration) -> PluginResolveConfigurationResult<Self::Configuration>;
async fn check_config_updates(&self, _plugin_config: ConfigKeyMap) -> Result<Vec<ConfigChange>> {
Ok(Vec::new())
}
async fn format(
&self,
request: FormatRequest<Self::Configuration>,
format_with_host: impl FnMut(HostFormatRequest) -> LocalBoxFuture<'static, FormatResult> + 'static,
) -> FormatResult;
}
#[cfg(feature = "wasm")]
#[derive(Clone, Serialize, serde::Deserialize, Debug, PartialEq, Eq)]
pub struct SyncPluginInfo {
#[serde(flatten)]
pub info: PluginInfo,
#[serde(flatten)]
pub file_matching: FileMatchingInfo,
}
#[cfg(feature = "wasm")]
pub trait SyncPluginHandler<TConfiguration: Clone + serde::Serialize> {
fn resolve_config(&mut self, config: ConfigKeyMap, global_config: &GlobalConfiguration) -> ResolveConfigurationResult<TConfiguration>;
fn plugin_info(&mut self) -> SyncPluginInfo;
fn license_text(&mut self) -> String;
fn format(
&mut self,
file_path: &std::path::Path,
file_text: &str,
config: &TConfiguration,
format_with_host: impl FnMut(&std::path::Path, String, &ConfigKeyMap) -> FormatResult,
) -> FormatResult;
}