autotel-plugins 0.4.0

package/dist/drizzle.cjs

@@ -0,0 +1,469 @@
+'use strict';
+
+var api = require('@opentelemetry/api');
+var traceHelpers = require('autotel/trace-helpers');
+
+// src/drizzle/index.ts
+
+// src/common/constants.ts
+var SEMATTRS_DB_SYSTEM = "db.system";
+var SEMATTRS_DB_OPERATION = "db.operation";
+var SEMATTRS_DB_STATEMENT = "db.statement";
+var SEMATTRS_DB_NAME = "db.name";
+var SEMATTRS_NET_PEER_NAME = "net.peer.name";
+var SEMATTRS_NET_PEER_PORT = "net.peer.port";
+var DEFAULT_TRACER_NAME = "autotel-plugins/drizzle";
+var DEFAULT_DB_SYSTEM = "postgresql";
+var INSTRUMENTED_FLAG = "__autotelDrizzleInstrumented";
+function extractQueryText(queryArg) {
+  if (typeof queryArg === "string") {
+    return queryArg;
+  }
+  if (queryArg && typeof queryArg === "object") {
+    if (typeof queryArg.sql === "string") {
+      return queryArg.sql;
+    }
+    if (typeof queryArg.text === "string") {
+      return queryArg.text;
+    }
+    if (typeof queryArg.queryChunks === "object") {
+      const drizzleQuery = queryArg;
+      if (typeof drizzleQuery.sql === "string") {
+        return drizzleQuery.sql;
+      }
+    }
+  }
+  return void 0;
+}
+function sanitizeQueryText(queryText, maxLength) {
+  if (queryText.length <= maxLength) {
+    return queryText;
+  }
+  return `${queryText.slice(0, Math.max(0, maxLength))}...`;
+}
+function extractOperation(queryText) {
+  const trimmed = queryText.trimStart();
+  const match = /^(?<op>\w+)/u.exec(trimmed);
+  return match?.groups?.op?.toUpperCase();
+}
+function instrumentDrizzle(client, config) {
+  if (!client) {
+    return client;
+  }
+  const hasQuery = typeof client.query === "function";
+  const hasExecute = typeof client.execute === "function";
+  if (!hasQuery && !hasExecute) {
+    return client;
+  }
+  if (client[INSTRUMENTED_FLAG]) {
+    return client;
+  }
+  const {
+    tracerName = DEFAULT_TRACER_NAME,
+    dbSystem = DEFAULT_DB_SYSTEM,
+    dbName,
+    captureQueryText = true,
+    maxQueryTextLength = 1e3,
+    peerName,
+    peerPort
+  } = config ?? {};
+  const tracer = api.trace.getTracer(tracerName);
+  const originalMethod = hasQuery ? client.query : client.execute;
+  if (!originalMethod) {
+    return client;
+  }
+  const instrumentedMethod = function instrumented(...incomingArgs) {
+    const args = [...incomingArgs];
+    let callback;
+    if (typeof args.at(-1) === "function") {
+      callback = args.pop();
+    }
+    const queryText = extractQueryText(args[0]);
+    const operation = queryText ? extractOperation(queryText) : void 0;
+    const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
+    const span = tracer.startSpan(spanName, { kind: api.SpanKind.CLIENT });
+    span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
+    if (operation) {
+      span.setAttribute(SEMATTRS_DB_OPERATION, operation);
+    }
+    if (dbName) {
+      span.setAttribute(SEMATTRS_DB_NAME, dbName);
+    }
+    if (captureQueryText && queryText !== void 0) {
+      const sanitized = sanitizeQueryText(queryText, maxQueryTextLength);
+      span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
+    }
+    if (peerName) {
+      span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
+    }
+    if (peerPort) {
+      span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
+    }
+    if (callback) {
+      return traceHelpers.runWithSpan(span, () => {
+        const wrappedCallback = (err, result) => {
+          traceHelpers.finalizeSpan(span, err);
+          if (callback) {
+            callback(err, result);
+          }
+        };
+        try {
+          return Reflect.apply(originalMethod, this, [
+            ...args,
+            wrappedCallback
+          ]);
+        } catch (error) {
+          traceHelpers.finalizeSpan(span, error);
+          throw error;
+        }
+      });
+    }
+    return traceHelpers.runWithSpan(span, () => {
+      try {
+        const result = originalMethod.apply(this, args);
+        return Promise.resolve(result).then((value) => {
+          traceHelpers.finalizeSpan(span);
+          return value;
+        }).catch((error) => {
+          traceHelpers.finalizeSpan(span, error);
+          throw error;
+        });
+      } catch (error) {
+        traceHelpers.finalizeSpan(span, error);
+        throw error;
+      }
+    });
+  };
+  client[INSTRUMENTED_FLAG] = true;
+  if (hasQuery) {
+    client.query = instrumentedMethod;
+  } else {
+    client.execute = instrumentedMethod;
+  }
+  return client;
+}
+function instrumentDrizzleClient(db, config) {
+  if (!db) {
+    return db;
+  }
+  if (db[INSTRUMENTED_FLAG]) {
+    return db;
+  }
+  const {
+    tracerName = DEFAULT_TRACER_NAME,
+    dbSystem = DEFAULT_DB_SYSTEM,
+    dbName,
+    captureQueryText = true,
+    maxQueryTextLength = 1e3,
+    peerName,
+    peerPort
+  } = config ?? {};
+  const tracer = api.trace.getTracer(tracerName);
+  let instrumented = false;
+  if (db.session && !instrumented) {
+    const session = db.session;
+    if (typeof session.prepareQuery === "function" && !session[INSTRUMENTED_FLAG]) {
+      const originalPrepareQuery = session.prepareQuery;
+      session.prepareQuery = function(...args) {
+        const prepared = originalPrepareQuery.apply(this, args);
+        if (prepared && typeof prepared.execute === "function") {
+          const originalPreparedExecute = prepared.execute;
+          prepared.execute = function(...executeArgs) {
+            const queryObj = args[0];
+            const queryText = queryObj?.sql || queryObj?.queryString || extractQueryText(queryObj);
+            const operation = queryText ? extractOperation(queryText) : void 0;
+            const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
+            const span = tracer.startSpan(spanName, { kind: api.SpanKind.CLIENT });
+            span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
+            if (operation) {
+              span.setAttribute(SEMATTRS_DB_OPERATION, operation);
+            }
+            if (dbName) {
+              span.setAttribute(SEMATTRS_DB_NAME, dbName);
+            }
+            if (captureQueryText && queryText !== void 0) {
+              const sanitized = sanitizeQueryText(
+                queryText,
+                maxQueryTextLength
+              );
+              span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
+            }
+            if (peerName) {
+              span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
+            }
+            if (peerPort) {
+              span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
+            }
+            return traceHelpers.runWithSpan(span, () => {
+              try {
+                const result = originalPreparedExecute.apply(this, executeArgs);
+                return Promise.resolve(result).then((value) => {
+                  traceHelpers.finalizeSpan(span);
+                  return value;
+                }).catch((error) => {
+                  traceHelpers.finalizeSpan(span, error);
+                  throw error;
+                });
+              } catch (error) {
+                traceHelpers.finalizeSpan(span, error);
+                throw error;
+              }
+            });
+          };
+        }
+        return prepared;
+      };
+      session[INSTRUMENTED_FLAG] = true;
+      instrumented = true;
+    }
+    if (typeof session.query === "function" && !session[INSTRUMENTED_FLAG + "_query"]) {
+      const originalQuery = session.query;
+      session.query = function(queryString, params) {
+        const operation = queryString ? extractOperation(queryString) : void 0;
+        const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
+        const span = tracer.startSpan(spanName, { kind: api.SpanKind.CLIENT });
+        span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
+        if (operation) {
+          span.setAttribute(SEMATTRS_DB_OPERATION, operation);
+        }
+        if (dbName) {
+          span.setAttribute(SEMATTRS_DB_NAME, dbName);
+        }
+        if (captureQueryText && queryString !== void 0) {
+          const sanitized = sanitizeQueryText(queryString, maxQueryTextLength);
+          span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
+        }
+        if (peerName) {
+          span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
+        }
+        if (peerPort) {
+          span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
+        }
+        return traceHelpers.runWithSpan(span, () => {
+          try {
+            const result = Reflect.apply(originalQuery, this, [
+              queryString,
+              params
+            ]);
+            return Promise.resolve(result).then((value) => {
+              traceHelpers.finalizeSpan(span);
+              return value;
+            }).catch((error) => {
+              traceHelpers.finalizeSpan(span, error);
+              throw error;
+            });
+          } catch (error) {
+            traceHelpers.finalizeSpan(span, error);
+            throw error;
+          }
+        });
+      };
+      session[INSTRUMENTED_FLAG + "_query"] = true;
+      instrumented = true;
+    }
+    if (typeof session.transaction === "function" && !session[INSTRUMENTED_FLAG + "_transaction"]) {
+      const originalTransaction = session.transaction;
+      session.transaction = function(transactionCallback, ...restArgs) {
+        const wrappedCallback = async function(tx) {
+          if (tx && (tx.session || tx._?.session || tx)) {
+            const txSession = tx.session || tx._?.session || tx;
+            if (typeof tx.execute === "function" && !tx[INSTRUMENTED_FLAG + "_execute"]) {
+              const originalTxExecute = tx.execute;
+              tx.execute = function(...executeArgs) {
+                const queryText = extractQueryText(executeArgs[0]);
+                const operation = queryText ? extractOperation(queryText) : void 0;
+                const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
+                const span = tracer.startSpan(spanName, {
+                  kind: api.SpanKind.CLIENT
+                });
+                span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
+                span.setAttribute("db.transaction", true);
+                if (operation) {
+                  span.setAttribute(SEMATTRS_DB_OPERATION, operation);
+                }
+                if (dbName) {
+                  span.setAttribute(SEMATTRS_DB_NAME, dbName);
+                }
+                if (captureQueryText && queryText !== void 0) {
+                  const sanitized = sanitizeQueryText(
+                    queryText,
+                    maxQueryTextLength
+                  );
+                  span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
+                }
+                if (peerName) {
+                  span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
+                }
+                if (peerPort) {
+                  span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
+                }
+                return traceHelpers.runWithSpan(span, () => {
+                  try {
+                    const result = originalTxExecute.apply(this, executeArgs);
+                    return Promise.resolve(result).then((value) => {
+                      traceHelpers.finalizeSpan(span);
+                      return value;
+                    }).catch((error) => {
+                      traceHelpers.finalizeSpan(span, error);
+                      throw error;
+                    });
+                  } catch (error) {
+                    traceHelpers.finalizeSpan(span, error);
+                    throw error;
+                  }
+                });
+              };
+              tx[INSTRUMENTED_FLAG + "_execute"] = true;
+            }
+            if (typeof txSession.prepareQuery === "function" && !txSession[INSTRUMENTED_FLAG + "_tx"]) {
+              const originalTxPrepareQuery = txSession.prepareQuery;
+              txSession.prepareQuery = function(...prepareArgs) {
+                const prepared = originalTxPrepareQuery.apply(
+                  this,
+                  prepareArgs
+                );
+                if (prepared && typeof prepared.execute === "function") {
+                  const originalPreparedExecute = prepared.execute;
+                  prepared.execute = function(...executeArgs) {
+                    const queryObj = prepareArgs[0];
+                    const queryText = queryObj?.sql || queryObj?.queryString || extractQueryText(queryObj);
+                    const operation = queryText ? extractOperation(queryText) : void 0;
+                    const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
+                    const span = tracer.startSpan(spanName, {
+                      kind: api.SpanKind.CLIENT
+                    });
+                    span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);
+                    span.setAttribute("db.transaction", true);
+                    if (operation) {
+                      span.setAttribute(SEMATTRS_DB_OPERATION, operation);
+                    }
+                    if (dbName) {
+                      span.setAttribute(SEMATTRS_DB_NAME, dbName);
+                    }
+                    if (captureQueryText && queryText !== void 0) {
+                      const sanitized = sanitizeQueryText(
+                        queryText,
+                        maxQueryTextLength
+                      );
+                      span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
+                    }
+                    if (peerName) {
+                      span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);
+                    }
+                    if (peerPort) {
+                      span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);
+                    }
+                    return traceHelpers.runWithSpan(span, () => {
+                      try {
+                        const result = originalPreparedExecute.apply(
+                          this,
+                          executeArgs
+                        );
+                        return Promise.resolve(result).then((value) => {
+                          traceHelpers.finalizeSpan(span);
+                          return value;
+                        }).catch((error) => {
+                          traceHelpers.finalizeSpan(span, error);
+                          throw error;
+                        });
+                      } catch (error) {
+                        traceHelpers.finalizeSpan(span, error);
+                        throw error;
+                      }
+                    });
+                  };
+                }
+                return prepared;
+              };
+              txSession[INSTRUMENTED_FLAG + "_tx"] = true;
+            }
+          }
+          return transactionCallback(tx);
+        };
+        return Reflect.apply(originalTransaction, this, [
+          wrappedCallback,
+          ...restArgs
+        ]);
+      };
+      session[INSTRUMENTED_FLAG + "_transaction"] = true;
+      instrumented = true;
+    }
+  }
+  if (db.$client && !instrumented) {
+    const client = db.$client;
+    if (typeof client.query === "function" || typeof client.execute === "function") {
+      instrumentDrizzle(client, config);
+      instrumented = true;
+    }
+  }
+  if (db._ && db._.session && typeof db._.session.execute === "function" && !instrumented) {
+    const session = db._.session;
+    if (session[INSTRUMENTED_FLAG]) {
+      return db;
+    }
+    const {
+      tracerName: tracerName2 = DEFAULT_TRACER_NAME,
+      dbSystem: dbSystem2 = DEFAULT_DB_SYSTEM,
+      dbName: dbName2,
+      captureQueryText: captureQueryText2 = true,
+      maxQueryTextLength: maxQueryTextLength2 = 1e3,
+      peerName: peerName2,
+      peerPort: peerPort2
+    } = config ?? {};
+    const tracer2 = api.trace.getTracer(tracerName2);
+    const originalExecute = session.execute;
+    if (!originalExecute) {
+      return db;
+    }
+    const instrumentedExecute = function instrumented2(...args) {
+      const queryText = extractQueryText(args[0]);
+      const operation = queryText ? extractOperation(queryText) : void 0;
+      const spanName = operation ? `drizzle.${operation.toLowerCase()}` : "drizzle.query";
+      const span = tracer2.startSpan(spanName, { kind: api.SpanKind.CLIENT });
+      span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem2);
+      if (operation) {
+        span.setAttribute(SEMATTRS_DB_OPERATION, operation);
+      }
+      if (dbName2) {
+        span.setAttribute(SEMATTRS_DB_NAME, dbName2);
+      }
+      if (captureQueryText2 && queryText !== void 0) {
+        const sanitized = sanitizeQueryText(queryText, maxQueryTextLength2);
+        span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);
+      }
+      if (peerName2) {
+        span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName2);
+      }
+      if (peerPort2) {
+        span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort2);
+      }
+      return traceHelpers.runWithSpan(span, () => {
+        try {
+          const result = originalExecute.apply(this, args);
+          return Promise.resolve(result).then((value) => {
+            traceHelpers.finalizeSpan(span);
+            return value;
+          }).catch((error) => {
+            traceHelpers.finalizeSpan(span, error);
+            throw error;
+          });
+        } catch (error) {
+          traceHelpers.finalizeSpan(span, error);
+          throw error;
+        }
+      });
+    };
+    session[INSTRUMENTED_FLAG] = true;
+    session.execute = instrumentedExecute;
+    instrumented = true;
+  }
+  if (instrumented) {
+    db[INSTRUMENTED_FLAG] = true;
+  }
+  return db;
+}
+
+exports.instrumentDrizzle = instrumentDrizzle;
+exports.instrumentDrizzleClient = instrumentDrizzleClient;
+//# sourceMappingURL=drizzle.cjs.map
+//# sourceMappingURL=drizzle.cjs.map

package/dist/drizzle.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/common/constants.ts","../src/drizzle/index.ts"],"names":["trace","SpanKind","runWithSpan","finalizeSpan","tracerName","dbSystem","dbName","captureQueryText","maxQueryTextLength","peerName","peerPort","tracer","instrumented"],"mappings":";;;;;;;;AAMO,IAAM,kBAAA,GAAqB,WAAA;AAC3B,IAAM,qBAAA,GAAwB,cAAA;AAC9B,IAAM,qBAAA,GAAwB,cAAA;AAC9B,IAAM,gBAAA,GAAmB,SAAA;AAMzB,IAAM,sBAAA,GAAyB,eAAA;AAC/B,IAAM,sBAAA,GAAyB,eAAA;ACFtC,IAAM,mBAAA,GAAsB,yBAAA;AAC5B,IAAM,iBAAA,GAAoB,YAAA;AAC1B,IAAM,iBAAA,GAAoB,8BAAA;AA6D1B,SAAS,iBAAiB,QAAA,EAAuC;AAC/D,EAAA,IAAI,OAAO,aAAa,QAAA,EAAU;AAChC,IAAA,OAAO,QAAA;AAAA,EACT;AACA,EAAA,IAAI,QAAA,IAAY,OAAO,QAAA,KAAa,QAAA,EAAU;AAE5C,IAAA,IAAI,OAAQ,QAAA,CAA+B,GAAA,KAAQ,QAAA,EAAU;AAC3D,MAAA,OAAQ,QAAA,CAA6B,GAAA;AAAA,IACvC;AAEA,IAAA,IAAI,OAAQ,QAAA,CAAgC,IAAA,KAAS,QAAA,EAAU;AAC7D,MAAA,OAAQ,QAAA,CAA8B,IAAA;AAAA,IACxC;AAEA,IAAA,IACE,OAAQ,QAAA,CAAuC,WAAA,KAAgB,QAAA,EAC/D;AAEA,MAAA,MAAM,YAAA,GAAe,QAAA;AACrB,MAAA,IAAI,OAAO,YAAA,CAAa,GAAA,KAAQ,QAAA,EAAU;AACxC,QAAA,OAAO,YAAA,CAAa,GAAA;AAAA,MACtB;AAAA,IACF;AAAA,EACF;AACA,EAAA,OAAO,MAAA;AACT;AAKA,SAAS,iBAAA,CAAkB,WAAmB,SAAA,EAA2B;AACvE,EAAA,IAAI,SAAA,CAAU,UAAU,SAAA,EAAW;AACjC,IAAA,OAAO,SAAA;AAAA,EACT;AACA,EAAA,OAAO,CAAA,EAAG,UAAU,KAAA,CAAM,CAAA,EAAG,KAAK,GAAA,CAAI,CAAA,EAAG,SAAS,CAAC,CAAC,CAAA,GAAA,CAAA;AACtD;AAKA,SAAS,iBAAiB,SAAA,EAAuC;AAC/D,EAAA,MAAM,OAAA,GAAU,UAAU,SAAA,EAAU;AACpC,EAAA,MAAM,KAAA,GAAQ,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AACzC,EAAA,OAAO,KAAA,EAAO,MAAA,EAAQ,EAAA,EAAI,WAAA,EAAY;AACxC;AA2EO,SAAS,iBAAA,CACd,QACA,MAAA,EACS;AACT,EAAA,IAAI,CAAC,MAAA,EAAQ;AACX,IAAA,OAAO,MAAA;AAAA,EACT;AAGA,EAAA,MAAM,QAAA,GAAW,OAAO,MAAA,CAAO,KAAA,KAAU,UAAA;AACzC,EAAA,MAAM,UAAA,GAAa,OAAO,MAAA,CAAO,OAAA,KAAY,UAAA;AAE7C,EAAA,IAAI,CAAC,QAAA,IAAY,CAAC,UAAA,EAAY;AAC5B,IAAA,OAAO,MAAA;AAAA,EACT;AAEA,EAAA,IAAI,MAAA,CAAO,iBAAiB,CAAA,EAAG;AAC7B,IAAA,OAAO,MAAA;AAAA,EACT;AAEA,EAAA,MAAM;AAAA,IACJ,UAAA,GAAa,mBAAA;AAAA,IACb,QAAA,GAAW,iBAAA;AAAA,IACX,MAAA;AAAA,IACA,gBAAA,GAAmB,IAAA;AAAA,IACnB,kBAAA,GAAqB,GAAA;AAAA,IACrB,QAAA;AAAA,IACA;AAAA,GACF,GAAI,UAAU,EAAC;AAEf,EAAA,MAAM,MAAA,GAASA,SAAA,CAAM,SAAA,CAAU,UAAU,CAAA;AAGzC,EAAA,MAAM,cAAA,GAAiB,QAAA,GAAW,MAAA,CAAO,KAAA,GAAQ,MAAA,CAAO,OAAA;AAExD,EAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,IAAA,OAAO,MAAA;AAAA,EACT;AAEA,EAAA,MAAM,kBAAA,GAAoC,SAAS,YAAA,CAAA,GAE9C,YAAA,EACH;AACA,IAAA,MAAM,IAAA,GAAO,CAAC,GAAG,YAAY,CAAA;AAC7B,IAAA,IAAI,QAAA;AAGJ,IAAA,IAAI,OAAO,IAAA,CAAK,EAAA,CAAG,EAAE,MAAM,UAAA,EAAY;AACrC,MAAA,QAAA,GAAW,KAAK,GAAA,EAAI;AAAA,IACtB;AAGA,IAAA,MAAM,SAAA,GAAY,gBAAA,CAAiB,IAAA,CAAK,CAAC,CAAC,CAAA;AAC1C,IAAA,MAAM,SAAA,GAAY,SAAA,GAAY,gBAAA,CAAiB,SAAS,CAAA,GAAI,MAAA;AAC5D,IAAA,MAAM,WAAW,SAAA,GACb,CAAA,QAAA,EAAW,SAAA,CAAU,WAAA,EAAa,CAAA,CAAA,GAClC,eAAA;AAGJ,IAAA,MAAM,IAAA,GAAO,OAAO,SAAA,CAAU,QAAA,EAAU,EAAE,IAAA,EAAMC,YAAA,CAAS,QAAQ,CAAA;AACjE,IAAA,IAAA,CAAK,YAAA,CAAa,oBAAoB,QAAQ,CAAA;AAE9C,IAAA,IAAI,SAAA,EAAW;AACb,MAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,IACpD;AAEA,IAAA,IAAI,MAAA,EAAQ;AACV,MAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,MAAM,CAAA;AAAA,IAC5C;AAEA,IAAA,IAAI,gBAAA,IAAoB,cAAc,MAAA,EAAW;AAC/C,MAAA,MAAM,SAAA,GAAY,iBAAA,CAAkB,SAAA,EAAW,kBAAkB,CAAA;AACjE,MAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,IACpD;AAEA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,IACpD;AAEA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,IACpD;AAGA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,OAAOC,wBAAA,CAAY,MAAM,MAAM;AAC7B,QAAA,MAAM,eAAA,GAAiC,CAAC,GAAA,EAAK,MAAA,KAAW;AACtD,UAAAC,yBAAA,CAAa,MAAM,GAAG,CAAA;AACtB,UAAA,IAAI,QAAA,EAAU;AACZ,YAAA,QAAA,CAAS,KAAK,MAAM,CAAA;AAAA,UACtB;AAAA,QACF,CAAA;AAEA,QAAA,IAAI;AACF,UAAA,OAAO,OAAA,CAAQ,KAAA,CAAM,cAAA,EAAgB,IAAA,EAAM;AAAA,YACzC,GAAG,IAAA;AAAA,YACH;AAAA,WACD,CAAA;AAAA,QACH,SAAS,KAAA,EAAO;AACd,UAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,UAAA,MAAM,KAAA;AAAA,QACR;AAAA,MACF,CAAC,CAAA;AAAA,IACH;AAGA,IAAA,OAAOD,wBAAA,CAAY,MAAM,MAAM;AAC7B,MAAA,IAAI;AACF,QAAA,MAAM,MAAA,GAAS,cAAA,CAAe,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AAC9C,QAAA,OAAO,QAAQ,OAAA,CAAQ,MAAM,CAAA,CAC1B,IAAA,CAAK,CAAC,KAAA,KAAU;AACf,UAAAC,yBAAA,CAAa,IAAI,CAAA;AACjB,UAAA,OAAO,KAAA;AAAA,QACT,CAAC,CAAA,CACA,KAAA,CAAM,CAAC,KAAA,KAAU;AAChB,UAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,UAAA,MAAM,KAAA;AAAA,QACR,CAAC,CAAA;AAAA,MACL,SAAS,KAAA,EAAO;AACd,QAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,QAAA,MAAM,KAAA;AAAA,MACR;AAAA,IACF,CAAC,CAAA;AAAA,EACH,CAAA;AAEA,EAAA,MAAA,CAAO,iBAAiB,CAAA,GAAI,IAAA;AAG5B,EAAA,IAAI,QAAA,EAAU;AACZ,IAAA,MAAA,CAAO,KAAA,GAAQ,kBAAA;AAAA,EACjB,CAAA,MAAO;AACL,IAAA,MAAA,CAAO,OAAA,GAAU,kBAAA;AAAA,EACnB;AAEA,EAAA,OAAO,MAAA;AACT;AA0EO,SAAS,uBAAA,CACd,IACA,MAAA,EACK;AACL,EAAA,IAAI,CAAC,EAAA,EAAI;AACP,IAAA,OAAO,EAAA;AAAA,EACT;AAGA,EAAA,IAAI,EAAA,CAAG,iBAAiB,CAAA,EAAG;AACzB,IAAA,OAAO,EAAA;AAAA,EACT;AAEA,EAAA,MAAM;AAAA,IACJ,UAAA,GAAa,mBAAA;AAAA,IACb,QAAA,GAAW,iBAAA;AAAA,IACX,MAAA;AAAA,IACA,gBAAA,GAAmB,IAAA;AAAA,IACnB,kBAAA,GAAqB,GAAA;AAAA,IACrB,QAAA;AAAA,IACA;AAAA,GACF,GAAI,UAAU,EAAC;AAEf,EAAA,MAAM,MAAA,GAASH,SAAA,CAAM,SAAA,CAAU,UAAU,CAAA;AACzC,EAAA,IAAI,YAAA,GAAe,KAAA;AAInB,EAAA,IAAK,EAAA,CAAW,OAAA,IAAW,CAAC,YAAA,EAAc;AACxC,IAAA,MAAM,UAAW,EAAA,CAAW,OAAA;AAG5B,IAAA,IACE,OAAO,OAAA,CAAQ,YAAA,KAAiB,cAChC,CAAC,OAAA,CAAQ,iBAAiB,CAAA,EAC1B;AACA,MAAA,MAAM,uBAAuB,OAAA,CAAQ,YAAA;AAErC,MAAA,OAAA,CAAQ,YAAA,GAAe,YAAa,IAAA,EAAa;AAC/C,QAAA,MAAM,QAAA,GAAW,oBAAA,CAAqB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AAGtD,QAAA,IAAI,QAAA,IAAY,OAAO,QAAA,CAAS,OAAA,KAAY,UAAA,EAAY;AACtD,UAAA,MAAM,0BAA0B,QAAA,CAAS,OAAA;AAEzC,UAAA,QAAA,CAAS,OAAA,GAAU,YAAwB,WAAA,EAAoB;AAE7D,YAAA,MAAM,QAAA,GAAW,KAAK,CAAC,CAAA;AACvB,YAAA,MAAM,YACJ,QAAA,EAAU,GAAA,IACV,QAAA,EAAU,WAAA,IACV,iBAAiB,QAAQ,CAAA;AAC3B,YAAA,MAAM,SAAA,GAAY,SAAA,GACd,gBAAA,CAAiB,SAAS,CAAA,GAC1B,MAAA;AACJ,YAAA,MAAM,WAAW,SAAA,GACb,CAAA,QAAA,EAAW,SAAA,CAAU,WAAA,EAAa,CAAA,CAAA,GAClC,eAAA;AAGJ,YAAA,MAAM,IAAA,GAAO,OAAO,SAAA,CAAU,QAAA,EAAU,EAAE,IAAA,EAAMC,YAAA,CAAS,QAAQ,CAAA;AACjE,YAAA,IAAA,CAAK,YAAA,CAAa,oBAAoB,QAAQ,CAAA;AAE9C,YAAA,IAAI,SAAA,EAAW;AACb,cAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,YACpD;AAEA,YAAA,IAAI,MAAA,EAAQ;AACV,cAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,MAAM,CAAA;AAAA,YAC5C;AAEA,YAAA,IAAI,gBAAA,IAAoB,cAAc,MAAA,EAAW;AAC/C,cAAA,MAAM,SAAA,GAAY,iBAAA;AAAA,gBAChB,SAAA;AAAA,gBACA;AAAA,eACF;AACA,cAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,YACpD;AAEA,YAAA,IAAI,QAAA,EAAU;AACZ,cAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,YACpD;AAEA,YAAA,IAAI,QAAA,EAAU;AACZ,cAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,YACpD;AAGA,YAAA,OAAOC,wBAAA,CAAY,MAAM,MAAM;AAC7B,cAAA,IAAI;AACF,gBAAA,MAAM,MAAA,GAAS,uBAAA,CAAwB,KAAA,CAAM,IAAA,EAAM,WAAW,CAAA;AAC9D,gBAAA,OAAO,QAAQ,OAAA,CAAQ,MAAM,CAAA,CAC1B,IAAA,CAAK,CAAC,KAAA,KAAU;AACf,kBAAAC,yBAAA,CAAa,IAAI,CAAA;AACjB,kBAAA,OAAO,KAAA;AAAA,gBACT,CAAC,CAAA,CACA,KAAA,CAAM,CAAC,KAAA,KAAU;AAChB,kBAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,kBAAA,MAAM,KAAA;AAAA,gBACR,CAAC,CAAA;AAAA,cACL,SAAS,KAAA,EAAO;AACd,gBAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,gBAAA,MAAM,KAAA;AAAA,cACR;AAAA,YACF,CAAC,CAAA;AAAA,UACH,CAAA;AAAA,QACF;AAEA,QAAA,OAAO,QAAA;AAAA,MACT,CAAA;AAEA,MAAA,OAAA,CAAQ,iBAAiB,CAAA,GAAI,IAAA;AAC7B,MAAA,YAAA,GAAe,IAAA;AAAA,IACjB;AAGA,IAAA,IACE,OAAO,QAAQ,KAAA,KAAU,UAAA,IACzB,CAAC,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,CAAA,EACrC;AACA,MAAA,MAAM,gBAAgB,OAAA,CAAQ,KAAA;AAE9B,MAAA,OAAA,CAAQ,KAAA,GAAQ,SAAqB,WAAA,EAAqB,MAAA,EAAe;AACvE,QAAA,MAAM,SAAA,GAAY,WAAA,GACd,gBAAA,CAAiB,WAAW,CAAA,GAC5B,MAAA;AACJ,QAAA,MAAM,WAAW,SAAA,GACb,CAAA,QAAA,EAAW,SAAA,CAAU,WAAA,EAAa,CAAA,CAAA,GAClC,eAAA;AAGJ,QAAA,MAAM,IAAA,GAAO,OAAO,SAAA,CAAU,QAAA,EAAU,EAAE,IAAA,EAAMF,YAAA,CAAS,QAAQ,CAAA;AACjE,QAAA,IAAA,CAAK,YAAA,CAAa,oBAAoB,QAAQ,CAAA;AAE9C,QAAA,IAAI,SAAA,EAAW;AACb,UAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,QACpD;AAEA,QAAA,IAAI,MAAA,EAAQ;AACV,UAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,MAAM,CAAA;AAAA,QAC5C;AAEA,QAAA,IAAI,gBAAA,IAAoB,gBAAgB,MAAA,EAAW;AACjD,UAAA,MAAM,SAAA,GAAY,iBAAA,CAAkB,WAAA,EAAa,kBAAkB,CAAA;AACnE,UAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,QACpD;AAEA,QAAA,IAAI,QAAA,EAAU;AACZ,UAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,QACpD;AAEA,QAAA,IAAI,QAAA,EAAU;AACZ,UAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,QACpD;AAGA,QAAA,OAAOC,wBAAA,CAAY,MAAM,MAAM;AAC7B,UAAA,IAAI;AACF,YAAA,MAAM,MAAA,GAAS,OAAA,CAAQ,KAAA,CAAM,aAAA,EAAe,IAAA,EAAM;AAAA,cAChD,WAAA;AAAA,cACA;AAAA,aACD,CAAA;AACD,YAAA,OAAO,QAAQ,OAAA,CAAQ,MAAM,CAAA,CAC1B,IAAA,CAAK,CAAC,KAAA,KAAU;AACf,cAAAC,yBAAA,CAAa,IAAI,CAAA;AACjB,cAAA,OAAO,KAAA;AAAA,YACT,CAAC,CAAA,CACA,KAAA,CAAM,CAAC,KAAA,KAAU;AAChB,cAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,cAAA,MAAM,KAAA;AAAA,YACR,CAAC,CAAA;AAAA,UACL,SAAS,KAAA,EAAO;AACd,YAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,YAAA,MAAM,KAAA;AAAA,UACR;AAAA,QACF,CAAC,CAAA;AAAA,MACH,CAAA;AAEA,MAAA,OAAA,CAAQ,iBAAA,GAAoB,QAAQ,CAAA,GAAI,IAAA;AACxC,MAAA,YAAA,GAAe,IAAA;AAAA,IACjB;AAGA,IAAA,IACE,OAAO,QAAQ,WAAA,KAAgB,UAAA,IAC/B,CAAC,OAAA,CAAQ,iBAAA,GAAoB,cAAc,CAAA,EAC3C;AACA,MAAA,MAAM,sBAAsB,OAAA,CAAQ,WAAA;AAEpC,MAAA,OAAA,CAAQ,WAAA,GAAc,SAEpB,mBAAA,EAAA,GACG,QAAA,EACH;AAEA,QAAA,MAAM,eAAA,GAAkB,eAAgB,EAAA,EAAS;AAE/C,UAAA,IAAI,OAAO,EAAA,CAAG,OAAA,IAAW,EAAA,CAAG,CAAA,EAAG,WAAW,EAAA,CAAA,EAAK;AAC7C,YAAA,MAAM,SAAA,GAAY,EAAA,CAAG,OAAA,IAAW,EAAA,CAAG,GAAG,OAAA,IAAW,EAAA;AAGjD,YAAA,IACE,OAAO,GAAG,OAAA,KAAY,UAAA,IACtB,CAAC,EAAA,CAAG,iBAAA,GAAoB,UAAU,CAAA,EAClC;AACA,cAAA,MAAM,oBAAoB,EAAA,CAAG,OAAA;AAE7B,cAAA,EAAA,CAAG,OAAA,GAAU,YAAwB,WAAA,EAAoB;AACvD,gBAAA,MAAM,SAAA,GAAY,gBAAA,CAAiB,WAAA,CAAY,CAAC,CAAC,CAAA;AACjD,gBAAA,MAAM,SAAA,GAAY,SAAA,GACd,gBAAA,CAAiB,SAAS,CAAA,GAC1B,MAAA;AACJ,gBAAA,MAAM,WAAW,SAAA,GACb,CAAA,QAAA,EAAW,SAAA,CAAU,WAAA,EAAa,CAAA,CAAA,GAClC,eAAA;AAGJ,gBAAA,MAAM,IAAA,GAAO,MAAA,CAAO,SAAA,CAAU,QAAA,EAAU;AAAA,kBACtC,MAAMF,YAAA,CAAS;AAAA,iBAChB,CAAA;AACD,gBAAA,IAAA,CAAK,YAAA,CAAa,oBAAoB,QAAQ,CAAA;AAC9C,gBAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,IAAI,CAAA;AAExC,gBAAA,IAAI,SAAA,EAAW;AACb,kBAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,gBACpD;AAEA,gBAAA,IAAI,MAAA,EAAQ;AACV,kBAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,MAAM,CAAA;AAAA,gBAC5C;AAEA,gBAAA,IAAI,gBAAA,IAAoB,cAAc,MAAA,EAAW;AAC/C,kBAAA,MAAM,SAAA,GAAY,iBAAA;AAAA,oBAChB,SAAA;AAAA,oBACA;AAAA,mBACF;AACA,kBAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,gBACpD;AAEA,gBAAA,IAAI,QAAA,EAAU;AACZ,kBAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,gBACpD;AAEA,gBAAA,IAAI,QAAA,EAAU;AACZ,kBAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,gBACpD;AAGA,gBAAA,OAAOC,wBAAA,CAAY,MAAM,MAAM;AAC7B,kBAAA,IAAI;AACF,oBAAA,MAAM,MAAA,GAAS,iBAAA,CAAkB,KAAA,CAAM,IAAA,EAAM,WAAW,CAAA;AACxD,oBAAA,OAAO,QAAQ,OAAA,CAAQ,MAAM,CAAA,CAC1B,IAAA,CAAK,CAAC,KAAA,KAAU;AACf,sBAAAC,yBAAA,CAAa,IAAI,CAAA;AACjB,sBAAA,OAAO,KAAA;AAAA,oBACT,CAAC,CAAA,CACA,KAAA,CAAM,CAAC,KAAA,KAAU;AAChB,sBAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,sBAAA,MAAM,KAAA;AAAA,oBACR,CAAC,CAAA;AAAA,kBACL,SAAS,KAAA,EAAO;AACd,oBAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,oBAAA,MAAM,KAAA;AAAA,kBACR;AAAA,gBACF,CAAC,CAAA;AAAA,cACH,CAAA;AAEA,cAAA,EAAA,CAAG,iBAAA,GAAoB,UAAU,CAAA,GAAI,IAAA;AAAA,YACvC;AAGA,YAAA,IACE,OAAO,UAAU,YAAA,KAAiB,UAAA,IAClC,CAAC,SAAA,CAAU,iBAAA,GAAoB,KAAK,CAAA,EACpC;AACA,cAAA,MAAM,yBAAyB,SAAA,CAAU,YAAA;AAEzC,cAAA,SAAA,CAAU,YAAA,GAAe,YAAa,WAAA,EAAoB;AACxD,gBAAA,MAAM,WAAW,sBAAA,CAAuB,KAAA;AAAA,kBACtC,IAAA;AAAA,kBACA;AAAA,iBACF;AAGA,gBAAA,IAAI,QAAA,IAAY,OAAO,QAAA,CAAS,OAAA,KAAY,UAAA,EAAY;AACtD,kBAAA,MAAM,0BAA0B,QAAA,CAAS,OAAA;AAEzC,kBAAA,QAAA,CAAS,OAAA,GAAU,YAEd,WAAA,EACH;AAEA,oBAAA,MAAM,QAAA,GAAW,YAAY,CAAC,CAAA;AAC9B,oBAAA,MAAM,YACJ,QAAA,EAAU,GAAA,IACV,QAAA,EAAU,WAAA,IACV,iBAAiB,QAAQ,CAAA;AAC3B,oBAAA,MAAM,SAAA,GAAY,SAAA,GACd,gBAAA,CAAiB,SAAS,CAAA,GAC1B,MAAA;AACJ,oBAAA,MAAM,WAAW,SAAA,GACb,CAAA,QAAA,EAAW,SAAA,CAAU,WAAA,EAAa,CAAA,CAAA,GAClC,eAAA;AAGJ,oBAAA,MAAM,IAAA,GAAO,MAAA,CAAO,SAAA,CAAU,QAAA,EAAU;AAAA,sBACtC,MAAMF,YAAA,CAAS;AAAA,qBAChB,CAAA;AACD,oBAAA,IAAA,CAAK,YAAA,CAAa,oBAAoB,QAAQ,CAAA;AAC9C,oBAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,IAAI,CAAA;AAExC,oBAAA,IAAI,SAAA,EAAW;AACb,sBAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,oBACpD;AAEA,oBAAA,IAAI,MAAA,EAAQ;AACV,sBAAA,IAAA,CAAK,YAAA,CAAa,kBAAkB,MAAM,CAAA;AAAA,oBAC5C;AAEA,oBAAA,IAAI,gBAAA,IAAoB,cAAc,MAAA,EAAW;AAC/C,sBAAA,MAAM,SAAA,GAAY,iBAAA;AAAA,wBAChB,SAAA;AAAA,wBACA;AAAA,uBACF;AACA,sBAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,oBACpD;AAEA,oBAAA,IAAI,QAAA,EAAU;AACZ,sBAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,oBACpD;AAEA,oBAAA,IAAI,QAAA,EAAU;AACZ,sBAAA,IAAA,CAAK,YAAA,CAAa,wBAAwB,QAAQ,CAAA;AAAA,oBACpD;AAGA,oBAAA,OAAOC,wBAAA,CAAY,MAAM,MAAM;AAC7B,sBAAA,IAAI;AACF,wBAAA,MAAM,SAAS,uBAAA,CAAwB,KAAA;AAAA,0BACrC,IAAA;AAAA,0BACA;AAAA,yBACF;AACA,wBAAA,OAAO,QAAQ,OAAA,CAAQ,MAAM,CAAA,CAC1B,IAAA,CAAK,CAAC,KAAA,KAAU;AACf,0BAAAC,yBAAA,CAAa,IAAI,CAAA;AACjB,0BAAA,OAAO,KAAA;AAAA,wBACT,CAAC,CAAA,CACA,KAAA,CAAM,CAAC,KAAA,KAAU;AAChB,0BAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,0BAAA,MAAM,KAAA;AAAA,wBACR,CAAC,CAAA;AAAA,sBACL,SAAS,KAAA,EAAO;AACd,wBAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,wBAAA,MAAM,KAAA;AAAA,sBACR;AAAA,oBACF,CAAC,CAAA;AAAA,kBACH,CAAA;AAAA,gBACF;AAEA,gBAAA,OAAO,QAAA;AAAA,cACT,CAAA;AAEA,cAAA,SAAA,CAAU,iBAAA,GAAoB,KAAK,CAAA,GAAI,IAAA;AAAA,YACzC;AAAA,UACF;AAGA,UAAA,OAAO,oBAAoB,EAAE,CAAA;AAAA,QAC/B,CAAA;AAGA,QAAA,OAAO,OAAA,CAAQ,KAAA,CAAM,mBAAA,EAAqB,IAAA,EAAM;AAAA,UAC9C,eAAA;AAAA,UACA,GAAG;AAAA,SACJ,CAAA;AAAA,MACH,CAAA;AAEA,MAAA,OAAA,CAAQ,iBAAA,GAAoB,cAAc,CAAA,GAAI,IAAA;AAC9C,MAAA,YAAA,GAAe,IAAA;AAAA,IACjB;AAAA,EACF;AAEA,EAAA,IAAI,EAAA,CAAG,OAAA,IAAW,CAAC,YAAA,EAAc;AAC/B,IAAA,MAAM,SAAS,EAAA,CAAG,OAAA;AAElB,IAAA,IACE,OAAO,MAAA,CAAO,KAAA,KAAU,cACxB,OAAO,MAAA,CAAO,YAAY,UAAA,EAC1B;AACA,MAAA,iBAAA,CAAkB,QAAQ,MAAM,CAAA;AAChC,MAAA,YAAA,GAAe,IAAA;AAAA,IACjB;AAAA,EACF;AAGA,EAAA,IACE,EAAA,CAAG,CAAA,IACH,EAAA,CAAG,CAAA,CAAE,OAAA,IACL,OAAO,EAAA,CAAG,CAAA,CAAE,OAAA,CAAQ,OAAA,KAAY,UAAA,IAChC,CAAC,YAAA,EACD;AACA,IAAA,MAAM,OAAA,GAAU,GAAG,CAAA,CAAE,OAAA;AAGrB,IAAA,IAAI,OAAA,CAAQ,iBAAiB,CAAA,EAAG;AAC9B,MAAA,OAAO,EAAA;AAAA,IACT;AAEA,IAAA,MAAM;AAAA,MACJ,YAAAC,WAAAA,GAAa,mBAAA;AAAA,MACb,UAAAC,SAAAA,GAAW,iBAAA;AAAA,MACX,MAAA,EAAAC,OAAAA;AAAA,MACA,kBAAAC,iBAAAA,GAAmB,IAAA;AAAA,MACnB,oBAAAC,mBAAAA,GAAqB,GAAA;AAAA,MACrB,QAAA,EAAAC,SAAAA;AAAA,MACA,QAAA,EAAAC;AAAA,KACF,GAAI,UAAU,EAAC;AAEf,IAAA,MAAMC,OAAAA,GAASX,SAAA,CAAM,SAAA,CAAUI,WAAU,CAAA;AACzC,IAAA,MAAM,kBAAkB,OAAA,CAAQ,OAAA;AAEhC,IAAA,IAAI,CAAC,eAAA,EAAiB;AACpB,MAAA,OAAO,EAAA;AAAA,IACT;AAEA,IAAA,MAAM,mBAAA,GAAqC,SAASQ,aAAAA,CAAAA,GAE/C,IAAA,EACH;AAEA,MAAA,MAAM,SAAA,GAAY,gBAAA,CAAiB,IAAA,CAAK,CAAC,CAAC,CAAA;AAC1C,MAAA,MAAM,SAAA,GAAY,SAAA,GAAY,gBAAA,CAAiB,SAAS,CAAA,GAAI,MAAA;AAC5D,MAAA,MAAM,WAAW,SAAA,GACb,CAAA,QAAA,EAAW,SAAA,CAAU,WAAA,EAAa,CAAA,CAAA,GAClC,eAAA;AAGJ,MAAA,MAAM,IAAA,GAAOD,QAAO,SAAA,CAAU,QAAA,EAAU,EAAE,IAAA,EAAMV,YAAA,CAAS,QAAQ,CAAA;AACjE,MAAA,IAAA,CAAK,YAAA,CAAa,oBAAoBI,SAAQ,CAAA;AAE9C,MAAA,IAAI,SAAA,EAAW;AACb,QAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,MACpD;AAEA,MAAA,IAAIC,OAAAA,EAAQ;AACV,QAAA,IAAA,CAAK,YAAA,CAAa,kBAAkBA,OAAM,CAAA;AAAA,MAC5C;AAEA,MAAA,IAAIC,iBAAAA,IAAoB,cAAc,MAAA,EAAW;AAC/C,QAAA,MAAM,SAAA,GAAY,iBAAA,CAAkB,SAAA,EAAWC,mBAAkB,CAAA;AACjE,QAAA,IAAA,CAAK,YAAA,CAAa,uBAAuB,SAAS,CAAA;AAAA,MACpD;AAEA,MAAA,IAAIC,SAAAA,EAAU;AACZ,QAAA,IAAA,CAAK,YAAA,CAAa,wBAAwBA,SAAQ,CAAA;AAAA,MACpD;AAEA,MAAA,IAAIC,SAAAA,EAAU;AACZ,QAAA,IAAA,CAAK,YAAA,CAAa,wBAAwBA,SAAQ,CAAA;AAAA,MACpD;AAGA,MAAA,OAAOR,wBAAA,CAAY,MAAM,MAAM;AAC7B,QAAA,IAAI;AACF,UAAA,MAAM,MAAA,GAAS,eAAA,CAAgB,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AAC/C,UAAA,OAAO,QAAQ,OAAA,CAAQ,MAAM,CAAA,CAC1B,IAAA,CAAK,CAAC,KAAA,KAAU;AACf,YAAAC,yBAAA,CAAa,IAAI,CAAA;AACjB,YAAA,OAAO,KAAA;AAAA,UACT,CAAC,CAAA,CACA,KAAA,CAAM,CAAC,KAAA,KAAU;AAChB,YAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,YAAA,MAAM,KAAA;AAAA,UACR,CAAC,CAAA;AAAA,QACL,SAAS,KAAA,EAAO;AACd,UAAAA,yBAAA,CAAa,MAAM,KAAK,CAAA;AACxB,UAAA,MAAM,KAAA;AAAA,QACR;AAAA,MACF,CAAC,CAAA;AAAA,IACH,CAAA;AAEA,IAAA,OAAA,CAAQ,iBAAiB,CAAA,GAAI,IAAA;AAC7B,IAAA,OAAA,CAAQ,OAAA,GAAU,mBAAA;AAClB,IAAA,YAAA,GAAe,IAAA;AAAA,EACjB;AAGA,EAAA,IAAI,YAAA,EAAc;AAChB,IAAA,EAAA,CAAG,iBAAiB,CAAA,GAAI,IAAA;AAAA,EAC1B;AAEA,EAAA,OAAO,EAAA;AACT","file":"drizzle.cjs","sourcesContent":["/**\n * OpenTelemetry semantic conventions for database operations.\n * These constants are shared across all plugins.\n */\n\n// Common database attributes\nexport const SEMATTRS_DB_SYSTEM = 'db.system' as const;\nexport const SEMATTRS_DB_OPERATION = 'db.operation' as const;\nexport const SEMATTRS_DB_STATEMENT = 'db.statement' as const;\nexport const SEMATTRS_DB_NAME = 'db.name' as const;\n\n// MongoDB-specific attributes\nexport const SEMATTRS_DB_MONGODB_COLLECTION = 'db.mongodb.collection' as const;\n\n// Network attributes\nexport const SEMATTRS_NET_PEER_NAME = 'net.peer.name' as const;\nexport const SEMATTRS_NET_PEER_PORT = 'net.peer.port' as const;\n","/* eslint-disable @typescript-eslint/no-explicit-any */\n// Note: `any` types are necessary for dynamic instrumentation patterns\n// where we need to wrap arbitrary methods and preserve their signatures\nimport { SpanKind, trace } from '@opentelemetry/api';\nimport {\n  SEMATTRS_DB_SYSTEM,\n  SEMATTRS_DB_OPERATION,\n  SEMATTRS_DB_STATEMENT,\n  SEMATTRS_DB_NAME,\n  SEMATTRS_NET_PEER_NAME,\n  SEMATTRS_NET_PEER_PORT,\n} from '../common/constants';\nimport { runWithSpan, finalizeSpan } from 'autotel/trace-helpers';\n\nconst DEFAULT_TRACER_NAME = 'autotel-plugins/drizzle';\nconst DEFAULT_DB_SYSTEM = 'postgresql';\nconst INSTRUMENTED_FLAG = '__autotelDrizzleInstrumented' as const;\n\ntype QueryCallback = (error: unknown, result: unknown) => void;\n\ntype QueryFunction = (...args: any[]) => any;\n\ninterface DrizzleClientLike {\n  query?: QueryFunction;\n  execute?: QueryFunction;\n  [INSTRUMENTED_FLAG]?: true;\n  [key: string]: any; // Allow other properties\n}\n\n/**\n * Configuration options for Drizzle instrumentation.\n */\nexport interface InstrumentDrizzleConfig {\n  /**\n   * Custom tracer name. Defaults to \"autotel-plugins/drizzle\".\n   */\n  tracerName?: string;\n\n  /**\n   * Database system identifier (e.g., \"postgresql\", \"mysql\", \"sqlite\").\n   * Defaults to \"postgresql\".\n   */\n  dbSystem?: string;\n\n  /**\n   * Database name to include in spans.\n   */\n  dbName?: string;\n\n  /**\n   * Whether to capture full SQL query text in spans.\n   * Defaults to true.\n   */\n  captureQueryText?: boolean;\n\n  /**\n   * Maximum length for captured query text. Queries longer than this\n   * will be truncated. Defaults to 1000 characters.\n   */\n  maxQueryTextLength?: number;\n\n  /**\n   * Remote hostname or IP address of the database server.\n   * Example: \"db.example.com\" or \"192.168.1.100\"\n   */\n  peerName?: string;\n\n  /**\n   * Remote port number of the database server.\n   * Example: 5432 for PostgreSQL, 3306 for MySQL\n   */\n  peerPort?: number;\n}\n\n/**\n * Extracts SQL query text from various query argument formats.\n */\nfunction extractQueryText(queryArg: unknown): string | undefined {\n  if (typeof queryArg === 'string') {\n    return queryArg;\n  }\n  if (queryArg && typeof queryArg === 'object') {\n    // Generic SQL object format (used by LibSQL, MySQL, and others)\n    if (typeof (queryArg as { sql?: unknown }).sql === 'string') {\n      return (queryArg as { sql: string }).sql;\n    }\n    // PostgreSQL-style query object\n    if (typeof (queryArg as { text?: unknown }).text === 'string') {\n      return (queryArg as { text: string }).text;\n    }\n    // Drizzle SQL object\n    if (\n      typeof (queryArg as { queryChunks?: unknown }).queryChunks === 'object'\n    ) {\n      // Drizzle query objects may have complex structure, try to extract meaningful info\n      const drizzleQuery = queryArg as Record<string, unknown>;\n      if (typeof drizzleQuery.sql === 'string') {\n        return drizzleQuery.sql;\n      }\n    }\n  }\n  return undefined;\n}\n\n/**\n * Sanitizes and truncates query text for safe inclusion in spans.\n */\nfunction sanitizeQueryText(queryText: string, maxLength: number): string {\n  if (queryText.length <= maxLength) {\n    return queryText;\n  }\n  return `${queryText.slice(0, Math.max(0, maxLength))}...`;\n}\n\n/**\n * Extracts the SQL operation (SELECT, INSERT, etc.) from query text.\n */\nfunction extractOperation(queryText: string): string | undefined {\n  const trimmed = queryText.trimStart();\n  const match = /^(?<op>\\w+)/u.exec(trimmed);\n  return match?.groups?.op?.toUpperCase();\n}\n\n/**\n * Instruments a database connection pool/client with OpenTelemetry tracing.\n *\n * This function wraps the connection's `query` and `execute` methods to create spans for each database\n * operation.\n * The instrumentation is idempotent - calling it multiple times on the same connection will only\n * instrument it once.\n *\n * @typeParam TClient - The type of the database connection pool or client\n * @param client - The database connection pool or client to instrument\n * @param config - Optional configuration for instrumentation behavior\n * @returns The instrumented pool/client (same instance, modified in place)\n *\n * @example\n * ```typescript\n * // PostgreSQL with node-postgres\n * import { drizzle } from 'drizzle-orm/node-postgres';\n * import { Pool } from 'pg';\n * import { instrumentDrizzle } from 'autotel-plugins/drizzle';\n *\n * const pool = new Pool({ connectionString: process.env.DATABASE_URL });\n * const instrumentedPool = instrumentDrizzle(pool, {\n *   dbSystem: 'postgresql',\n *   dbName: 'myapp',\n *   peerName: 'db.example.com',\n *   peerPort: 5432,\n * });\n * const db = drizzle({ client: instrumentedPool });\n * ```\n *\n * @example\n * ```typescript\n * // MySQL with mysql2\n * import { drizzle } from 'drizzle-orm/mysql2';\n * import mysql from 'mysql2/promise';\n * import { instrumentDrizzle } from 'autotel-plugins/drizzle';\n *\n * const connection = await mysql.createConnection({\n *   host: 'localhost',\n *   user: 'root',\n *   database: 'mydb',\n * });\n * const instrumentedConnection = instrumentDrizzle(connection, { dbSystem: 'mysql' });\n * const db = drizzle({ client: instrumentedConnection });\n * ```\n *\n * @example\n * ```typescript\n * // SQLite with better-sqlite3\n * import { drizzle } from 'drizzle-orm/better-sqlite3';\n * import Database from 'better-sqlite3';\n * import { instrumentDrizzle } from 'autotel-plugins/drizzle';\n *\n * const sqlite = new Database('sqlite.db');\n * const instrumentedSqlite = instrumentDrizzle(sqlite, { dbSystem: 'sqlite' });\n * const db = drizzle({ client: instrumentedSqlite });\n * ```\n *\n * @example\n * ```typescript\n * // LibSQL/Turso\n * import { drizzle } from 'drizzle-orm/libsql';\n * import { createClient } from '@libsql/client';\n * import { instrumentDrizzle } from 'autotel-plugins/drizzle';\n *\n * const client = createClient({\n *   url: process.env.DATABASE_URL!,\n *   authToken: process.env.DATABASE_AUTH_TOKEN,\n * });\n * const instrumentedClient = instrumentDrizzle(client, { dbSystem: 'sqlite' });\n * const db = drizzle({ client: instrumentedClient });\n * ```\n */\nexport function instrumentDrizzle<TClient extends DrizzleClientLike>(\n  client: TClient,\n  config?: InstrumentDrizzleConfig,\n): TClient {\n  if (!client) {\n    return client;\n  }\n\n  // Check if client has query or execute method\n  const hasQuery = typeof client.query === 'function';\n  const hasExecute = typeof client.execute === 'function';\n\n  if (!hasQuery && !hasExecute) {\n    return client;\n  }\n\n  if (client[INSTRUMENTED_FLAG]) {\n    return client;\n  }\n\n  const {\n    tracerName = DEFAULT_TRACER_NAME,\n    dbSystem = DEFAULT_DB_SYSTEM,\n    dbName,\n    captureQueryText = true,\n    maxQueryTextLength = 1000,\n    peerName,\n    peerPort,\n  } = config ?? {};\n\n  const tracer = trace.getTracer(tracerName);\n\n  // Store the original method (query or execute)\n  const originalMethod = hasQuery ? client.query : client.execute;\n\n  if (!originalMethod) {\n    return client;\n  }\n\n  const instrumentedMethod: QueryFunction = function instrumented(\n    this: any,\n    ...incomingArgs: any[]\n  ) {\n    const args = [...incomingArgs];\n    let callback: QueryCallback | undefined;\n\n    // Detect callback pattern\n    if (typeof args.at(-1) === 'function') {\n      callback = args.pop() as QueryCallback;\n    }\n\n    // Extract query information\n    const queryText = extractQueryText(args[0]);\n    const operation = queryText ? extractOperation(queryText) : undefined;\n    const spanName = operation\n      ? `drizzle.${operation.toLowerCase()}`\n      : 'drizzle.query';\n\n    // Start span\n    const span = tracer.startSpan(spanName, { kind: SpanKind.CLIENT });\n    span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);\n\n    if (operation) {\n      span.setAttribute(SEMATTRS_DB_OPERATION, operation);\n    }\n\n    if (dbName) {\n      span.setAttribute(SEMATTRS_DB_NAME, dbName);\n    }\n\n    if (captureQueryText && queryText !== undefined) {\n      const sanitized = sanitizeQueryText(queryText, maxQueryTextLength);\n      span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);\n    }\n\n    if (peerName) {\n      span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);\n    }\n\n    if (peerPort) {\n      span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);\n    }\n\n    // Callback-based pattern\n    if (callback) {\n      return runWithSpan(span, () => {\n        const wrappedCallback: QueryCallback = (err, result) => {\n          finalizeSpan(span, err);\n          if (callback) {\n            callback(err, result);\n          }\n        };\n\n        try {\n          return Reflect.apply(originalMethod, this, [\n            ...args,\n            wrappedCallback,\n          ]);\n        } catch (error) {\n          finalizeSpan(span, error);\n          throw error;\n        }\n      });\n    }\n\n    // Promise-based pattern\n    return runWithSpan(span, () => {\n      try {\n        const result = originalMethod.apply(this, args);\n        return Promise.resolve(result)\n          .then((value) => {\n            finalizeSpan(span);\n            return value;\n          })\n          .catch((error) => {\n            finalizeSpan(span, error);\n            throw error;\n          });\n      } catch (error) {\n        finalizeSpan(span, error);\n        throw error;\n      }\n    });\n  };\n\n  client[INSTRUMENTED_FLAG] = true;\n\n  // Replace the original method with the instrumented one\n  if (hasQuery) {\n    client.query = instrumentedMethod;\n  } else {\n    client.execute = instrumentedMethod;\n  }\n\n  return client;\n}\n\n/**\n * Interface for Drizzle database instances with minimal type requirements.\n */\ninterface DrizzleDbLike {\n  $client?: DrizzleClientLike | any; // Allow any client type\n  execute?: QueryFunction; // Direct execute method on db\n  transaction?: QueryFunction; // Transaction method on db\n  _?: {\n    session?: {\n      execute?: QueryFunction;\n      [INSTRUMENTED_FLAG]?: true;\n      [key: string]: any;\n    };\n    [key: string]: any;\n  };\n  [INSTRUMENTED_FLAG]?: true;\n  [key: string]: any; // Allow other properties\n}\n\n/**\n * Instruments a Drizzle database instance with OpenTelemetry tracing.\n *\n * This function instruments the database at the session level, automatically tracing all database\n * operations including query builders, direct SQL execution, and transactions.\n *\n * The instrumentation is idempotent - calling it multiple times on the same\n * database will only instrument it once.\n *\n * @typeParam TDb - The type of the Drizzle database instance\n * @param db - The Drizzle database instance to instrument\n * @param config - Optional configuration for instrumentation behavior\n * @returns The instrumented database instance (same instance, modified in place)\n *\n * @example\n * ```typescript\n * // PostgreSQL with postgres.js\n * import { drizzle } from 'drizzle-orm/postgres-js';\n * import postgres from 'postgres';\n * import { instrumentDrizzleClient } from 'autotel-plugins/drizzle';\n *\n * // Using connection string\n * const db = drizzle(process.env.DATABASE_URL!);\n * instrumentDrizzleClient(db, { dbSystem: 'postgresql' });\n *\n * // Or with a client instance\n * const queryClient = postgres(process.env.DATABASE_URL!);\n * const db = drizzle({ client: queryClient });\n * instrumentDrizzleClient(db, { dbSystem: 'postgresql' });\n * ```\n *\n * @example\n * ```typescript\n * // PostgreSQL with node-postgres (pg)\n * import { drizzle } from 'drizzle-orm/node-postgres';\n * import { Pool } from 'pg';\n * import { instrumentDrizzleClient } from 'autotel-plugins/drizzle';\n *\n * // Using connection string\n * const db = drizzle(process.env.DATABASE_URL!);\n * instrumentDrizzleClient(db, { dbSystem: 'postgresql' });\n *\n * // Or with a pool\n * const pool = new Pool({ connectionString: process.env.DATABASE_URL });\n * const db = drizzle({ client: pool });\n * instrumentDrizzleClient(db, {\n *   dbSystem: 'postgresql',\n *   dbName: 'myapp',\n *   peerName: 'db.example.com',\n *   peerPort: 5432,\n * });\n * ```\n */\nexport function instrumentDrizzleClient<TDb extends DrizzleDbLike>(\n  db: TDb,\n  config?: InstrumentDrizzleConfig,\n): TDb {\n  if (!db) {\n    return db;\n  }\n\n  // Check if already instrumented\n  if (db[INSTRUMENTED_FLAG]) {\n    return db;\n  }\n\n  const {\n    tracerName = DEFAULT_TRACER_NAME,\n    dbSystem = DEFAULT_DB_SYSTEM,\n    dbName,\n    captureQueryText = true,\n    maxQueryTextLength = 1000,\n    peerName,\n    peerPort,\n  } = config ?? {};\n\n  const tracer = trace.getTracer(tracerName);\n  let instrumented = false;\n\n  // First priority: Instrument the session directly\n  // This is where all queries actually go through\n  if ((db as any).session && !instrumented) {\n    const session = (db as any).session;\n\n    // Check if session has prepareQuery method (used by select/insert/update/delete)\n    if (\n      typeof session.prepareQuery === 'function' &&\n      !session[INSTRUMENTED_FLAG]\n    ) {\n      const originalPrepareQuery = session.prepareQuery;\n\n      session.prepareQuery = function (...args: any[]) {\n        const prepared = originalPrepareQuery.apply(this, args);\n\n        // Wrap the prepared query's execute method\n        if (prepared && typeof prepared.execute === 'function') {\n          const originalPreparedExecute = prepared.execute;\n\n          prepared.execute = function (this: any, ...executeArgs: any[]) {\n            // Extract query information from the query object\n            const queryObj = args[0]; // The query object passed to prepareQuery\n            const queryText =\n              queryObj?.sql ||\n              queryObj?.queryString ||\n              extractQueryText(queryObj);\n            const operation = queryText\n              ? extractOperation(queryText)\n              : undefined;\n            const spanName = operation\n              ? `drizzle.${operation.toLowerCase()}`\n              : 'drizzle.query';\n\n            // Start span\n            const span = tracer.startSpan(spanName, { kind: SpanKind.CLIENT });\n            span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);\n\n            if (operation) {\n              span.setAttribute(SEMATTRS_DB_OPERATION, operation);\n            }\n\n            if (dbName) {\n              span.setAttribute(SEMATTRS_DB_NAME, dbName);\n            }\n\n            if (captureQueryText && queryText !== undefined) {\n              const sanitized = sanitizeQueryText(\n                queryText,\n                maxQueryTextLength,\n              );\n              span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);\n            }\n\n            if (peerName) {\n              span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);\n            }\n\n            if (peerPort) {\n              span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);\n            }\n\n            // Execute the prepared query\n            return runWithSpan(span, () => {\n              try {\n                const result = originalPreparedExecute.apply(this, executeArgs);\n                return Promise.resolve(result)\n                  .then((value) => {\n                    finalizeSpan(span);\n                    return value;\n                  })\n                  .catch((error) => {\n                    finalizeSpan(span, error);\n                    throw error;\n                  });\n              } catch (error) {\n                finalizeSpan(span, error);\n                throw error;\n              }\n            });\n          };\n        }\n\n        return prepared;\n      };\n\n      session[INSTRUMENTED_FLAG] = true;\n      instrumented = true;\n    }\n\n    // Also instrument direct query method if exists\n    if (\n      typeof session.query === 'function' &&\n      !session[INSTRUMENTED_FLAG + '_query']\n    ) {\n      const originalQuery = session.query;\n\n      session.query = function (this: any, queryString: string, params: any[]) {\n        const operation = queryString\n          ? extractOperation(queryString)\n          : undefined;\n        const spanName = operation\n          ? `drizzle.${operation.toLowerCase()}`\n          : 'drizzle.query';\n\n        // Start span\n        const span = tracer.startSpan(spanName, { kind: SpanKind.CLIENT });\n        span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);\n\n        if (operation) {\n          span.setAttribute(SEMATTRS_DB_OPERATION, operation);\n        }\n\n        if (dbName) {\n          span.setAttribute(SEMATTRS_DB_NAME, dbName);\n        }\n\n        if (captureQueryText && queryString !== undefined) {\n          const sanitized = sanitizeQueryText(queryString, maxQueryTextLength);\n          span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);\n        }\n\n        if (peerName) {\n          span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);\n        }\n\n        if (peerPort) {\n          span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);\n        }\n\n        // Execute the query\n        return runWithSpan(span, () => {\n          try {\n            const result = Reflect.apply(originalQuery, this, [\n              queryString,\n              params,\n            ]);\n            return Promise.resolve(result)\n              .then((value) => {\n                finalizeSpan(span);\n                return value;\n              })\n              .catch((error) => {\n                finalizeSpan(span, error);\n                throw error;\n              });\n          } catch (error) {\n            finalizeSpan(span, error);\n            throw error;\n          }\n        });\n      };\n\n      session[INSTRUMENTED_FLAG + '_query'] = true;\n      instrumented = true;\n    }\n\n    // Instrument transaction method to ensure transaction sessions are also instrumented\n    if (\n      typeof session.transaction === 'function' &&\n      !session[INSTRUMENTED_FLAG + '_transaction']\n    ) {\n      const originalTransaction = session.transaction;\n\n      session.transaction = function (\n        this: any,\n        transactionCallback: any,\n        ...restArgs: any[]\n      ) {\n        // Wrap the transaction callback to instrument the tx object\n        const wrappedCallback = async function (tx: any) {\n          // Instrument the transaction's session if it has one\n          if (tx && (tx.session || tx._?.session || tx)) {\n            const txSession = tx.session || tx._?.session || tx;\n\n            // Instrument tx.execute if it exists\n            if (\n              typeof tx.execute === 'function' &&\n              !tx[INSTRUMENTED_FLAG + '_execute']\n            ) {\n              const originalTxExecute = tx.execute;\n\n              tx.execute = function (this: any, ...executeArgs: any[]) {\n                const queryText = extractQueryText(executeArgs[0]);\n                const operation = queryText\n                  ? extractOperation(queryText)\n                  : undefined;\n                const spanName = operation\n                  ? `drizzle.${operation.toLowerCase()}`\n                  : 'drizzle.query';\n\n                // Start span\n                const span = tracer.startSpan(spanName, {\n                  kind: SpanKind.CLIENT,\n                });\n                span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);\n                span.setAttribute('db.transaction', true);\n\n                if (operation) {\n                  span.setAttribute(SEMATTRS_DB_OPERATION, operation);\n                }\n\n                if (dbName) {\n                  span.setAttribute(SEMATTRS_DB_NAME, dbName);\n                }\n\n                if (captureQueryText && queryText !== undefined) {\n                  const sanitized = sanitizeQueryText(\n                    queryText,\n                    maxQueryTextLength,\n                  );\n                  span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);\n                }\n\n                if (peerName) {\n                  span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);\n                }\n\n                if (peerPort) {\n                  span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);\n                }\n\n                // Execute the query\n                return runWithSpan(span, () => {\n                  try {\n                    const result = originalTxExecute.apply(this, executeArgs);\n                    return Promise.resolve(result)\n                      .then((value) => {\n                        finalizeSpan(span);\n                        return value;\n                      })\n                      .catch((error) => {\n                        finalizeSpan(span, error);\n                        throw error;\n                      });\n                  } catch (error) {\n                    finalizeSpan(span, error);\n                    throw error;\n                  }\n                });\n              };\n\n              tx[INSTRUMENTED_FLAG + '_execute'] = true;\n            }\n\n            // Also instrument txSession.prepareQuery if it exists\n            if (\n              typeof txSession.prepareQuery === 'function' &&\n              !txSession[INSTRUMENTED_FLAG + '_tx']\n            ) {\n              const originalTxPrepareQuery = txSession.prepareQuery;\n\n              txSession.prepareQuery = function (...prepareArgs: any[]) {\n                const prepared = originalTxPrepareQuery.apply(\n                  this,\n                  prepareArgs,\n                );\n\n                // Wrap the prepared query's execute method\n                if (prepared && typeof prepared.execute === 'function') {\n                  const originalPreparedExecute = prepared.execute;\n\n                  prepared.execute = function (\n                    this: any,\n                    ...executeArgs: any[]\n                  ) {\n                    // Extract query information from the query object\n                    const queryObj = prepareArgs[0]; // The query object passed to prepareQuery\n                    const queryText =\n                      queryObj?.sql ||\n                      queryObj?.queryString ||\n                      extractQueryText(queryObj);\n                    const operation = queryText\n                      ? extractOperation(queryText)\n                      : undefined;\n                    const spanName = operation\n                      ? `drizzle.${operation.toLowerCase()}`\n                      : 'drizzle.query';\n\n                    // Start span\n                    const span = tracer.startSpan(spanName, {\n                      kind: SpanKind.CLIENT,\n                    });\n                    span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);\n                    span.setAttribute('db.transaction', true);\n\n                    if (operation) {\n                      span.setAttribute(SEMATTRS_DB_OPERATION, operation);\n                    }\n\n                    if (dbName) {\n                      span.setAttribute(SEMATTRS_DB_NAME, dbName);\n                    }\n\n                    if (captureQueryText && queryText !== undefined) {\n                      const sanitized = sanitizeQueryText(\n                        queryText,\n                        maxQueryTextLength,\n                      );\n                      span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);\n                    }\n\n                    if (peerName) {\n                      span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);\n                    }\n\n                    if (peerPort) {\n                      span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);\n                    }\n\n                    // Execute the prepared query\n                    return runWithSpan(span, () => {\n                      try {\n                        const result = originalPreparedExecute.apply(\n                          this,\n                          executeArgs,\n                        );\n                        return Promise.resolve(result)\n                          .then((value) => {\n                            finalizeSpan(span);\n                            return value;\n                          })\n                          .catch((error) => {\n                            finalizeSpan(span, error);\n                            throw error;\n                          });\n                      } catch (error) {\n                        finalizeSpan(span, error);\n                        throw error;\n                      }\n                    });\n                  };\n                }\n\n                return prepared;\n              };\n\n              txSession[INSTRUMENTED_FLAG + '_tx'] = true;\n            }\n          }\n\n          // Call the original callback with the instrumented tx\n          return transactionCallback(tx);\n        };\n\n        // Call the original transaction with the wrapped callback\n        return Reflect.apply(originalTransaction, this, [\n          wrappedCallback,\n          ...restArgs,\n        ]);\n      };\n\n      session[INSTRUMENTED_FLAG + '_transaction'] = true;\n      instrumented = true;\n    }\n  }\n\n  if (db.$client && !instrumented) {\n    const client = db.$client;\n    // Check if client has query or execute function\n    if (\n      typeof client.query === 'function' ||\n      typeof client.execute === 'function'\n    ) {\n      instrumentDrizzle(client, config);\n      instrumented = true;\n    }\n  }\n\n  // Third priority: Try to instrument via session.execute as fallback\n  if (\n    db._ &&\n    db._.session &&\n    typeof db._.session.execute === 'function' &&\n    !instrumented\n  ) {\n    const session = db._.session;\n\n    // Check if already instrumented\n    if (session[INSTRUMENTED_FLAG]) {\n      return db;\n    }\n\n    const {\n      tracerName = DEFAULT_TRACER_NAME,\n      dbSystem = DEFAULT_DB_SYSTEM,\n      dbName,\n      captureQueryText = true,\n      maxQueryTextLength = 1000,\n      peerName,\n      peerPort,\n    } = config ?? {};\n\n    const tracer = trace.getTracer(tracerName);\n    const originalExecute = session.execute;\n\n    if (!originalExecute) {\n      return db;\n    }\n\n    const instrumentedExecute: QueryFunction = function instrumented(\n      this: any,\n      ...args: any[]\n    ) {\n      // Extract query information\n      const queryText = extractQueryText(args[0]);\n      const operation = queryText ? extractOperation(queryText) : undefined;\n      const spanName = operation\n        ? `drizzle.${operation.toLowerCase()}`\n        : 'drizzle.query';\n\n      // Start span\n      const span = tracer.startSpan(spanName, { kind: SpanKind.CLIENT });\n      span.setAttribute(SEMATTRS_DB_SYSTEM, dbSystem);\n\n      if (operation) {\n        span.setAttribute(SEMATTRS_DB_OPERATION, operation);\n      }\n\n      if (dbName) {\n        span.setAttribute(SEMATTRS_DB_NAME, dbName);\n      }\n\n      if (captureQueryText && queryText !== undefined) {\n        const sanitized = sanitizeQueryText(queryText, maxQueryTextLength);\n        span.setAttribute(SEMATTRS_DB_STATEMENT, sanitized);\n      }\n\n      if (peerName) {\n        span.setAttribute(SEMATTRS_NET_PEER_NAME, peerName);\n      }\n\n      if (peerPort) {\n        span.setAttribute(SEMATTRS_NET_PEER_PORT, peerPort);\n      }\n\n      // Promise-based pattern (session.execute is typically promise-based)\n      return runWithSpan(span, () => {\n        try {\n          const result = originalExecute.apply(this, args);\n          return Promise.resolve(result)\n            .then((value) => {\n              finalizeSpan(span);\n              return value;\n            })\n            .catch((error) => {\n              finalizeSpan(span, error);\n              throw error;\n            });\n        } catch (error) {\n          finalizeSpan(span, error);\n          throw error;\n        }\n      });\n    };\n\n    session[INSTRUMENTED_FLAG] = true;\n    session.execute = instrumentedExecute;\n    instrumented = true;\n  }\n\n  // Mark the db as instrumented if we instrumented anything\n  if (instrumented) {\n    db[INSTRUMENTED_FLAG] = true;\n  }\n\n  return db;\n}\n"]}