@playpilot/tpi 8.40.0-beta.4 → 8.40.0-beta.5

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@playpilot/tpi",
-	"version": "8.40.0-beta.4",
+	"version": "8.40.0-beta.5",
 	"type": "module",
 	"scripts": {
 		"dev": "vite dev",

package/src/lib/injection.ts

@@ -30,39 +30,15 @@ export function injectLinksInDocument(elements: HTMLElement[], injections: LinkI
 
   const validInjections = filterInvalidInTextInjections(mergedInjections)
   const foundInjections = validInjections.filter(i => fullText.includes(cleanPhrase(i.sentence)))
-
-  const failedMessages: Record<string, string> = {}
-  const aiInjectionSidsCount: Record<string, number> = {}
-  let aiParticipantInjectionsCount = 0
+  const { injections: limitedInjections, failedMessages }  = limitInjectionsByCount(foundInjections)
 
   let injectionIndex = -1 // This index is used in Option 3 below.
-  for (const injection of sortInjections(foundInjections)) {
+  for (const injection of sortInjections(limitedInjections)) {
     injectionIndex++
 
-    if (!injection.manual) {
-      const previousCount = aiInjectionSidsCount[injection.sid] || 0
-
-      const exceedsUniqueCount =
-        window.PlayPilotLinkInjections?.config?.unique_injection_count_limit &&
-        previousCount >= window.PlayPilotLinkInjections?.config?.unique_injection_count_limit
-
-      const exceedsTotalParticipantsCount =
-        injection.type === 'participant' &&
-        window.PlayPilotLinkInjections?.config?.total_participant_injection_count_limit &&
-        aiParticipantInjectionsCount >= window.PlayPilotLinkInjections?.config?.total_participant_injection_count_limit
-
-      if (exceedsUniqueCount || exceedsTotalParticipantsCount) {
-        failedMessages[injection.key] = 'Injection was withheld because it exceeded the maximum unique count.'
-        continue
-      } else {
-        aiInjectionSidsCount[injection.sid] = previousCount + 1
-        if (injection.type === 'participant') aiParticipantInjectionsCount++
-      }
-
-      if (injection.type === 'participant' && !doesParticipantMeetThreshold(mergedInjections, injection.participant_details!)) {
-        failedMessages[injection.key] = 'Participant does not meet popularity threshold.'
-        continue
-      }
+    if (!injection.manual && injection.type === 'participant' && !doesParticipantMeetThreshold(mergedInjections, injection.participant_details!)) {
+      failedMessages[injection.key] = 'Participant does not meet popularity threshold.'
+      continue
     }
 
     const elementIndex = elements.findIndex(element => cleanPhrase(element.innerText).includes(cleanPhrase(injection.sentence)))
@@ -468,6 +444,41 @@ export function filterRemovedAndInactiveInjections(injections: LinkInjection[]):
   })
 }
 
+export function limitInjectionsByCount(injections: LinkInjection[]): { injections: LinkInjection[], failedMessages: Record<string, string> } {
+  const limitedInjections: LinkInjection[] = []
+  const failedMessages: Record<string, string> = {}
+
+  const aiInjectionSidsCount: Record<string, number> = {}
+  let aiParticipantInjectionsCount = 0
+
+  for (const injection of injections) {
+    if (!injection.manual) {
+      const previousCount = aiInjectionSidsCount[injection.sid] || 0
+
+      const exceedsUniqueCount =
+        window.PlayPilotLinkInjections?.config?.unique_injection_count_limit &&
+        previousCount >= window.PlayPilotLinkInjections?.config?.unique_injection_count_limit
+
+      const exceedsTotalParticipantsCount =
+        injection.type === 'participant' &&
+        window.PlayPilotLinkInjections?.config?.total_participant_injection_count_limit &&
+        aiParticipantInjectionsCount >= window.PlayPilotLinkInjections?.config?.total_participant_injection_count_limit
+
+      if (exceedsUniqueCount || exceedsTotalParticipantsCount) {
+        failedMessages[injection.key] = `Injection was withheld because it exceeded the maximum ${exceedsUniqueCount ? 'unique' : 'total'} count.`
+        continue
+      } else {
+        aiInjectionSidsCount[injection.sid] = previousCount + 1
+        if (injection.type === 'participant') aiParticipantInjectionsCount++
+      }
+    }
+
+    limitedInjections.push(injection)
+  }
+
+  return { injections: limitedInjections, failedMessages }
+}
+
 /**
  * Injections are sorted first by their sentence and second by their title length. The first sorting, sentence, is so that injections remain
  * roughly grouped. The second sorting is to prevent shorter titles from merging into longer ones. For instance;