core.rs 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760
  1. use rand::Rng;
  2. use std::fs::File;
  3. use std::io::Read;
  4. use std::rc::Rc;
  5. use std::time::{SystemTime, UNIX_EPOCH};
  6. use crate::printer::pr_seq;
  7. use crate::reader::read_str;
  8. use crate::types::MalErr::ErrMalVal;
  9. use crate::types::MalVal::{
  10. Atom, Bool, Func, Hash, Int, List, MalFunc, Nil, Str, Sym, Vector, ZKScalar,
  11. };
  12. use crate::types::{MalArgs, MalRet, MalVal, _assoc, _dissoc, atom, error, func, hash_map};
  13. use bls12_381;
  14. use ff::{Field, PrimeField};
  15. use sapvi::bls_extensions::BlsStringConversion;
  16. use std::ops::{AddAssign, MulAssign, SubAssign};
  17. macro_rules! fn_t_int_int {
  18. ($ret:ident, $fn:expr) => {{
  19. |a: MalArgs| match (a[0].clone(), a[1].clone()) {
  20. (Int(a0), Int(a1)) => Ok($ret($fn(a0, a1))),
  21. _ => error("expecting (int,int) args"),
  22. }
  23. }};
  24. }
  25. macro_rules! fn_is_type {
  26. ($($ps:pat),*) => {{
  27. |a:MalArgs| { Ok(Bool(match a[0] { $($ps => true,)* _ => false})) }
  28. }};
  29. ($p:pat if $e:expr) => {{
  30. |a:MalArgs| { Ok(Bool(match a[0] { $p if $e => true, _ => false})) }
  31. }};
  32. ($p:pat if $e:expr,$($ps:pat),*) => {{
  33. |a:MalArgs| { Ok(Bool(match a[0] { $p if $e => true, $($ps => true,)* _ => false})) }
  34. }};
  35. }
  36. macro_rules! fn_str {
  37. ($fn:expr) => {{
  38. |a: MalArgs| match a[0].clone() {
  39. Str(a0) => $fn(a0),
  40. _ => error("expecting (str) arg"),
  41. }
  42. }};
  43. }
  44. fn symbol(a: MalArgs) -> MalRet {
  45. match a[0] {
  46. Str(ref s) => Ok(Sym(s.to_string())),
  47. _ => error("illegal symbol call"),
  48. }
  49. }
  50. fn slurp(f: String) -> MalRet {
  51. let mut s = String::new();
  52. match File::open(f).and_then(|mut f| f.read_to_string(&mut s)) {
  53. Ok(_) => Ok(Str(s)),
  54. Err(e) => error(&e.to_string()),
  55. }
  56. }
  57. fn time_ms(_a: MalArgs) -> MalRet {
  58. let ms_e = match SystemTime::now().duration_since(UNIX_EPOCH) {
  59. Ok(d) => d,
  60. Err(e) => return error(&format!("{:?}", e)),
  61. };
  62. Ok(Int(
  63. ms_e.as_secs() as i64 * 1000 + ms_e.subsec_nanos() as i64 / 1_000_000
  64. ))
  65. }
  66. fn get(a: MalArgs) -> MalRet {
  67. match (a[0].clone(), a[1].clone()) {
  68. (Nil, _) => Ok(Nil),
  69. (Hash(ref hm, _), Str(ref s)) => match hm.get(s) {
  70. Some(mv) => Ok(mv.clone()),
  71. None => Ok(Nil),
  72. },
  73. _ => error("illegal get args"),
  74. }
  75. }
  76. fn assoc(a: MalArgs) -> MalRet {
  77. match a[0] {
  78. Hash(ref hm, _) => _assoc((**hm).clone(), a[1..].to_vec()),
  79. _ => error("assoc on non-Hash Map"),
  80. }
  81. }
  82. fn dissoc(a: MalArgs) -> MalRet {
  83. match a[0] {
  84. Hash(ref hm, _) => _dissoc((**hm).clone(), a[1..].to_vec()),
  85. _ => error("dissoc on non-Hash Map"),
  86. }
  87. }
  88. fn contains_q(a: MalArgs) -> MalRet {
  89. match (a[0].clone(), a[1].clone()) {
  90. (Hash(ref hm, _), Str(ref s)) => Ok(Bool(hm.contains_key(s))),
  91. _ => error("illegal get args"),
  92. }
  93. }
  94. fn keys(a: MalArgs) -> MalRet {
  95. match a[0] {
  96. Hash(ref hm, _) => Ok(list!(hm.keys().map(|k| { Str(k.to_string()) }).collect())),
  97. _ => error("keys requires Hash Map"),
  98. }
  99. }
  100. fn vals(a: MalArgs) -> MalRet {
  101. match a[0] {
  102. Hash(ref hm, _) => Ok(list!(hm.values().map(|v| { v.clone() }).collect())),
  103. _ => error("keys requires Hash Map"),
  104. }
  105. }
  106. fn vec(a: MalArgs) -> MalRet {
  107. match a[0] {
  108. List(ref v, _) | Vector(ref v, _) => Ok(vector!(v.to_vec())),
  109. _ => error("non-seq passed to vec"),
  110. }
  111. }
  112. fn cons(a: MalArgs) -> MalRet {
  113. match a[1].clone() {
  114. List(v, _) | Vector(v, _) => {
  115. let mut new_v = vec![a[0].clone()];
  116. new_v.extend_from_slice(&v);
  117. Ok(list!(new_v.to_vec()))
  118. }
  119. _ => error("cons expects seq as second arg"),
  120. }
  121. }
  122. fn concat(a: MalArgs) -> MalRet {
  123. let mut new_v = vec![];
  124. for seq in a.iter() {
  125. match seq {
  126. List(v, _) | Vector(v, _) => new_v.extend_from_slice(v),
  127. _ => return error("non-seq passed to concat"),
  128. }
  129. }
  130. Ok(list!(new_v.to_vec()))
  131. }
  132. fn nth(a: MalArgs) -> MalRet {
  133. match (a[0].clone(), a[1].clone()) {
  134. (List(seq, _), Int(idx)) | (Vector(seq, _), Int(idx)) => {
  135. if seq.len() <= idx as usize {
  136. return error("nth: index out of range");
  137. }
  138. Ok(seq[idx as usize].clone())
  139. }
  140. _ => error("invalid args to nth"),
  141. }
  142. }
  143. fn unpack_bits(a: MalArgs) -> MalRet {
  144. let mut result = vec![];
  145. match a[0].clone() {
  146. Str(ref s) => {
  147. let value = bls12_381::Scalar::from_string(s);
  148. for (_, bit) in value.to_le_bits().into_iter().cloned().enumerate() {
  149. match bit {
  150. true => result.push(bls12_381::Scalar::one()),
  151. false => result.push(bls12_381::Scalar::zero()),
  152. }
  153. }
  154. Ok(list!(result
  155. .iter()
  156. .map(|a| Str(std::string::ToString::to_string(&a)[2..].to_string()))
  157. .collect::<Vec<MalVal>>()))
  158. }
  159. ZKScalar(ref s) => {
  160. for (_, bit) in s.to_le_bits().into_iter().cloned().enumerate() {
  161. match bit {
  162. true => result.push(bls12_381::Scalar::one()),
  163. false => result.push(bls12_381::Scalar::zero()),
  164. }
  165. }
  166. Ok(list!(result
  167. .iter()
  168. .map(|a| Str(std::string::ToString::to_string(&a)[2..].to_string()))
  169. .collect::<Vec<MalVal>>()))
  170. }
  171. _ => error("invalid args to unpack-bits"),
  172. }
  173. }
  174. fn last(a: MalArgs) -> MalRet {
  175. match a[0].clone() {
  176. List(ref seq, _) | Vector(ref seq, _) if seq.len() == 0 => Ok(Nil),
  177. List(ref seq, _) | Vector(ref seq, _) => Ok(seq[seq.len() - 1].clone()),
  178. Nil => Ok(Nil),
  179. _ => error("invalid args to first"),
  180. }
  181. }
  182. fn first(a: MalArgs) -> MalRet {
  183. match a[0].clone() {
  184. List(ref seq, _) | Vector(ref seq, _) if seq.len() == 0 => Ok(Nil),
  185. List(ref seq, _) | Vector(ref seq, _) => Ok(seq[0].clone()),
  186. Nil => Ok(Nil),
  187. _ => error("invalid args to first"),
  188. }
  189. }
  190. fn second(a: MalArgs) -> MalRet {
  191. match a[0].clone() {
  192. List(ref seq, _) | Vector(ref seq, _) if seq.len() < 2 => Ok(Nil),
  193. List(ref seq, _) | Vector(ref seq, _) => Ok(seq[1].clone()),
  194. Nil => Ok(Nil),
  195. _ => error("invalid args to first"),
  196. }
  197. }
  198. fn rest(a: MalArgs) -> MalRet {
  199. match a[0].clone() {
  200. List(ref seq, _) | Vector(ref seq, _) => {
  201. if seq.len() > 1 {
  202. Ok(list!(seq[1..].to_vec()))
  203. } else {
  204. Ok(list![])
  205. }
  206. }
  207. Nil => Ok(list![]),
  208. _ => error("invalid args to first"),
  209. }
  210. }
  211. fn apply(a: MalArgs) -> MalRet {
  212. match a[a.len() - 1] {
  213. List(ref v, _) | Vector(ref v, _) => {
  214. let f = &a[0];
  215. let mut fargs = a[1..a.len() - 1].to_vec();
  216. fargs.extend_from_slice(&v);
  217. f.apply(fargs)
  218. }
  219. _ => error("apply called with non-seq"),
  220. }
  221. }
  222. fn map(a: MalArgs) -> MalRet {
  223. match a[1] {
  224. List(ref v, _) | Vector(ref v, _) => {
  225. let mut res = vec![];
  226. for mv in v.iter() {
  227. res.push(a[0].apply(vec![mv.clone()])?)
  228. }
  229. Ok(list!(res))
  230. }
  231. _ => error("map called with non-seq"),
  232. }
  233. }
  234. fn conj(a: MalArgs) -> MalRet {
  235. match a[0] {
  236. List(ref v, _) => {
  237. let sl = a[1..]
  238. .iter()
  239. .rev()
  240. .map(|a| a.clone())
  241. .collect::<Vec<MalVal>>();
  242. Ok(list!([&sl[..], v].concat()))
  243. }
  244. Vector(ref v, _) => Ok(vector!([v, &a[1..]].concat())),
  245. _ => error("conj: called with non-seq"),
  246. }
  247. }
  248. fn sub_scalar(a: MalArgs) -> MalRet {
  249. match (a[0].clone(), a[1].clone()) {
  250. (Func(_, _), ZKScalar(a1)) => {
  251. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  252. if let ZKScalar(mut a0) = values[0] {
  253. a0.sub_assign(a1);
  254. Ok(ZKScalar(a0))
  255. } else {
  256. error("scalar sub expect (zkscalar, zkscalar) found (func, zkscalar)")
  257. }
  258. } else {
  259. error("scalar sub expect (zkscalar, zkscalar)")
  260. }
  261. }
  262. (Func(_, _), Str(a1)) => {
  263. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  264. let s1 = bls12_381::Scalar::from_string(&a1);
  265. if let ZKScalar(mut a0) = values[0] {
  266. a0.sub_assign(s1);
  267. Ok(ZKScalar(a0))
  268. } else {
  269. error("scalar sub expect (zkscalar, zkscalar) found (func, zkscalar)")
  270. }
  271. } else {
  272. error("scalar sub expect (zkscalar, zkscalar)")
  273. }
  274. }
  275. (ZKScalar(mut a0), ZKScalar(a1)) => {
  276. a0.sub_assign(a1);
  277. Ok(ZKScalar(a0))
  278. }
  279. (Str(a0), ZKScalar(a1)) => {
  280. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  281. s0.sub_assign(s1);
  282. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  283. }
  284. (ZKScalar(a0), Str(a1)) => {
  285. let (mut s0, s1) = (a0, bls12_381::Scalar::from_string(&a1));
  286. s0.sub_assign(s1);
  287. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  288. }
  289. (Str(a0), Str(a1)) => {
  290. let (mut s0, s1) = (
  291. bls12_381::Scalar::from_string(&a0),
  292. bls12_381::Scalar::from_string(&a1),
  293. );
  294. s0.sub_assign(s1);
  295. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  296. }
  297. _ => error(&format!("scalar sub expect (zkscalar, zkscalar) found \n {:?}", a).to_string()),
  298. }
  299. }
  300. fn mul_scalar(a: MalArgs) -> MalRet {
  301. match (a[0].clone(), a[1].clone()) {
  302. (Func(_, _), ZKScalar(a1)) => {
  303. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  304. if let ZKScalar(mut a0) = values[0] {
  305. a0.mul_assign(a1);
  306. Ok(ZKScalar(a0))
  307. } else {
  308. error("scalar mul expect (zkscalar, zkscalar) found (func, zkscalar)")
  309. }
  310. } else {
  311. error("scalar mul expect (zkscalar, zkscalar)")
  312. }
  313. }
  314. (ZKScalar(a1), Func(_, _)) => {
  315. if let Vector(ref values, _) = a[1].apply(vec![]).unwrap() {
  316. if let ZKScalar(mut a0) = values[0] {
  317. a0.mul_assign(a1);
  318. Ok(ZKScalar(a0))
  319. } else {
  320. error("scalar mul expect (zkscalar, zkscalar) found (func, zkscalar)")
  321. }
  322. } else {
  323. error("scalar mul expect (zkscalar, zkscalar)")
  324. }
  325. }
  326. (ZKScalar(mut a0), ZKScalar(a1)) => {
  327. a0.mul_assign(a1);
  328. Ok(ZKScalar(a0))
  329. }
  330. (Str(a0), Str(a1)) => {
  331. let (mut s0, s1) = (
  332. bls12_381::Scalar::from_string(&a0),
  333. bls12_381::Scalar::from_string(&a1),
  334. );
  335. s0.mul_assign(s1);
  336. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  337. }
  338. (ZKScalar(a0), Str(a1)) => {
  339. let (mut s0, s1) = (a0,
  340. bls12_381::Scalar::from_string(&a1),
  341. );
  342. s0.mul_assign(s1);
  343. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  344. }
  345. (Str(a0), ZKScalar(a1)) => {
  346. let (mut s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  347. s0.mul_assign(s1);
  348. Ok(Str(std::string::ToString::to_string(&s0)[2..].to_string()))
  349. }
  350. _ => error(
  351. &format!("scalar mul expect (zkscalar, zkscalar) \n {:?}", a).to_string())
  352. }
  353. }
  354. fn div_scalar(a: MalArgs) -> MalRet {
  355. match (a[0].clone(), a[1].clone()) {
  356. (ZKScalar(s0), ZKScalar(s1)) => {
  357. let ret = s1.invert().map(|other| *&s0 * other);
  358. if bool::from(ret.is_some()) {
  359. Ok(Str(
  360. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  361. ))
  362. } else {
  363. error("DivisionByZero")
  364. }
  365. }
  366. (Str(a0), ZKScalar(a1)) => {
  367. let (s0, s1) = (bls12_381::Scalar::from_string(&a0), a1);
  368. let ret = s1.invert().map(|other| *&s0 * other);
  369. if bool::from(ret.is_some()) {
  370. Ok(Str(
  371. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  372. ))
  373. } else {
  374. error("DivisionByZero")
  375. }
  376. }
  377. (Str(a0), Str(a1)) => {
  378. let (s0, s1) = (
  379. bls12_381::Scalar::from_string(&a0),
  380. bls12_381::Scalar::from_string(&a1),
  381. );
  382. let ret = s1.invert().map(|other| *&s0 * other);
  383. if bool::from(ret.is_some()) {
  384. Ok(Str(
  385. std::string::ToString::to_string(&ret.unwrap())[2..].to_string()
  386. ))
  387. } else {
  388. error("DivisionByZero")
  389. }
  390. }
  391. _ => error(
  392. &format!("scalar div expect (zkscalar, zkscalar) \n {:?}", a).to_string())
  393. }
  394. }
  395. fn range(a: MalArgs) -> MalRet {
  396. let mut result = vec![];
  397. match (a[0].clone(), a[1].clone()) {
  398. (Int(a0), Int(a1)) => {
  399. for n in a0..a1 {
  400. result.push(n);
  401. }
  402. Ok(list!(result.iter().map(|_a| Nil).collect::<Vec<MalVal>>()))
  403. }
  404. _ => error("expected int int"),
  405. }
  406. }
  407. fn scalar_zero(a: MalArgs) -> MalRet {
  408. match a.len() {
  409. 0 => Ok(vector![vec![ZKScalar(bls12_381::Scalar::zero())]]),
  410. _ => Ok(vector![vec![
  411. ZKScalar(bls12_381::Scalar::zero()),
  412. a[0].clone()
  413. ]]),
  414. }
  415. }
  416. fn scalar_one(a: MalArgs) -> MalRet {
  417. match a.len() {
  418. 0 => Ok(vector![vec![ZKScalar(bls12_381::Scalar::one())]]),
  419. _ => Ok(vector![vec![
  420. ZKScalar(bls12_381::Scalar::one()),
  421. a[0].clone()
  422. ]]),
  423. }
  424. }
  425. fn cs_one(_a: MalArgs) -> MalRet {
  426. Ok(vector![vec![Sym("cs::one".to_string())]])
  427. }
  428. fn negate_from(a: MalArgs) -> MalRet {
  429. match a[0].clone() {
  430. ZKScalar(a0) => Ok(ZKScalar(a0.neg())),
  431. _ => match a[0].apply(vec![])? {
  432. List(v, _) | Vector(v, _) => match v[0] {
  433. ZKScalar(val) => Ok(vector![vec![ZKScalar(val.neg())]]),
  434. _ => error("not scalar"),
  435. },
  436. _ => return error("non zkscalar passed to negate"),
  437. },
  438. }
  439. }
  440. fn scalar_from(a: MalArgs) -> MalRet {
  441. match a[0].clone() {
  442. ZKScalar(s0) => Ok(ZKScalar(s0)),
  443. Str(a0) => {
  444. let s0 = bls12_381::Scalar::from_string(&a0.to_string());
  445. Ok(ZKScalar(s0))
  446. }
  447. Int(a0) => {
  448. let s0 = bls12_381::Scalar::from(a0 as u64);
  449. Ok(ZKScalar(s0))
  450. }
  451. _ => error("scalar from expected (string or int)"),
  452. }
  453. }
  454. fn scalar_square(a: MalArgs) -> MalRet {
  455. match a[0].clone() {
  456. ZKScalar(a0) => {
  457. let z0 = a0.clone();
  458. Ok(ZKScalar(z0.square()))
  459. }
  460. Str(a0) => {
  461. let s0 = bls12_381::Scalar::from_string(&a0);
  462. Ok(ZKScalar(s0.square()))
  463. }
  464. _ => error(
  465. &format!("scalar square expect (zkscalar or string) found \n {:?}", a).to_string(),
  466. ),
  467. }
  468. }
  469. fn scalar_double(a: MalArgs) -> MalRet {
  470. match a[0].clone() {
  471. ZKScalar(a0) => {
  472. let z0 = a0.clone();
  473. Ok(ZKScalar(z0.double()))
  474. }
  475. Str(a0) => {
  476. let s0 = bls12_381::Scalar::from_string(&a0);
  477. Ok(ZKScalar(s0.double()))
  478. }
  479. _ => error(
  480. &format!("scalar double expect (zkscalar or string) found \n {:?}", a).to_string(),
  481. ),
  482. }
  483. }
  484. fn scalar_invert(a: MalArgs) -> MalRet {
  485. match a[0].clone() {
  486. Func(_, _) => {
  487. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  488. if let ZKScalar(a0) = values[0] {
  489. if a0.is_zero() {
  490. error(
  491. &format!("scalar invert divizion by zero \n {:?}", a0).to_string())
  492. } else {
  493. Ok(ZKScalar(a0.invert().unwrap()))
  494. }
  495. } else {
  496. error(
  497. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string())
  498. }
  499. } else {
  500. error(
  501. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string())
  502. }
  503. }
  504. ZKScalar(a0) => {
  505. let z0 = a0.clone();
  506. Ok(ZKScalar(z0.invert().unwrap()))
  507. }
  508. Str(a0) => {
  509. let s0 = bls12_381::Scalar::from_string(&a0);
  510. Ok(ZKScalar(s0.invert().unwrap()))
  511. }
  512. _ => error(
  513. &format!("scalar invert expect (zkscalar or string) found \n {:?}", a).to_string(),
  514. ),
  515. }
  516. }
  517. fn scalar_is_zero(a: MalArgs) -> MalRet {
  518. match a[0].clone() {
  519. Func(_, _) => {
  520. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  521. if let ZKScalar(a0) = values[0] {
  522. Ok(Bool(a0.is_zero()))
  523. } else {
  524. error(
  525. &format!("scalar is zero expect (zkscalar or string) found \n {:?}", a).to_string())
  526. }
  527. } else {
  528. error(
  529. &format!("scalar is zero expect (zkscalar or string) found \n {:?}", a).to_string())
  530. }
  531. }
  532. ZKScalar(a0) => {
  533. let z0 = a0.clone();
  534. Ok(Bool(z0.is_zero()))
  535. }
  536. Str(a0) => {
  537. let s0 = bls12_381::Scalar::from_string(&a0);
  538. Ok(Bool(s0.is_zero()))
  539. }
  540. _ => error(
  541. &format!("scalar is zero expect (zkscalar or string) found \n {:?}", a).to_string(),
  542. ),
  543. }
  544. }
  545. fn add_scalar(a: MalArgs) -> MalRet {
  546. match (a[0].clone(), a[1].clone()) {
  547. (Func(_, _), ZKScalar(a1)) => {
  548. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  549. if let ZKScalar(mut a0) = values[0] {
  550. a0.add_assign(a1);
  551. Ok(ZKScalar(a0))
  552. } else {
  553. error("scalar add expect (zkscalar, zkscalar) found (func, zkscalar)")
  554. }
  555. } else {
  556. error("scalar add expect (zkscalar, zkscalar)")
  557. }
  558. }
  559. (Func(_, _), Str(a1)) => {
  560. if let Vector(ref values, _) = a[0].apply(vec![]).unwrap() {
  561. if let ZKScalar(mut a0) = values[0] {
  562. let s1 = bls12_381::Scalar::from_string(&a1);
  563. a0.add_assign(s1);
  564. Ok(ZKScalar(a0))
  565. } else {
  566. error("scalar add expect (zkscalar, zkscalar) found (func, zkscalar)")
  567. }
  568. } else {
  569. error("scalar add expect (zkscalar, zkscalar)")
  570. }
  571. }
  572. (ZKScalar(a0), ZKScalar(a1)) => {
  573. let (mut z0, z1) = (a0.clone(), a1.clone());
  574. z0.add_assign(z1);
  575. Ok(ZKScalar(z0))
  576. }
  577. (Str(a0), Str(a1)) => {
  578. let (mut s0, s1) = (
  579. bls12_381::Scalar::from_string(&a0),
  580. bls12_381::Scalar::from_string(&a1),
  581. );
  582. s0.add_assign(s1);
  583. Ok(ZKScalar(s0))
  584. }
  585. _ => error(&format!("scalar add expect (zkscalar, zkscalar) found \n {:?}", a).to_string()),
  586. }
  587. }
  588. fn seq(a: MalArgs) -> MalRet {
  589. match a[0] {
  590. List(ref v, _) | Vector(ref v, _) if v.len() == 0 => Ok(Nil),
  591. List(ref v, _) | Vector(ref v, _) => Ok(list!(v.to_vec())),
  592. Str(ref s) if s.len() == 0 => Ok(Nil),
  593. Str(ref s) if !a[0].keyword_q() => {
  594. Ok(list!(s.chars().map(|c| { Str(c.to_string()) }).collect()))
  595. }
  596. Nil => Ok(Nil),
  597. _ => error("seq: called with non-seq"),
  598. }
  599. }
  600. fn gen_rand(a: MalArgs) -> MalRet {
  601. let mut rng = rand::thread_rng();
  602. Ok(MalVal::Int(rng.gen::<i64>()))
  603. }
  604. pub fn ns() -> Vec<(&'static str, MalVal)> {
  605. vec![
  606. ("=", func(|a| Ok(Bool(a[0] == a[1])))),
  607. ("throw", func(|a| Err(ErrMalVal(a[0].clone())))),
  608. ("nil?", func(fn_is_type!(Nil))),
  609. ("true?", func(fn_is_type!(Bool(true)))),
  610. ("false?", func(fn_is_type!(Bool(false)))),
  611. ("symbol", func(symbol)),
  612. ("symbol?", func(fn_is_type!(Sym(_)))),
  613. (
  614. "string?",
  615. func(fn_is_type!(Str(ref s) if !s.starts_with("\u{29e}"))),
  616. ),
  617. ("keyword", func(|a| a[0].keyword())),
  618. (
  619. "keyword?",
  620. func(fn_is_type!(Str(ref s) if s.starts_with("\u{29e}"))),
  621. ),
  622. ("number?", func(fn_is_type!(Int(_)))),
  623. (
  624. "fn?",
  625. func(fn_is_type!(MalFunc{is_macro,..} if !is_macro,Func(_,_))),
  626. ),
  627. (
  628. "macro?",
  629. func(fn_is_type!(MalFunc{is_macro,..} if is_macro)),
  630. ),
  631. ("pr-str", func(|a| Ok(Str(pr_seq(&a, true, "", "", " "))))),
  632. ("str", func(|a| Ok(Str(pr_seq(&a, false, "", "", ""))))),
  633. (
  634. "prn",
  635. func(|a| {
  636. println!("{}", pr_seq(&a, true, "", "", " "));
  637. Ok(Nil)
  638. }),
  639. ),
  640. (
  641. "println",
  642. func(|a| {
  643. println!("{}", pr_seq(&a, false, "", "", " "));
  644. Ok(Nil)
  645. }),
  646. ),
  647. ("read-string", func(fn_str!(|s| { read_str(s) }))),
  648. ("slurp", func(fn_str!(|f| { slurp(f) }))),
  649. ("<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  650. ("<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  651. (">", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  652. (">=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  653. ("+", func(add_scalar)),
  654. ("-", func(sub_scalar)),
  655. ("*", func(mul_scalar)),
  656. ("/", func(div_scalar)),
  657. ("time-ms", func(time_ms)),
  658. ("i+", func(fn_t_int_int!(Int, |i, j| { i + j }))),
  659. ("i-", func(fn_t_int_int!(Int, |i, j| { i - j }))),
  660. ("i*", func(fn_t_int_int!(Int, |i, j| { i * j }))),
  661. ("i/", func(fn_t_int_int!(Int, |i, j| { i / j }))),
  662. ("i<", func(fn_t_int_int!(Bool, |i, j| { i < j }))),
  663. ("i<=", func(fn_t_int_int!(Bool, |i, j| { i <= j }))),
  664. ("i>", func(fn_t_int_int!(Bool, |i, j| { i > j }))),
  665. ("i>=", func(fn_t_int_int!(Bool, |i, j| { i >= j }))),
  666. ("time-ms", func(time_ms)),
  667. ("sequential?", func(fn_is_type!(List(_, _), Vector(_, _)))),
  668. ("list", func(|a| Ok(list!(a)))),
  669. ("list?", func(fn_is_type!(List(_, _)))),
  670. ("vector", func(|a| Ok(vector!(a)))),
  671. ("vector?", func(fn_is_type!(Vector(_, _)))),
  672. ("hash-map", func(|a| hash_map(a))),
  673. ("map?", func(fn_is_type!(Hash(_, _)))),
  674. ("assoc", func(assoc)),
  675. ("dissoc", func(dissoc)),
  676. ("get", func(get)),
  677. ("contains?", func(contains_q)),
  678. ("keys", func(keys)),
  679. ("vals", func(vals)),
  680. ("vec", func(vec)),
  681. ("cons", func(cons)),
  682. ("concat", func(concat)),
  683. ("empty?", func(|a| a[0].empty_q())),
  684. ("nth", func(nth)),
  685. ("first", func(first)),
  686. ("last", func(last)),
  687. ("rest", func(rest)),
  688. ("count", func(|a| a[0].count())),
  689. ("apply", func(apply)),
  690. ("map", func(map)),
  691. ("conj", func(conj)),
  692. ("seq", func(seq)),
  693. ("meta", func(|a| a[0].get_meta())),
  694. ("with-meta", func(|a| a[0].clone().with_meta(&a[1]))),
  695. ("atom", func(|a| Ok(atom(&a[0])))),
  696. ("atom?", func(fn_is_type!(Atom(_)))),
  697. ("deref", func(|a| a[0].deref())),
  698. ("reset!", func(|a| a[0].reset_bang(&a[1]))),
  699. ("swap!", func(|a| a[0].swap_bang(&a[1..].to_vec()))),
  700. ("unpack-bits", func(unpack_bits)),
  701. ("range", func(range)),
  702. ("scalar::one", func(scalar_one)),
  703. ("neg", func(negate_from)),
  704. ("scalar::zero", func(scalar_zero)),
  705. ("scalar", func(scalar_from)),
  706. ("square", func(scalar_square)),
  707. ("cs::one", func(cs_one)),
  708. ("second", func(second)),
  709. ("genrand", func(gen_rand)),
  710. ("double", func(scalar_double)),
  711. ("invert", func(scalar_invert)),
  712. ("zero?", func(scalar_is_zero)),
  713. ]
  714. }