|
| 1 | +using System; |
| 2 | +using System.Linq.Expressions; |
| 3 | +using System.Reflection; |
| 4 | +using SQLitePCL.pretty.Orm.Attributes; |
| 5 | +using System.Linq; |
| 6 | + |
| 7 | +namespace SQLitePCL.pretty.Orm.Sql |
| 8 | +{ |
| 9 | + internal static partial class SqlCompiler |
| 10 | + { |
| 11 | + private static string CompileExpr(this Expression This) |
| 12 | + { |
| 13 | + if (This is BinaryExpression) |
| 14 | + { |
| 15 | + var bin = (BinaryExpression)This; |
| 16 | + |
| 17 | + var leftExpr = bin.Left.CompileExpr(); |
| 18 | + var rightExpr = bin.Right.CompileExpr(); |
| 19 | + |
| 20 | + if (rightExpr == "NULL" && bin.NodeType == ExpressionType.Equal) |
| 21 | + { |
| 22 | + if (bin.NodeType == ExpressionType.Equal) |
| 23 | + { |
| 24 | + return "(" + leftExpr + "IS NULL)"; |
| 25 | + } |
| 26 | + else if (rightExpr == "NULL" && bin.NodeType == ExpressionType.NotEqual) |
| 27 | + { |
| 28 | + return "(" + leftExpr + "IS NOT NULL)"; |
| 29 | + } |
| 30 | + } |
| 31 | + |
| 32 | + return "(" + leftExpr + " " + GetSqlName(bin) + " " + rightExpr + ")"; |
| 33 | + } |
| 34 | + else if (This is ParameterExpression) |
| 35 | + { |
| 36 | + var param = (ParameterExpression)This; |
| 37 | + return ":" + param.Name; |
| 38 | + } |
| 39 | + else if (This is MemberExpression) |
| 40 | + { |
| 41 | + var member = (MemberExpression)This; |
| 42 | + |
| 43 | + if (member.Expression != null && member.Expression.NodeType == ExpressionType.Parameter) |
| 44 | + { |
| 45 | + // This is a column in the table, output the column name |
| 46 | + var tableName = TableMapping.Get(member.Expression.Type).TableName; |
| 47 | + var columnName = ((PropertyInfo) member.Member).GetColumnName(); |
| 48 | + return "\"" + tableName + "\".\"" + columnName + "\""; |
| 49 | + } |
| 50 | + else |
| 51 | + { |
| 52 | + return member.EvaluateExpression().ConvertToSQLiteValue().ToSqlString(); |
| 53 | + } |
| 54 | + } |
| 55 | + else if (This.NodeType == ExpressionType.Not) |
| 56 | + { |
| 57 | + var operandExpr = ((UnaryExpression)This).Operand; |
| 58 | + return "NOT(" + operandExpr.CompileExpr() + ")"; |
| 59 | + } |
| 60 | + else if (This is ConstantExpression) |
| 61 | + { |
| 62 | + return This.EvaluateExpression().ConvertToSQLiteValue().ToSqlString(); |
| 63 | + } |
| 64 | + else if (This is MethodCallExpression) |
| 65 | + { |
| 66 | + var call = (MethodCallExpression)This; |
| 67 | + var args = new String[call.Arguments.Count]; |
| 68 | + |
| 69 | + var obj = call.Object != null ? call.Object.CompileExpr() : null; |
| 70 | + |
| 71 | + for (var i = 0; i < args.Length; i++) |
| 72 | + { |
| 73 | + args[i] = call.Arguments[i].CompileExpr(); |
| 74 | + } |
| 75 | + |
| 76 | + if (call.Method.Name == "Like" && args.Length == 2) |
| 77 | + { |
| 78 | + return "(" + args[0] + " LIKE " + args[1] + ")"; |
| 79 | + } |
| 80 | + else if (call.Method.Name == "Contains" && args.Length == 2) |
| 81 | + { |
| 82 | + return "(" + args[1] + " IN " + args[0] + ")"; |
| 83 | + } |
| 84 | + else if (call.Method.Name == "Contains" && args.Length == 1) |
| 85 | + { |
| 86 | + if (call.Object != null && call.Object.Type == typeof(string)) |
| 87 | + { |
| 88 | + return "(" + obj + " LIKE ('%' || " + args[0] + " || '%'))"; |
| 89 | + } |
| 90 | + else |
| 91 | + { |
| 92 | + return "(" + args[0] + " IN " + obj + ")"; |
| 93 | + } |
| 94 | + } |
| 95 | + else if (call.Method.Name == "StartsWith" && args.Length == 1) |
| 96 | + { |
| 97 | + return "(" + obj + " LIKE (" + args[0] + " || '%'))"; |
| 98 | + } |
| 99 | + else if (call.Method.Name == "EndsWith" && args.Length == 1) |
| 100 | + { |
| 101 | + return "(" + obj + " LIKE ('%' || " + args[0] + "))"; |
| 102 | + } |
| 103 | + else if (call.Method.Name == "Equals" && args.Length == 1) |
| 104 | + { |
| 105 | + return "(" + obj + " = (" + args[0] + "))"; |
| 106 | + } |
| 107 | + else if (call.Method.Name == "Is" && args.Length == 2) |
| 108 | + { |
| 109 | + return "(" + args[0] + " IS " + args[1] + ")"; |
| 110 | + } |
| 111 | + else if (call.Method.Name == "IsNot" && args.Length == 2) |
| 112 | + { |
| 113 | + return "(" + args[0] + " IS NOT " + args[1] + ")"; |
| 114 | + } |
| 115 | + } |
| 116 | + else if (This.NodeType == ExpressionType.Convert) |
| 117 | + { |
| 118 | + var u = (UnaryExpression)This; |
| 119 | + var ty = u.Type; |
| 120 | + var value = EvaluateExpression(u.Operand); |
| 121 | + |
| 122 | + return value.ConvertTo(ty).ConvertToSQLiteValue().ToSqlString(); |
| 123 | + } |
| 124 | + else if (This.NodeType == ExpressionType.Default) |
| 125 | + { |
| 126 | + var d = (DefaultExpression)This; |
| 127 | + var ty = d.Type; |
| 128 | + if (ty == typeof(void)) |
| 129 | + { |
| 130 | + return ""; |
| 131 | + } |
| 132 | + } |
| 133 | + else if (This.NodeType == ExpressionType.Lambda) |
| 134 | + { |
| 135 | + var expr = (LambdaExpression) This; |
| 136 | + return CompileExpr(expr.Body); |
| 137 | + } |
| 138 | + |
| 139 | + throw new NotSupportedException("Cannot compile: " + This.NodeType.ToString()); |
| 140 | + } |
| 141 | + |
| 142 | + private static object EvaluateExpression(this Expression expr) |
| 143 | + { |
| 144 | + if (expr is ConstantExpression) |
| 145 | + { |
| 146 | + var c = (ConstantExpression) expr; |
| 147 | + return c.Value; |
| 148 | + } |
| 149 | + else if (expr is MemberExpression) |
| 150 | + { |
| 151 | + var memberExpr = (MemberExpression) expr; |
| 152 | + var obj = EvaluateExpression(memberExpr.Expression); |
| 153 | + |
| 154 | + if (memberExpr.Member is PropertyInfo) |
| 155 | + { |
| 156 | + var m = (PropertyInfo) memberExpr.Member; |
| 157 | + return m.GetValue(obj, null); |
| 158 | + } |
| 159 | + |
| 160 | + else if (memberExpr.Member is FieldInfo) |
| 161 | + { |
| 162 | + var m = (FieldInfo) memberExpr.Member; |
| 163 | + return m.GetValue(obj); |
| 164 | + } |
| 165 | + } |
| 166 | + |
| 167 | + throw new NotSupportedException("Cannot compile: " + expr.NodeType.ToString()); |
| 168 | + } |
| 169 | + |
| 170 | + private static string ToSqlString(this ISQLiteValue value) |
| 171 | + { |
| 172 | + switch (value.SQLiteType) |
| 173 | + { |
| 174 | + case SQLiteType.Null: |
| 175 | + return "NULL"; |
| 176 | + case SQLiteType.Text: |
| 177 | + case SQLiteType.Blob: |
| 178 | + return "\"" + value.ToString() + "\""; |
| 179 | + default: |
| 180 | + return value.ToString(); |
| 181 | + } |
| 182 | + } |
| 183 | + |
| 184 | + private static ISQLiteValue ConvertToSQLiteValue(this object This) |
| 185 | + { |
| 186 | + if (This == null) { return SQLiteValue.Null; } |
| 187 | + |
| 188 | + Type t = This.GetType(); |
| 189 | + |
| 190 | + if (typeof(string) == t) { return ((string) This).ToSQLiteValue(); } |
| 191 | + else if ( |
| 192 | + (typeof(Int32) == t) |
| 193 | + || (typeof(Boolean) == t) |
| 194 | + || (typeof(Byte) == t) |
| 195 | + || (typeof(UInt16) == t) |
| 196 | + || (typeof(Int16) == t) |
| 197 | + || (typeof(sbyte) == t) |
| 198 | + || (typeof(Int64) == t) |
| 199 | + || (typeof(UInt32) == t)) { return ((long)(Convert.ChangeType(This, typeof(long)))).ToSQLiteValue(); } |
| 200 | + else if ((typeof(double) == t) || (typeof(float) == t) || (typeof(decimal) == t)) { return ((double)(Convert.ChangeType(This, typeof(double)))).ToSQLiteValue(); } |
| 201 | + else if (typeof(byte[]) == t) { return ((byte[]) This).ToSQLiteValue(); } |
| 202 | + else if (t.GetTypeInfo().ImplementedInterfaces.Contains(typeof(ISQLiteValue))) { return (ISQLiteValue) This; } |
| 203 | + else if (This is TimeSpan) { return ((TimeSpan) This).ToSQLiteValue(); } |
| 204 | + else if (This is DateTime) { return ((DateTime) This).ToSQLiteValue(); } |
| 205 | + else if (This is DateTimeOffset) { return ((DateTimeOffset) This).ToSQLiteValue(); } |
| 206 | + else if (This is Guid) { return ((Guid) This).ToSQLiteValue(); } |
| 207 | + //else if (obj is Stream) { return ((Stream) obj); } |
| 208 | + else if (This is Uri) { return ((Uri) This).ToSQLiteValue(); } |
| 209 | + else |
| 210 | + { |
| 211 | + throw new ArgumentException("Invalid type conversion" + t); |
| 212 | + } |
| 213 | + } |
| 214 | + |
| 215 | + private static string GetSqlName(Expression prop) |
| 216 | + { |
| 217 | + var n = prop.NodeType; |
| 218 | + |
| 219 | + switch (n) |
| 220 | + { |
| 221 | + case ExpressionType.GreaterThan: return ">"; |
| 222 | + case ExpressionType.GreaterThanOrEqual: return ">="; |
| 223 | + case ExpressionType.LessThan: return "<"; |
| 224 | + case ExpressionType.LessThanOrEqual: return "<="; |
| 225 | + case ExpressionType.And: return "&"; |
| 226 | + case ExpressionType.AndAlso: return "and"; |
| 227 | + case ExpressionType.Or: return "|"; |
| 228 | + case ExpressionType.OrElse: return "or"; |
| 229 | + case ExpressionType.Equal: return "="; |
| 230 | + case ExpressionType.NotEqual: return "!="; |
| 231 | + default: |
| 232 | + throw new NotSupportedException ("Cannot get SQL for: " + n); |
| 233 | + } |
| 234 | + } |
| 235 | + } |
| 236 | +} |
| 237 | + |
0 commit comments