push some stuff

packages/datadog-instrumentations/src/passport.js

@@ -1,7 +1,7 @@
 'use strict'
 
 const shimmer = require('../../datadog-shimmer')
-const { addHook } = require('./helpers/instrument')
+const { channel, addHook } = require('./helpers/instrument')
 
 /* TODO: test with:
 passport-jwt JWTs
@@ -17,15 +17,17 @@
 koa-passport
 */
 
-function wrapDone (done) {
+const onPassportDeserializeUserChannel = channel('datadog:passport:deserializeUser:finish')
+
+function wrapDone (req, done) {
   // eslint-disable-next-line n/handle-callback-err
   return function wrappedDone (err, user) {
     if (user) {
       const abortController = new AbortController()
 
       // express-session middleware sets req.sessionID, it's required to use passport sessions anyway so might as well use it ?
       // what if session IDs are using rolling sessions or always changing or something idk ?
-      channel.publish({ req, user, sessionId: req.sessionID, abortController })
+      onPassportDeserializeUserChannel.publish({ req, user, sessionId: req.sessionID, abortController })
 
       if (abortController.signal.aborted) return
     }
@@ -48,7 +50,7 @@
   }
 }
 
 
 const { block } = require('../../dd-trace/src/appsec/blocking')
 const { getRootSpan } = require('../../dd-trace/src/appsec/sdk/utils')
 
@@ -68,9 +70,9 @@
       strategy._verify
 
       return function wrappedMiddleware (req, res, next) {
         return middleware(req, res, function wrappedNext (err) {
           console.log('NEW', req.user)
           if (req.user?.name === 'bitch') {
 
             return block(req, res, getRootSpan(global._ddtrace))
           }

packages/dd-trace/src/appsec/user_tracking.js

@@ -194,5 +194,6 @@ function trackUser (user, abortController) {
 
 module.exports = {
   setCollectionMode,
-  trackLogin
+  trackLogin,
+  trackUser
 }