From d718df88cd6477b5f6e6744c2a69af2d37ab2536 Mon Sep 17 00:00:00 2001
From: Ben Wilson <ben@registrationmax.com>
Date: Tue, 1 Aug 2023 16:46:22 -0400
Subject: [PATCH] refine user agent

---
 src/models/stratum-messages/SubscriptionMessage.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/models/stratum-messages/SubscriptionMessage.ts b/src/models/stratum-messages/SubscriptionMessage.ts
index 994cf07..5402372 100644
--- a/src/models/stratum-messages/SubscriptionMessage.ts
+++ b/src/models/stratum-messages/SubscriptionMessage.ts
@@ -14,7 +14,7 @@ export class SubscriptionMessage extends StratumBaseMessage {
     @IsString()
     @MaxLength(128)
     @Transform(({ value, key, obj, type }) => {
-        return obj.params[0] == null ? 'default' : obj.params[0]
+        return obj.params[0] == null ? 'unknown' : SubscriptionMessage.refineUserAgent(obj.params);
     })
     public userAgent: string;
 
@@ -38,4 +38,15 @@ export class SubscriptionMessage extends StratumBaseMessage {
 
 
     }
+
+    public static refineUserAgent(userAgent: string): string {
+        userAgent = userAgent.split(' ')[0].split('/')[0].split('V')[0];
+
+        if (userAgent.includes('bosminer')) {
+            userAgent = 'Braiins OS';
+        } else if (userAgent.includes('cpuminer')) {
+            userAgent = 'cpuminer';
+        }
+        return userAgent;
+    }
 }
\ No newline at end of file