1 /* 2 * Hunt - A high-level D Programming Language Web framework that encourages rapid development and clean, pragmatic design. 3 * 4 * Copyright (C) 2015-2019, HuntLabs 5 * 6 * Website: https://www.huntlabs.net/ 7 * 8 * Licensed under the Apache-2.0 License. 9 * 10 */ 11 12 module hunt.framework.view.Uninode; 13 14 public 15 { 16 import hunt.framework.util.uninode.Core; 17 import hunt.framework.util.uninode.Serialization : 18 serialize = serializeToUniNode, 19 deserialize = deserializeUniNode; 20 } 21 22 23 import std.array; 24 import std.algorithm : among, map, sort; 25 import std.conv : to; 26 import std.format: fmt = format; 27 import std.typecons : Tuple, tuple; 28 29 import hunt.logging.ConsoleLogger; 30 31 import hunt.framework.view.Lexer; 32 import hunt.framework.view.Exception : TemplateRenderException, 33 assertTemplate = assertTemplateRender; 34 35 36 bool isNumericNode(ref UniNode n) 37 { 38 return cast(bool)n.kind.among!( 39 UniNode.Kind.integer, 40 UniNode.Kind.uinteger, 41 UniNode.Kind.floating 42 ); 43 } 44 45 46 bool isIntNode(ref UniNode n) 47 { 48 return cast(bool)n.kind.among!( 49 UniNode.Kind.integer, 50 UniNode.Kind.uinteger 51 ); 52 } 53 54 55 bool isFloatNode(ref UniNode n) 56 { 57 return n.kind == UniNode.Kind.floating; 58 } 59 60 61 bool isIterableNode(ref UniNode n) 62 { 63 return cast(bool)n.kind.among!( 64 UniNode.Kind.array, 65 UniNode.Kind.object, 66 UniNode.Kind.text 67 ); 68 } 69 70 void toIterableNode(ref UniNode n) 71 { 72 switch (n.kind) with (UniNode.Kind) 73 { 74 case array: 75 return; 76 case text: 77 auto a = n.get!string.map!(a => UniNode(cast(string)[a])).array; 78 if(!a.empty()) 79 n = UniNode(a); 80 return; 81 case object: 82 UniNode[] arr; 83 auto items = n.get!(UniNode[string]); 84 if(items !is null) { 85 foreach (key, val; items) 86 arr ~= UniNode([UniNode(key), val]); 87 n = UniNode(arr); 88 } 89 return; 90 default: 91 throw new TemplateRenderException("Can't implicity convert type %s to iterable".fmt(n.kind)); 92 } 93 } 94 95 void toCommonNumType(ref UniNode n1, ref UniNode n2) 96 { 97 assertTemplate(n1.isNumericNode, "Not a numeric type of %s".fmt(n1)); 98 assertTemplate(n2.isNumericNode, "Not a numeric type of %s".fmt(n2)); 99 100 if (n1.isIntNode && n2.isFloatNode) 101 { 102 n1 = UniNode(n1.get!long.to!double); 103 return; 104 } 105 106 if (n1.isFloatNode && n2.isIntNode) 107 { 108 n2 = UniNode(n2.get!long.to!double); 109 return; 110 } 111 } 112 113 114 void toCommonCmpType(ref UniNode n1, ref UniNode n2) 115 { 116 if (n1.isNumericNode && n2.isNumericNode) 117 { 118 toCommonNumType(n1, n2); 119 return; 120 } 121 if (n1.kind != n2.kind) 122 throw new TemplateRenderException("Not comparable types %s and %s".fmt(n1.kind, n2.kind)); 123 } 124 125 126 void toBoolType(ref UniNode n) 127 { 128 switch (n.kind) with (UniNode.Kind) 129 { 130 case boolean: 131 return; 132 case integer: 133 case uinteger: 134 n = UniNode(n.get!long != 0); 135 return; 136 case floating: 137 n = UniNode(n.get!double != 0); 138 return; 139 case text: 140 n = UniNode(n.get!string.length > 0); 141 return; 142 case array: 143 case object: 144 n = UniNode(n.length > 0); 145 return; 146 case nil: 147 n = UniNode(false); 148 return; 149 default: 150 throw new TemplateRenderException("Can't cast type %s to bool".fmt(n.kind)); 151 } 152 } 153 154 155 void toStringType(ref UniNode n) 156 { 157 import std.algorithm : map; 158 import std.string : join; 159 160 string getString(UniNode n) 161 { 162 bool quotes = n.kind == UniNode.Kind.text; 163 n.toStringType; 164 if (quotes) 165 return "'" ~ n.get!string ~ "'"; 166 else 167 return n.get!string; 168 } 169 170 string doSwitch() 171 { 172 final switch (n.kind) with (UniNode.Kind) 173 { 174 case nil: return ""; 175 case boolean: return n.get!bool.to!string; 176 case integer: return n.get!long.to!string; 177 case uinteger: return n.get!ulong.to!string; 178 case floating: return n.get!double.to!string; 179 case text: return n.get!string; 180 case raw: return n.get!(ubyte[]).to!string; 181 case array: return "["~n.get!(UniNode[]).map!(a => getString(a)).join(", ").to!string~"]"; 182 case object: 183 string[] results; 184 Tuple!(string, UniNode)[] sorted = []; 185 foreach (string key, ref value; n) 186 results ~= key ~ ": " ~ getString(value); 187 return "{" ~ results.join(", ").to!string ~ "}"; 188 } 189 } 190 191 n = UniNode(doSwitch()); 192 } 193 194 195 string getAsString(UniNode n) 196 { 197 n.toStringType; 198 return n.get!string; 199 } 200 201 202 void checkNodeType(ref UniNode n, UniNode.Kind kind, Position pos) 203 { 204 if (n.kind != kind) 205 assertTemplate(0, "Unexpected expression type `%s`, expected `%s`".fmt(n.kind, kind), pos); 206 } 207 208 209 210 UniNode unary(string op)(UniNode lhs) 211 if (op.among!(Operator.Plus, 212 Operator.Minus) 213 ) 214 { 215 assertTemplate(lhs.isNumericNode, "Expected int got %s".fmt(lhs.kind)); 216 217 if (lhs.isIntNode) 218 return UniNode(mixin(op ~ "lhs.get!long")); 219 else 220 return UniNode(mixin(op ~ "lhs.get!double")); 221 } 222 223 224 225 UniNode unary(string op)(UniNode lhs) 226 if (op == Operator.Not) 227 { 228 lhs.toBoolType; 229 return UniNode(!lhs.get!bool); 230 } 231 232 233 234 UniNode binary(string op)(UniNode lhs, UniNode rhs) 235 if (op.among!(Operator.Plus, 236 Operator.Minus, 237 Operator.Mul) 238 ) 239 { 240 toCommonNumType(lhs, rhs); 241 if (lhs.isIntNode) 242 return UniNode(mixin("lhs.get!long" ~ op ~ "rhs.get!long")); 243 else 244 return UniNode(mixin("lhs.get!double" ~ op ~ "rhs.get!double")); 245 } 246 247 248 249 UniNode binary(string op)(UniNode lhs, UniNode rhs) 250 if (op == Operator.DivInt) 251 { 252 assertTemplate(lhs.isIntNode, "Expected int got %s".fmt(lhs.kind)); 253 assertTemplate(rhs.isIntNode, "Expected int got %s".fmt(rhs.kind)); 254 return UniNode(lhs.get!long / rhs.get!long); 255 } 256 257 258 259 UniNode binary(string op)(UniNode lhs, UniNode rhs) 260 if (op == Operator.DivFloat 261 || op == Operator.Rem) 262 { 263 toCommonNumType(lhs, rhs); 264 265 if (lhs.isIntNode) 266 { 267 assertTemplate(rhs.get!long != 0, "Division by zero!"); 268 return UniNode(mixin("lhs.get!long" ~ op ~ "rhs.get!long")); 269 } 270 else 271 { 272 assertTemplate(rhs.get!double != 0, "Division by zero!"); 273 return UniNode(mixin("lhs.get!double" ~ op ~ "rhs.get!double")); 274 } 275 } 276 277 278 279 UniNode binary(string op)(UniNode lhs, UniNode rhs) 280 if (op == Operator.Pow) 281 { 282 toCommonNumType(lhs, rhs); 283 if (lhs.isIntNode) 284 return UniNode(lhs.get!long ^^ rhs.get!long); 285 else 286 return UniNode(lhs.get!double ^^ rhs.get!double); 287 } 288 289 290 291 UniNode binary(string op)(UniNode lhs, UniNode rhs) 292 if (op.among!(Operator.Eq, Operator.NotEq)) 293 { 294 toCommonCmpType(lhs, rhs); 295 return UniNode(mixin("lhs" ~ op ~ "rhs")); 296 } 297 298 299 300 UniNode binary(string op)(UniNode lhs, UniNode rhs) 301 if (op.among!(Operator.Less, 302 Operator.LessEq, 303 Operator.Greater, 304 Operator.GreaterEq) 305 ) 306 { 307 toCommonCmpType(lhs, rhs); 308 switch (lhs.kind) with (UniNode.Kind) 309 { 310 case integer: 311 case uinteger: 312 return UniNode(mixin("lhs.get!long" ~ op ~ "rhs.get!long")); 313 case floating: 314 return UniNode(mixin("lhs.get!double" ~ op ~ "rhs.get!double")); 315 case text: 316 return UniNode(mixin("lhs.get!string" ~ op ~ "rhs.get!string")); 317 default: 318 throw new TemplateRenderException("Not comparable type %s".fmt(lhs.kind)); 319 } 320 } 321 322 323 324 UniNode binary(string op)(UniNode lhs, UniNode rhs) 325 if (op == Operator.Or) 326 { 327 lhs.toBoolType; 328 rhs.toBoolType; 329 return UniNode(lhs.get!bool || rhs.get!bool); 330 } 331 332 333 334 UniNode binary(string op)(UniNode lhs, UniNode rhs) 335 if (op == Operator.And) 336 { 337 lhs.toBoolType; 338 rhs.toBoolType; 339 return UniNode(lhs.get!bool && rhs.get!bool); 340 } 341 342 343 344 UniNode binary(string op)(UniNode lhs, UniNode rhs) 345 if (op == Operator.Concat) 346 { 347 lhs.toStringType; 348 rhs.toStringType; 349 return UniNode(lhs.get!string ~ rhs.get!string); 350 } 351 352 353 354 UniNode binary(string op)(UniNode lhs, UniNode rhs) 355 if (op == Operator.In) 356 { 357 import std.algorithm.searching : countUntil; 358 359 switch (rhs.kind) with (UniNode.Kind) 360 { 361 case array: 362 foreach(val; rhs) 363 { 364 if (val == lhs) 365 return UniNode(true); 366 } 367 return UniNode(false); 368 case object: 369 if (lhs.kind != UniNode.Kind.text) 370 return UniNode(false); 371 return UniNode(cast(bool)(lhs.get!string in rhs)); 372 case text: 373 if (lhs.kind != UniNode.Kind.text) 374 return UniNode(false); 375 return UniNode(rhs.get!string.countUntil(lhs.get!string) >= 0); 376 default: 377 return UniNode(false); 378 } 379 }