Merge pull request #2618 from pigalot/Corp-Api-Updates

Corp api updates

src/Corporation/OfficeSpace.ts

@@ -173,6 +173,40 @@ export class OfficeSpace {
     return false;
   }
 
+  setEmployeeToJob(job: string, amount: number): boolean {
+    let unassignedCount = 0;
+    let jobCount = 0;
+    for (let i = 0; i < this.employees.length; ++i) {
+      if (this.employees[i].pos === EmployeePositions.Unassigned) {
+        unassignedCount++;
+      } else if (this.employees[i].pos === job) {
+        jobCount++;
+      }
+    }
+
+    if ((jobCount + unassignedCount) < amount) return false;
+
+    for (let i = 0; i < this.employees.length; ++i) {
+      if (this.employees[i].pos === EmployeePositions.Unassigned) {
+        if (jobCount <= amount) {
+          this.employees[i].pos = job;
+          jobCount++;
+          unassignedCount--;
+        }
+        if (jobCount === amount) break;
+      } else if (this.employees[i].pos === job) {
+        if (jobCount >= amount) {
+          this.employees[i].pos = EmployeePositions.Unassigned;
+          jobCount--;
+          unassignedCount++;
+        }
+        if (jobCount === amount) break;
+      }
+    }
+    if (jobCount !== amount) return false;
+    return true;
+  }
+
   toJSON(): any {
     return Generic_toJSON("OfficeSpace", this);
   }