✨ feat: Add server useObfs hook

src/helper/handlers.ts

@@ -1,5 +1,6 @@
 import { AuthMethod } from './authMethod'
 import Connection from './connection'
+import { ObfsBuilder } from '../obfs/obfs'
 
 interface Req {
   connect: (connection: Connection) => void
@@ -15,13 +16,16 @@ export class Handlers {
   public userId: (userId: string) => boolean
   public auth: AuthMethod[]
   public req: Req
+  public obfs: ObfsBuilder[]
   constructor(
     req: Req,
     auth: AuthMethod[] = [],
+    obfs: ObfsBuilder[] = [],
     userId: (userId: string) => boolean = () => true
   ) {
     this.req = req
     this.auth = auth
+    this.obfs = obfs
     this.userId = userId
   }
 }

src/server/server.ts

@@ -5,6 +5,7 @@ import { Handlers } from '../helper/handlers'
 import { AuthMethod } from '../helper/authMethod'
 import { connect, associate, bind } from './handlers'
 import Event from '../helper/event'
+import { ObfsBuilder } from '../obfs/obfs'
 
 type ConnectionListener = ((socket: net.Socket) => void) | undefined
 
@@ -104,6 +105,11 @@ export class Server {
     return this
   }
 
+  public useObfs(method: ObfsBuilder): Server {
+    this.handlers.obfs.push(method)
+    return this
+  }
+
   /**
    * Get the primary's server handle, and listen on it
    * @param port - Port to listen on

src/server/state/socks5.ts

@@ -11,6 +11,9 @@ export class ObfsState extends State {
   private obfsMethods = [none(), http()]
 
   parse(): void {
+    if (this.context.handlers.obfs.length > 0) {
+      this.obfsMethods = this.context.handlers.obfs
+    }
     const message = this.context.cat()
     for (const m of this.obfsMethods) {
       const method = m(this.context, ObfsMethod.SERVER)