git.delta.rocks / jrsonnet / refs/commits / fad91f8a3923

--- /dev/null
+++ b/crates/nixlike/Cargo.toml
@@ -0,0 +1,14 @@
+[package]
+name = "nixlike"
+version = "0.1.0"
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+anyhow = "1.0.44"
+dprint-core = "0.46.2"
+peg = "0.7.0"
+serde = "1.0.130"
+serde_json = "1.0.68"
+thiserror = "1.0.29"

--- /dev/null
+++ b/crates/nixlike/src/de_impl.rs
@@ -0,0 +1,368 @@
+use std::{
+	collections::BTreeMap,
+	convert::{TryFrom, TryInto},
+};
+
+use serde::{
+	de::{self, MapAccess, SeqAccess},
+	Deserializer,
+};
+
+use crate::{Error, Value};
+
+struct ObjectAccess {
+	iter: std::collections::btree_map::IntoIter<String, Value>,
+	value: Option<Value>,
+}
+impl ObjectAccess {
+	fn new(v: BTreeMap<String, Value>) -> Self {
+		Self {
+			iter: v.into_iter(),
+			value: None,
+		}
+	}
+}
+
+impl<'de> MapAccess<'de> for ObjectAccess {
+	type Error = Error;
+
+	fn next_key_seed<K>(&mut self, seed: K) -> Result<Option<K::Value>, Self::Error>
+	where
+		K: de::DeserializeSeed<'de>,
+	{
+		if let Some((k, v)) = self.iter.next() {
+			let _ = self.value.insert(v);
+			Ok(Some(seed.deserialize(Value::String(k))?))
+		} else {
+			Ok(None)
+		}
+	}
+
+	fn next_value_seed<V>(&mut self, seed: V) -> Result<V::Value, Self::Error>
+	where
+		V: de::DeserializeSeed<'de>,
+	{
+		seed.deserialize(self.value.take().unwrap())
+	}
+}
+
+struct ArrayAccess {
+	iter: std::vec::IntoIter<Value>,
+}
+impl ArrayAccess {
+	fn new(v: Vec<Value>) -> Self {
+		Self {
+			iter: v.into_iter(),
+		}
+	}
+}
+
+impl<'de> SeqAccess<'de> for ArrayAccess {
+	type Error = Error;
+
+	fn next_element_seed<T>(&mut self, seed: T) -> Result<Option<T::Value>, Self::Error>
+	where
+		T: de::DeserializeSeed<'de>,
+	{
+		if let Some(v) = self.iter.next() {
+			Ok(Some(seed.deserialize(v)?))
+		} else {
+			Ok(None)
+		}
+	}
+}
+
+impl Value {
+	fn parse_int<T: TryFrom<i64>>(&self) -> Result<T, Error> {
+		match self {
+			Value::Number(n) => Ok((*n).try_into().map_err(|_| Error::BadNumber)?),
+			_ => Err(Error::Expected("integer")),
+		}
+	}
+	fn parse_boolean(self) -> Result<bool, Error> {
+		match self {
+			Value::Boolean(b) => Ok(b),
+			_ => Err(Error::Expected("boolean")),
+		}
+	}
+	pub fn parse_string(&self) -> Result<&str, Error> {
+		match self {
+			Value::String(s) => Ok(s),
+			_ => Err(Error::Expected("string")),
+		}
+	}
+	fn parse_char(self) -> Result<char, Error> {
+		match self {
+			Value::String(s) if s.chars().count() == 1 => Ok(s.chars().next().unwrap()),
+			_ => Err(Error::Expected("char")),
+		}
+	}
+	fn parse_array(self) -> Result<Vec<Value>, Error> {
+		match self {
+			Value::Array(s) => Ok(s),
+			_ => Err(Error::Expected("array")),
+		}
+	}
+	fn parse_object(self) -> Result<BTreeMap<String, Value>, Error> {
+		match self {
+			Value::Object(s) => Ok(s),
+			_ => Err(Error::Expected("object")),
+		}
+	}
+	fn parse_null(self) -> Result<(), Error> {
+		match self {
+			Value::Null => Ok(()),
+			_ => Err(Error::Expected("null")),
+		}
+	}
+}
+
+impl de::Error for Error {
+	fn custom<T>(msg: T) -> Self
+	where
+		T: std::fmt::Display,
+	{
+		Self::Custom(format!("{}", msg))
+	}
+}
+
+impl<'de> Deserializer<'de> for Value {
+	type Error = Error;
+
+	fn deserialize_any<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		match self {
+			Value::Number(f) => visitor.visit_i64(f),
+			Value::String(s) => visitor.visit_str(&s),
+			Value::Boolean(b) => visitor.visit_bool(b),
+			Value::Object(o) => visitor.visit_map(ObjectAccess::new(o)),
+			Value::Array(a) => visitor.visit_seq(ArrayAccess::new(a)),
+			Value::Null => visitor.visit_none(),
+		}
+	}
+
+	fn deserialize_bool<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_bool(self.parse_boolean()?)
+	}
+
+	fn deserialize_i8<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_i8(self.parse_int()?)
+	}
+
+	fn deserialize_i16<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_i16(self.parse_int()?)
+	}
+
+	fn deserialize_i32<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_i32(self.parse_int()?)
+	}
+
+	fn deserialize_i64<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_i64(self.parse_int()?)
+	}
+
+	fn deserialize_u8<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_u8(self.parse_int()?)
+	}
+
+	fn deserialize_u16<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_u16(self.parse_int()?)
+	}
+
+	fn deserialize_u32<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_u32(self.parse_int()?)
+	}
+
+	fn deserialize_u64<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_u64(self.parse_int()?)
+	}
+
+	fn deserialize_f32<V>(self, _visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		todo!()
+	}
+
+	fn deserialize_f64<V>(self, _visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		todo!()
+	}
+
+	fn deserialize_char<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_char(self.parse_char()?)
+	}
+
+	fn deserialize_str<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_str(self.parse_string()?)
+	}
+
+	fn deserialize_string<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_string(self.parse_string()?.to_owned())
+	}
+
+	fn deserialize_bytes<V>(self, _visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		todo!()
+	}
+
+	fn deserialize_byte_buf<V>(self, _visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		todo!()
+	}
+
+	fn deserialize_option<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		match self {
+			Value::Null => visitor.visit_none(),
+			v => visitor.visit_some(v),
+		}
+	}
+
+	fn deserialize_unit<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		self.parse_null()?;
+		visitor.visit_unit()
+	}
+
+	fn deserialize_unit_struct<V>(
+		self,
+		_name: &'static str,
+		visitor: V,
+	) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		self.deserialize_unit(visitor)
+	}
+
+	fn deserialize_newtype_struct<V>(
+		self,
+		_name: &'static str,
+		visitor: V,
+	) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_newtype_struct(self)
+	}
+
+	fn deserialize_seq<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_seq(self.parse_array().map(ArrayAccess::new)?)
+	}
+
+	fn deserialize_tuple<V>(self, _len: usize, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		self.deserialize_seq(visitor)
+	}
+
+	fn deserialize_tuple_struct<V>(
+		self,
+		_name: &'static str,
+		_len: usize,
+		visitor: V,
+	) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		self.deserialize_seq(visitor)
+	}
+
+	fn deserialize_map<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		visitor.visit_map(self.parse_object().map(ObjectAccess::new)?)
+	}
+
+	fn deserialize_struct<V>(
+		self,
+		_name: &'static str,
+		_fields: &'static [&'static str],
+		visitor: V,
+	) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		self.deserialize_map(visitor)
+	}
+
+	fn deserialize_enum<V>(
+		self,
+		_name: &'static str,
+		_variants: &'static [&'static str],
+		_visitor: V,
+	) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		todo!()
+	}
+
+	fn deserialize_identifier<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		self.deserialize_str(visitor)
+	}
+
+	fn deserialize_ignored_any<V>(self, visitor: V) -> Result<V::Value, Self::Error>
+	where
+		V: serde::de::Visitor<'de>,
+	{
+		self.deserialize_any(visitor)
+	}
+}

--- /dev/null
+++ b/crates/nixlike/src/lib.rs
@@ -0,0 +1,125 @@
+use std::collections::BTreeMap;
+
+use peg::str::LineCol;
+use se_impl::MySerialize;
+use serde::{Deserialize, Serialize};
+
+mod de_impl;
+mod se_impl;
+mod to_string;
+
+#[derive(thiserror::Error, Debug)]
+pub enum Error {
+	#[error("bad number")]
+	BadNumber,
+	#[error("expected {0}")]
+	Expected(&'static str),
+	#[error("parse error")]
+	ParseError(#[from] peg::error::ParseError<LineCol>),
+	#[error("{0}")]
+	Custom(String),
+	#[error("io: {0}")]
+	Io(#[from] std::io::Error),
+	#[error("fmt: {0}")]
+	Fmt(#[from] std::fmt::Error),
+}
+
+#[derive(Debug)]
+pub enum Value {
+	Number(i64),
+	String(String),
+	Boolean(bool),
+	Object(BTreeMap<String, Value>),
+	Array(Vec<Value>),
+	Null,
+}
+
+peg::parser! {
+pub grammar nixlike() for str {
+	rule number() -> i64
+		= v:$(['0'..='9' | '+' | '-']+) {? v.parse().map_err(|_| "<number>")}
+	rule string() -> String
+		= "\"" v:$((!"\"" [_])+) "\"" { v.to_owned() }
+	rule boolean() -> bool
+		= "true" {true}
+		/ "false" {false}
+	rule indent() -> String
+		= s:$(['a'..='z' | 'A'..='Z' | '0'..='9' | '_' | '-']+) { s.to_owned() }
+	rule object() -> BTreeMap<String, Value>
+		= "{" _
+			e:(k:indent()++(_ "." _) _ "=" _ v:value() _ ";" _ {(k, v)})*
+		"}" {?
+			let mut out = BTreeMap::new();
+			for (k, v) in e {
+				let mut map = &mut out;
+				for v in k.iter().take(k.len() - 1) {
+					map = match map.entry(v.clone()).or_insert_with(|| Value::Object(Default::default())) {
+						Value::Object(v) => v,
+						_ => return Err("expected object"),
+					}
+				}
+
+				let key = k.into_iter().last().unwrap();
+				if map.contains_key(&key) {
+					return Err("can't override object");
+				}
+				map.insert(key, v);
+			}
+			Ok(out)
+		}
+
+	rule array() -> Vec<Value>
+		= "[" _ v:value()**_ _ "]" {v}
+
+	rule value() -> Value
+		= o:object() { Value::Object(o) }
+		/ a:array() { Value::Array(a) }
+		/ s:string() { Value::String(s) }
+		/ b:boolean() { Value::Boolean(b) }
+		/ n:number() { Value::Number(n) }
+
+	pub rule root() -> Value
+		= _ v:value() _ { v }
+
+	rule _()
+		= [' ' | '\t' | '\n']*
+}
+}
+
+pub fn parse_str<'de, D: Deserialize<'de>>(s: &str) -> Result<D, Error> {
+	let value = nixlike::root(s)?;
+	D::deserialize(value)
+}
+
+pub fn parse_value<'de, D: Deserialize<'de>>(value: Value) -> Result<D, Error> {
+	D::deserialize(value)
+}
+
+pub fn serialize_value_pretty(value: Value) -> Result<String, Error> {
+	to_string::write_nix(&value)
+}
+
+pub fn serialize<S: Serialize>(value: S) -> Result<String, Error> {
+	let value: Value = value.serialize(MySerialize)?;
+	serialize_value_pretty(value)
+}
+
+#[test]
+fn test() {
+	let v: serde_json::Value = parse_str(
+		r#"
+			{
+				b.c = 2;
+				b.d = "hello";
+				c = {
+					k = 123;
+					p = 231;
+					ll = [1 2 3 [] [[4 5 6]] ];
+				};
+			}
+		"#,
+	)
+	.unwrap();
+	let s: String = serialize(v).unwrap();
+	println!("{}", s);
+}

--- /dev/null
+++ b/crates/nixlike/src/to_string.rs
@@ -0,0 +1,74 @@
+use crate::{Error, Value};
+
+fn write_nix_obj_key_buf(
+	k: &str,
+	v: &Value,
+	out: &mut String,
+	indent: &mut String,
+) -> Result<(), Error> {
+	use std::fmt::Write;
+	write!(out, "{}", k)?;
+	match v {
+		Value::Object(o) if o.len() == 1 => {
+			let (k, v) = o.iter().next().unwrap();
+			write!(out, ".")?;
+			write_nix_obj_key_buf(k, v, out, indent)?;
+		}
+		v => {
+			write!(out, " = ")?;
+			write_nix_buf(v, out, indent)?;
+			writeln!(out, ";")?;
+		}
+	}
+	Ok(())
+}
+
+fn write_nix_buf(value: &Value, out: &mut String, indent: &mut String) -> Result<(), Error> {
+	use std::fmt::Write;
+	match value {
+		Value::Null => write!(out, "null")?,
+		Value::Boolean(v) => write!(out, "{:?}", v)?,
+		Value::Number(n) => write!(out, "{}", n)?,
+		Value::String(s) => write!(out, "{:?}", s)?,
+		Value::Array(a) => {
+			if a.is_empty() {
+				write!(out, "[ ]")?;
+			} else {
+				writeln!(out, "[")?;
+				let old_len = indent.len();
+				indent.push_str("  ");
+				for item in a {
+					write!(out, "{}", indent)?;
+					write_nix_buf(item, out, indent)?;
+					writeln!(out)?;
+				}
+				indent.truncate(old_len);
+				write!(out, "{}]", indent)?;
+			}
+		}
+		Value::Object(obj) => {
+			if obj.is_empty() {
+				write!(out, "{{ }}")?;
+			} else {
+				writeln!(out, "{{")?;
+				let old_len = indent.len();
+				indent.push_str("  ");
+				for (k, v) in obj {
+					write!(out, "{}", indent)?;
+					write_nix_obj_key_buf(k, v, out, indent)?;
+				}
+				indent.truncate(old_len);
+				write!(out, "{}}}", indent)?;
+			}
+		}
+	};
+	Ok(())
+}
+
+pub fn write_nix(value: &Value) -> Result<String, Error> {
+	let mut out = String::new();
+	let mut indent = String::new();
+
+	write_nix_buf(value, &mut out, &mut indent)?;
+	Ok(out)
+}