greybel-mock-environment 1.1.0 → 1.1.2

package/dist/environment.d.ts

@@ -3,7 +3,7 @@ import EmailGenerator from './generators/email';
 import NetworkGenerator from './generators/network';
 import UserGenerator from './generators/user';
 import VulnerabilityGenerator from './generators/vulnerability';
-import { Computer, EMail, Port, Router, Switch, User } from './types';
+import { Computer, EMail, Port, Router, Session, Switch } from './types';
 export default class MockEnvironment {
     seed: string;
     basicGenerator: BasicGenerator;
@@ -11,7 +11,8 @@ export default class MockEnvironment {
     userGenerator: UserGenerator;
     vulnerabilityGenerator: VulnerabilityGenerator;
     networkGenerator: NetworkGenerator;
-    localComputer: Computer;
+    sessions: Session[];
+    localSession: Session;
     constructor(seed: string, localUser: {
         username: string;
         password: string;
@@ -25,10 +26,8 @@ export default class MockEnvironment {
         computer: Computer;
     } | null;
     setupLibraries(): void;
-    getLocal(): {
-        computer: Computer;
-        user: User;
-    };
+    getLatestSession(): Session;
+    getLocal(): Session;
     connectLocal(router: Router): void;
     getRouterByIp(ipAddress: string): Router | null;
     getSwitchByIp(ipAddress: string): Switch | null;

package/dist/environment.js

@@ -40,28 +40,37 @@ var utils_1 = require("./utils");
 var MockEnvironment = /** @class */ (function () {
     function MockEnvironment(seed, localUser) {
         if (seed === void 0) { seed = 'test'; }
-        this.seed = seed;
-        this.basicGenerator = new basic_1.default({ seed: seed });
-        this.emailGenerator = new email_1.default({
+        var me = this;
+        me.seed = seed;
+        me.basicGenerator = new basic_1.default({ seed: seed });
+        me.emailGenerator = new email_1.default({
             seed: seed,
-            basicGenerator: this.basicGenerator
+            basicGenerator: me.basicGenerator
         });
-        this.userGenerator = new user_1.default({
+        me.userGenerator = new user_1.default({
             seed: seed,
-            basicGenerator: this.basicGenerator,
-            emailGenerator: this.emailGenerator
+            basicGenerator: me.basicGenerator,
+            emailGenerator: me.emailGenerator
         });
-        this.vulnerabilityGenerator = new vulnerability_1.default({
+        me.vulnerabilityGenerator = new vulnerability_1.default({
             seed: seed,
-            basicGenerator: this.basicGenerator
+            basicGenerator: me.basicGenerator
         });
-        this.networkGenerator = new network_1.default({
+        me.networkGenerator = new network_1.default({
             seed: seed,
-            basicGenerator: this.basicGenerator,
-            userGenerator: this.userGenerator,
-            emailGenerator: this.emailGenerator
+            basicGenerator: me.basicGenerator,
+            userGenerator: me.userGenerator,
+            emailGenerator: me.emailGenerator
+        });
+        var user = me.userGenerator.generate(localUser.username, localUser.password);
+        var computer = me.networkGenerator.generateLocalComputer([user], 'test');
+        var location = computer.getHome(user);
+        me.localSession = new types_1.Session({
+            user: user,
+            computer: computer,
+            location: location
         });
-        this.localComputer = this.networkGenerator.generateLocalComputer([this.userGenerator.generate(localUser.username, localUser.password)], 'test');
+        me.sessions.push(me.localSession);
     }
     MockEnvironment.prototype.getComputerInLan = function (ipAddress, router) {
         var me = this;
@@ -135,13 +144,11 @@ var MockEnvironment = /** @class */ (function () {
             remote: true
         });
     };
+    MockEnvironment.prototype.getLatestSession = function () {
+        return this.sessions[this.sessions.length - 1];
+    };
     MockEnvironment.prototype.getLocal = function () {
-        var computer = this.localComputer;
-        var user = computer.users[1];
-        return {
-            computer: computer,
-            user: user
-        };
+        return this.localSession;
     };
     MockEnvironment.prototype.connectLocal = function (router) {
         var me = this;
@@ -149,15 +156,15 @@ var MockEnvironment = /** @class */ (function () {
         if (!serverMap.has(router.publicIp)) {
             throw new Error("Router with ip ".concat(router.publicIp, " is unknown in serverMap."));
         }
-        var localComputer = me.localComputer;
-        if (localComputer.router) {
-            localComputer.router.devices.delete(localComputer.localIp);
+        var computer = me.localSession.computer;
+        if (computer.router) {
+            computer.router.devices.delete(computer.localIp);
         }
         var serverMapGenerator = serverMap.get(router.publicIp);
         var localIp = serverMapGenerator.generateLocalIp(router);
-        localComputer.router = router;
-        localComputer.localIp = localIp;
-        router.devices.set(localIp, localComputer);
+        computer.router = router;
+        computer.localIp = localIp;
+        router.devices.set(localIp, computer);
     };
     MockEnvironment.prototype.getRouterByIp = function (ipAddress) {
         var me = this;
@@ -178,7 +185,7 @@ var MockEnvironment = /** @class */ (function () {
         if (!(0, utils_1.isValidIp)(ipAddress) || !(0, utils_1.isLanIp)(ipAddress)) {
             return null;
         }
-        var localRouter = me.localComputer.router;
+        var localRouter = me.getLatestSession().computer.router;
         try {
             for (var _b = __values(localRouter.devices), _c = _b.next(); !_c.done; _c = _b.next()) {
                 var _d = __read(_c.value, 2), localIp = _d[0], device = _d[1];

package/dist/fs.d.ts

@@ -1,36 +1,3 @@
-import * as Type from './types';
-export declare function getUserFolder(parent: Type.FSEntity, user: string): Type.Folder;
-export declare function getDefaultFileSystem(users: Type.User[]): Type.Folder;
-export interface PermissionSegment {
-    x: boolean;
-    w: boolean;
-    r: boolean;
-}
-export interface PermissionMap {
-    u: PermissionSegment;
-    g: PermissionSegment;
-    o: PermissionSegment;
-}
-export declare function getPermissionSegmentValueByString(flag: PermissionSegment, flagType: string): boolean;
-export declare function setPermissionSegmentValueByString(flag: PermissionSegment, flagType: string, value: boolean): void;
-export declare function getPermissionSegmentByString(flag: PermissionMap, userType: string): PermissionSegment;
-export declare function transformFlagsToPermissions(permissionMap: PermissionMap): string;
-export declare function parsePermissions(file: Type.FSEntity): PermissionMap;
-export declare function getPermissions(user: Type.User, file: Type.FSEntity): PermissionSegment;
-export declare function getFile(entity: Type.FSEntity, path: string[]): Type.FSEntity | null;
-export declare function hasFile(folder: Type.Folder, fileName: string): boolean;
-export declare function getFileIndex(folder: Type.Folder, fileName: string): {
-    isFolder: boolean;
-    index: number;
-};
-export declare function putFile(folder: Type.Folder, file: Type.File): void;
-export declare function removeFile(folder: Type.Folder, fileName: string): boolean;
-export declare function traverseChildren(entity: Type.FSEntity, callback: (v: Type.FSEntity) => void, skip?: boolean): void;
-export declare function copyFile(entity: Type.FSEntity, parent: Type.FSEntity): Type.FSEntity;
-export declare function getHomePath(user: Type.User, computer: Type.Computer): string[] | null;
-export declare function getFilePath(entity: Type.FSEntity): string[] | null;
-export declare function getTraversalPath(path: string | null, currentLocation: string[] | null): null | string[];
-export declare function getFileLibrary(file: Type.File): Type.Library | null;
-export declare function getServiceLibrary(service: Type.Service): Type.Library | null;
-export declare function getUserByVulnerability(vulActionUser: Type.VulnerabilityActionUser, computer: Type.Computer): Type.User;
-export declare function changePassword(computer: Type.Computer, username: string, password: string): boolean;
+import { Folder, FSEntity, User } from './types';
+export declare function getUserFolder(parent: FSEntity, user: string): Folder;
+export declare function getDefaultFileSystem(users: User[]): Folder;