Initial commit of antigravity2 project

2026-06-11 22:52:58 +07:00
commit de35383342
3016 changed files with 484480 additions and 0 deletions
@@ -0,0 +1,102 @@
+'use strict';
+
+const Packet = require('../packets/packet.js');
+const CommandCode = require('../constants/commands.js');
+const StringParser = require('../parsers/string.js');
+const CharsetToEncoding = require('../constants/charset_encodings.js');
+const ClientConstants = require('../constants/client.js');
+const Types = require('../constants/types.js');
+const { toParameter } = require('./encode_parameter.js');
+
+class Query {
+  constructor(sql, charsetNumber, attributes, clientFlags) {
+    this.query = sql;
+    this.charsetNumber = charsetNumber;
+    this.encoding = CharsetToEncoding[charsetNumber];
+    this.attributes = attributes;
+    this.clientFlags = clientFlags || 0;
+  }
+
+  serializeToBuffer(buffer) {
+    const useQueryAttributes =
+      this.clientFlags & ClientConstants.CLIENT_QUERY_ATTRIBUTES;
+    const sqlBuf = StringParser.encode(this.query, this.encoding);
+    const packet = new Packet(0, buffer, 0, buffer.length);
+    packet.offset = 4;
+    packet.writeInt8(CommandCode.QUERY);
+
+    if (useQueryAttributes) {
+      const attrs = this.attributes;
+      const names = attrs ? Object.keys(attrs) : [];
+      const paramCount = names.length;
+
+      packet.writeLengthCodedNumber(paramCount);
+      packet.writeLengthCodedNumber(1); // parameter_set_count, always 1
+
+      if (paramCount > 0) {
+        const parameters = names.map((name) =>
+          toParameter(attrs[name], this.encoding, 'local')
+        );
+
+        // null bitmap
+        let bitmap = 0;
+        let bitValue = 1;
+        parameters.forEach((parameter) => {
+          if (parameter.type === Types.NULL) {
+            bitmap += bitValue;
+          }
+          bitValue *= 2;
+          if (bitValue === 256) {
+            packet.writeInt8(bitmap);
+            bitmap = 0;
+            bitValue = 1;
+          }
+        });
+        if (bitValue !== 1) {
+          packet.writeInt8(bitmap);
+        }
+
+        packet.writeInt8(1); // new_params_bind_flag
+
+        // types and names
+        for (let i = 0; i < paramCount; i++) {
+          packet.writeInt8(parameters[i].type);
+          packet.writeInt8(0); // unsigned flag
+          packet.writeLengthCodedString(names[i], this.encoding);
+        }
+
+        // values
+        parameters.forEach((parameter) => {
+          if (parameter.type !== Types.NULL) {
+            parameter.writer.call(packet, parameter.value);
+          }
+        });
+      }
+    }
+
+    packet.writeBuffer(sqlBuf);
+    return packet;
+  }
+
+  toPacket() {
+    const useQueryAttributes =
+      this.clientFlags & ClientConstants.CLIENT_QUERY_ATTRIBUTES;
+
+    if (!useQueryAttributes) {
+      const buf = StringParser.encode(this.query, this.encoding);
+      const length = 5 + buf.length;
+      const buffer = Buffer.allocUnsafe(length);
+      const packet = new Packet(0, buffer, 0, length);
+      packet.offset = 4;
+      packet.writeInt8(CommandCode.QUERY);
+      packet.writeBuffer(buf);
+      return packet;
+    }
+
+    // dry run to calculate required buffer length
+    const p = this.serializeToBuffer(Packet.MockBuffer());
+    return this.serializeToBuffer(Buffer.allocUnsafe(p.offset));
+  }
+}
+
+module.exports = Query;