refactor: clean unused code in core thread, add new hook

Author: GrinZero

packages/network-debugger/src/core/dc.ts

@@ -1,3 +1,6 @@
+/**
+ * @deprecated information is not complete
+ */
 import { ClientRequest, IncomingMessage } from 'http'
 import { MainProcess } from './fork'
 import * as dc from 'node:diagnostics_channel'

packages/network-debugger/src/core/fork.ts

@@ -118,41 +118,21 @@ export class MainProcess {
   }
 
   public sendRequest(type: RequestType, request: RequestDetail) {
-    this.send({
-      type,
-      data: request
-    })
-    return this
-  }
-
-  public initRequest(request: RequestDetail) {
-    this.sendRequest('initRequest', request)
-    return this
-  }
-
-  public registerRequest(request: RequestDetail) {
     const currentCell = getCurrentCell()
     let req = request
     if (currentCell) {
       currentCell.request = req
-      const pipes = currentCell.pipes.filter((p) => p.type === 'regsiter').map((p) => p.pipe)
+      const pipes = currentCell.pipes.filter((p) => p.type === type).map((p) => p.pipe)
       pipes.forEach((pipe) => {
         req = pipe(req)
       })
       currentCell.request = req
     }
 
-    this.sendRequest('registerRequest', req)
-    return this
-  }
-
-  public updateRequest(request: RequestDetail) {
-    this.sendRequest('updateRequest', request)
-    return this
-  }
-
-  public endRequest(request: RequestDetail) {
-    this.sendRequest('endRequest', request)
+    this.send({
+      type,
+      data: request
+    })
     return this
   }
 

packages/network-debugger/src/core/hooks/cell.ts

@@ -1,4 +1,5 @@
 import { type RequestDetail } from '../../common'
+import { RequestType } from '../fork'
 
 export interface Pipe<T = RequestDetail> {
   (req: T): T
@@ -8,7 +9,7 @@ export interface Cell {
   request: RequestDetail
   pipes: Array<{
     pipe: Pipe<RequestDetail>
-    type: 'regsiter'
+    type: RequestType
   }>
   /**
    * @default false

packages/network-debugger/src/core/hooks/index.ts

@@ -1,3 +1,4 @@
 export * from './useRegisterRequest'
+export * from './useRequestPipe'
 export * from './useAbortRequest'
 export * from './cell'

packages/network-debugger/src/core/hooks/useRegisterRequest.ts

@@ -14,7 +14,7 @@ export const useRegisterRequest: RegisterRequestHook = (pipe) => {
 
   cell.pipes.push({
     pipe,
-    type: 'regsiter'
+    type: 'registerRequest'
   })
 
   return

packages/network-debugger/src/core/hooks/useRequestPipe.ts

@@ -0,0 +1,22 @@
+import { RequestDetail } from '../../common'
+import { RequestType } from '../fork'
+import { getCurrentCell } from './cell'
+
+export interface RequestPipeHook {
+  (type: RequestType, pipe: (req: RequestDetail) => RequestDetail): void
+}
+
+export const useRequestPipe: RequestPipeHook = (type, pipe) => {
+  const cell = getCurrentCell()
+
+  if (!cell) {
+    throw new Error('useRequestPipe must be used in request handler')
+  }
+
+  cell.pipes.push({
+    pipe,
+    type
+  })
+
+  return
+}