Updated 'core'.

[hub.git] / application / hub / main / states / dht / class_BaseDhtState.php

diff --git a/application/hub/main/states/dht/class_BaseDhtState.php b/application/hub/main/states/dht/class_BaseDhtState.php
index a78f9dc161cc425ef0f639ca07cce656452c1883..44e0b5584ffe60c00aa8a540bb0ea810fb169d6b 100644 (file)
--- a/application/hub/main/states/dht/class_BaseDhtState.php
+++ b/application/hub/main/states/dht/class_BaseDhtState.php
@@ -2,11 +2,12 @@
 /**
  * A general DHT state class
  *
- * @author             Roland Haeder <webmaster@ship-simu.org>
+ * @author             Roland Haeder <webmaster@shipsimu.org>
  * @version            0.0.0
- * @copyright  Copyright (c) 2007, 2008 Roland Haeder, 2009 - 2012 Hub Developer Team
+ * @copyright  Copyright (c) 2007, 2008 Roland Haeder, 2009 - 2015 Hub Developer Team
  * @license            GNU GPL 3.0 or any newer version
- * @link               http://www.ship-simu.org
+ * @link               http://www.shipsimu.org
+ * @todo               Create generic DHT interface
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -38,15 +39,29 @@ class BaseDhtState extends BaseState {
         * is every other state.
         *
         * @return      void
-        * @throws      InvalidStateException   If the state is not 'virgin'
+        * @throws      UnexpectedStateException        If the state is not 'virgin'
         */
-       public function validateDhtStateIsVirgin () {
+       public function validateDhtStateIsVirginOrInit () {
                // Just compare it...
-               if (!$this instanceof DhtVirginState) {
+               if ((!$this instanceof DhtVirginState) && (!$this instanceof DhtInitState)) {
                        // Throw the exception
-                       throw new InvalidStateException($this, self::EXCEPTION_INVALID_STATE);
+                       throw new UnexpectedStateException($this, self::EXCEPTION_INVALID_STATE);
                } // END - if
        }
+
+       /**
+        * Checks whether the DHT's state is 'booting' by comparing it with the
+        * state's class name.
+        *
+        * @return      $isBooting      Whether this DHT's state is 'booting'
+        */
+       public function ifDhtIsBooting () {
+               // Check state
+               $isBooting = ($this instanceof DhtBootingState);
+
+               // Return status
+               return $isBooting;
+       }
 }
 
 // [EOF]