From 98c304ccd4d606b2bcc134379d8f3a3e47e81c29 Mon Sep 17 00:00:00 2001 From: Rahul Rudragoudar Date: Wed, 21 Apr 2021 14:39:20 +0530 Subject: [PATCH] Pick a random captcha provider instead of params Signed-off-by: Rahul Rudragoudar --- src/main/scala/lc/background/taskThread.scala | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/src/main/scala/lc/background/taskThread.scala b/src/main/scala/lc/background/taskThread.scala index 3508d02..6e9dbd8 100644 --- a/src/main/scala/lc/background/taskThread.scala +++ b/src/main/scala/lc/background/taskThread.scala @@ -30,11 +30,16 @@ class BackgroundTask(throttle: Int, timeLimit: Int) { } private def getRandomParam(): Parameters = { - val level = Config.allowedLevels.toList(Config.getNextRandomInt(Config.allowedLevels.size)) - val media = Config.allowedMedia.toList(Config.getNextRandomInt(Config.allowedMedia.size)) - val inputType = Config.allowedInputType.toList(Config.getNextRandomInt(Config.allowedInputType.size)) + val captcha = pickRandom(Config.captchaConfig) + val level = pickRandom(captcha.allowedLevels) + val media = pickRandom(captcha.allowedMedia) + val inputType = pickRandom(captcha.allowedInputType) - Parameters(level, media, inputType, Some(Size(0,0))) + Parameters(level, media, inputType, Some(Size(0, 0))) + } + + private def pickRandom[T](list: List[T]): T = { + list(Config.getNextRandomInt(list.size)) } def beginThread(delay: Int): Unit = {