Enable autoscaling based on worksets

docs/webhook_events.md

@@ -1245,20 +1245,26 @@ Each event will be submitted via HTTP POST to the user provided URL.
                     "title": "Ephemeral Os Disks",
                     "type": "boolean"
                 },
+                "extra_available_size": {
+                    "description": "number of unused but available nodes",
+                    "minimum": 0,
+                    "title": "Extra Available Size",
+                    "type": "integer"
+                },
                 "image": {
                     "title": "Image",
                     "type": "string"
                 },
                 "max_size": {
                     "default": 1000,
-                    "maximum": 1000,
+                    "description": "maximum size of the pool",
                     "minimum": 0,
                     "title": "Max Size",
                     "type": "integer"
                 },
                 "min_size": {
                     "default": 0,
-                    "maximum": 1000,
+                    "description": "minimum size of the pool",
                     "minimum": 0,
                     "title": "Min Size",
                     "type": "integer"
@@ -1268,6 +1274,8 @@ Each event will be submitted via HTTP POST to the user provided URL.
                     "type": "string"
                 },
                 "scaleset_size": {
+                    "default": 1,
+                    "description": "unused",
                     "title": "Scaleset Size",
                     "type": "integer"
                 },
@@ -1283,7 +1291,6 @@ Each event will be submitted via HTTP POST to the user provided URL.
             },
             "required": [
                 "image",
-                "scaleset_size",
                 "vm_sku"
             ],
             "title": "AutoScaleConfig",
@@ -4791,20 +4798,26 @@ Each event will be submitted via HTTP POST to the user provided URL.
                     "title": "Ephemeral Os Disks",
                     "type": "boolean"
                 },
+                "extra_available_size": {
+                    "description": "number of unused but available nodes",
+                    "minimum": 0,
+                    "title": "Extra Available Size",
+                    "type": "integer"
+                },
                 "image": {
                     "title": "Image",
                     "type": "string"
                 },
                 "max_size": {
                     "default": 1000,
-                    "maximum": 1000,
+                    "description": "maximum size of the pool",
                     "minimum": 0,
                     "title": "Max Size",
                     "type": "integer"
                 },
                 "min_size": {
                     "default": 0,
-                    "maximum": 1000,
+                    "description": "minimum size of the pool",
                     "minimum": 0,
                     "title": "Min Size",
                     "type": "integer"
@@ -4814,6 +4827,8 @@ Each event will be submitted via HTTP POST to the user provided URL.
                     "type": "string"
                 },
                 "scaleset_size": {
+                    "default": 1,
+                    "description": "unused",
                     "title": "Scaleset Size",
                     "type": "integer"
                 },
@@ -4829,7 +4844,6 @@ Each event will be submitted via HTTP POST to the user provided URL.
             },
             "required": [
                 "image",
-                "scaleset_size",
                 "vm_sku"
             ],
             "title": "AutoScaleConfig",

src/api-service/__app__/onefuzzlib/azure/queue.py

@@ -12,6 +12,7 @@
 
 from azure.core.exceptions import ResourceExistsError, ResourceNotFoundError
 from azure.storage.queue import (
+    QueueMessage,
     QueueSasPermissions,
     QueueServiceClient,
     generate_queue_sas,
@@ -172,9 +173,16 @@ def peek_queue(
             result.append(object_type.parse_obj(raw))
     except ResourceNotFoundError:
         return result
+
     return result
 
 
+def decode_message(message: QueueMessage, object_type: Type[A]) -> A:
+    decoded = base64.b64decode(message.content)
+    raw = json.loads(decoded)
+    return object_type.parse_obj(raw)
+
+
 def queue_object(
     name: QueueNameType,
     message: BaseModel,

src/api-service/__app__/onefuzzlib/tasks/main.py

@@ -18,7 +18,6 @@
 from onefuzztypes.models import Error
 from onefuzztypes.models import Task as BASE_TASK
 from onefuzztypes.models import TaskConfig, TaskVm, UserInfo
-from onefuzztypes.primitives import PoolName
 
 from ..azure.image import get_os
 from ..azure.queue import create_queue, delete_queue
@@ -171,23 +170,6 @@ def get_by_task_id(cls, task_id: UUID) -> Union[Error, "Task"]:
         task = tasks[0]
         return task
 
-    @classmethod
-    def get_tasks_by_pool_name(cls, pool_name: PoolName) -> List["Task"]:
-        tasks = cls.search_states(states=TaskState.available())
-        if not tasks:
-            return []
-
-        pool_tasks = []
-
-        for task in tasks:
-            task_pool = task.get_pool()
-            if not task_pool:
-                continue
-            if pool_name == task_pool.name:
-                pool_tasks.append(task)
-
-        return pool_tasks
-
     def mark_stopping(self) -> None:
         if self.state in TaskState.shutting_down():
             logging.debug(