Switch from toml::from_slice to toml::from_str (#5659)

This commit is contained in:
Pascal Kuthe 2023-01-24 17:07:01 +01:00 committed by GitHub
parent 64ec0256d3
commit e9dc9f4935
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 25 additions and 16 deletions

View File

@ -1,6 +1,9 @@
use std::str::from_utf8;
/// Default built-in languages.toml. /// Default built-in languages.toml.
pub fn default_lang_config() -> toml::Value { pub fn default_lang_config() -> toml::Value {
toml::from_slice(include_bytes!("../../languages.toml")) let default_config = include_bytes!("../../languages.toml");
toml::from_str(from_utf8(default_config).unwrap())
.expect("Could not parse built-in languages.toml to valid toml") .expect("Could not parse built-in languages.toml to valid toml")
} }
@ -11,8 +14,8 @@ pub fn user_lang_config() -> Result<toml::Value, toml::de::Error> {
.chain([crate::config_dir()].into_iter()) .chain([crate::config_dir()].into_iter())
.map(|path| path.join("languages.toml")) .map(|path| path.join("languages.toml"))
.filter_map(|file| { .filter_map(|file| {
std::fs::read(&file) std::fs::read_to_string(&file)
.map(|config| toml::from_slice(&config)) .map(|config| toml::from_str(&config))
.ok() .ok()
}) })
.collect::<Result<Vec<_>, _>>()? .collect::<Result<Vec<_>, _>>()?

View File

@ -179,6 +179,8 @@ fn get_name(v: &Value) -> Option<&str> {
#[cfg(test)] #[cfg(test)]
mod merge_toml_tests { mod merge_toml_tests {
use std::str;
use super::merge_toml_values; use super::merge_toml_values;
use toml::Value; use toml::Value;
@ -191,8 +193,9 @@ fn language_toml_map_merges() {
indent = { tab-width = 4, unit = " ", test = "aaa" } indent = { tab-width = 4, unit = " ", test = "aaa" }
"#; "#;
let base: Value = toml::from_slice(include_bytes!("../../languages.toml")) let base = include_bytes!("../../languages.toml");
.expect("Couldn't parse built-in languages config"); let base = str::from_utf8(base).expect("Couldn't parse built-in languages config");
let base: Value = toml::from_str(base).expect("Couldn't parse built-in languages config");
let user: Value = toml::from_str(USER).unwrap(); let user: Value = toml::from_str(USER).unwrap();
let merged = merge_toml_values(base, user, 3); let merged = merge_toml_values(base, user, 3);
@ -224,8 +227,9 @@ fn language_toml_nested_array_merges() {
language-server = { command = "deno", args = ["lsp"] } language-server = { command = "deno", args = ["lsp"] }
"#; "#;
let base: Value = toml::from_slice(include_bytes!("../../languages.toml")) let base = include_bytes!("../../languages.toml");
.expect("Couldn't parse built-in languages config"); let base = str::from_utf8(base).expect("Couldn't parse built-in languages config");
let base: Value = toml::from_str(base).expect("Couldn't parse built-in languages config");
let user: Value = toml::from_str(USER).unwrap(); let user: Value = toml::from_str(USER).unwrap();
let merged = merge_toml_values(base, user, 3); let merged = merge_toml_values(base, user, 3);

View File

@ -1,6 +1,7 @@
use std::{ use std::{
collections::HashMap, collections::HashMap,
path::{Path, PathBuf}, path::{Path, PathBuf},
str,
}; };
use anyhow::{anyhow, Context, Result}; use anyhow::{anyhow, Context, Result};
@ -15,12 +16,13 @@
pub use crate::graphics::{Color, Modifier, Style}; pub use crate::graphics::{Color, Modifier, Style};
pub static DEFAULT_THEME_DATA: Lazy<Value> = Lazy::new(|| { pub static DEFAULT_THEME_DATA: Lazy<Value> = Lazy::new(|| {
toml::from_slice(include_bytes!("../../theme.toml")).expect("Failed to parse default theme") let bytes = include_bytes!("../../theme.toml");
toml::from_str(str::from_utf8(bytes).unwrap()).expect("Failed to parse base default theme")
}); });
pub static BASE16_DEFAULT_THEME_DATA: Lazy<Value> = Lazy::new(|| { pub static BASE16_DEFAULT_THEME_DATA: Lazy<Value> = Lazy::new(|| {
toml::from_slice(include_bytes!("../../base16_theme.toml")) let bytes = include_bytes!("../../base16_theme.toml");
.expect("Failed to parse base 16 default theme") toml::from_str(str::from_utf8(bytes).unwrap()).expect("Failed to parse base 16 default theme")
}); });
pub static DEFAULT_THEME: Lazy<Theme> = Lazy::new(|| Theme { pub static DEFAULT_THEME: Lazy<Theme> = Lazy::new(|| Theme {
@ -148,8 +150,8 @@ fn merge_themes(&self, parent_theme_toml: Value, theme_toml: Value) -> Value {
// Loads the theme data as `toml::Value` first from the user_dir then in default_dir // Loads the theme data as `toml::Value` first from the user_dir then in default_dir
fn load_toml(&self, path: PathBuf) -> Result<Value> { fn load_toml(&self, path: PathBuf) -> Result<Value> {
let data = std::fs::read(&path)?; let data = std::fs::read_to_string(&path)?;
let value = toml::from_slice(data.as_slice())?; let value = toml::from_str(&data)?;
Ok(value) Ok(value)
} }

View File

@ -39,6 +39,6 @@ pub fn find_files(dir: &Path, filename: &str) -> Vec<PathBuf> {
} }
pub fn lang_config() -> LangConfig { pub fn lang_config() -> LangConfig {
let bytes = std::fs::read(path::lang_config()).unwrap(); let text = std::fs::read_to_string(path::lang_config()).unwrap();
toml::from_slice(&bytes).unwrap() toml::from_str(&text).unwrap()
} }

View File

@ -156,8 +156,8 @@ pub fn lint(file: String) -> Result<(), DynError> {
return Ok(()); return Ok(());
} }
let path = path::themes().join(file.clone() + ".toml"); let path = path::themes().join(file.clone() + ".toml");
let theme = std::fs::read(&path).unwrap(); let theme = std::fs::read_to_string(&path).unwrap();
let theme: Theme = toml::from_slice(&theme).expect("Failed to parse theme"); let theme: Theme = toml::from_str(&theme).expect("Failed to parse theme");
let mut messages: Vec<String> = vec![]; let mut messages: Vec<String> = vec![];
get_rules().iter().for_each(|lint| match lint { get_rules().iter().for_each(|lint| match lint {