ARConfiguration: Changed config assignment to be persistent, and added handling for requireConnectionForControl.
ARConfiguration: Changed config assignment to be persistent, and added handling for requireConnectionForControl.

--- a/ARConfiguration.cs
+++ b/ARConfiguration.cs
@@ -19,6 +19,20 @@
 		private Rect configWindowPos;
 
 		private IButton toolbarButton;
+
+		private KSP.IO.PluginConfiguration _config;
+		private KSP.IO.PluginConfiguration config
+		{
+			get
+			{
+				if (this._config == null)
+				{
+					this._config = KSP.IO.PluginConfiguration.CreateForType<AntennaRelay>();
+				}
+
+				return this._config;
+			}
+		}
 
 		public void Awake()
 		{
@@ -46,14 +60,10 @@
 					this.showConfigWindow = !this.showConfigWindow;
 				};
 
-				var config = KSP.IO.PluginConfiguration.CreateForType<AntennaRelay>();
-
-				config.load();
-
-				this.configWindowPos = config.GetValue<Rect>("configWindowPos", this.configWindowPos);
-				AntennaRelay.requireLineOfSight = config.GetValue<bool>("requireLineOfSight", false);
-
-				config.save();
+				this.configWindowPos = this.LoadConfigValue("configWindowPos", this.configWindowPos);
+				AntennaRelay.requireLineOfSight = this.LoadConfigValue("requireLineOfSight", false);
+				ARFlightController.requireConnectionForControl =
+					this.LoadConfigValue("requireConnectionForControl", false);
 			}
 
 			if (this.showConfigWindow)
@@ -81,11 +91,27 @@
 			GUILayout.BeginVertical(GUILayout.ExpandHeight(true));
 
 			GUILayout.BeginHorizontal(GUILayout.ExpandWidth(true));
+
 			bool requireLineOfSight = GUILayout.Toggle(AntennaRelay.requireLineOfSight, "Require Line of Sight");
 			if (requireLineOfSight != AntennaRelay.requireLineOfSight)
 			{
 				AntennaRelay.requireLineOfSight = requireLineOfSight;
 				this.SaveConfigValue("requireLineOfSight", requireLineOfSight);
+			}
+
+			GUILayout.EndHorizontal();
+
+			GUILayout.BeginHorizontal(GUILayout.ExpandWidth(true));
+
+			bool requireConnectionForControl =
+				GUILayout.Toggle(
+					ARFlightController.requireConnectionForControl,
+					"Require Connection for Probe Control"
+				);
+			if (requireConnectionForControl != ARFlightController.requireConnectionForControl)
+			{
+				ARFlightController.requireConnectionForControl = requireConnectionForControl;
+				this.SaveConfigValue("requireConnectionForControl", requireConnectionForControl);
 			}
 
 			GUILayout.EndHorizontal();
@@ -105,22 +131,18 @@
 
 		private T LoadConfigValue<T>(string key, T defaultValue)
 		{
-			var config = KSP.IO.PluginConfiguration.CreateForType<AntennaRelay>();
-
-			config.load();
+			this.config.load();
 
 			return config.GetValue(key, defaultValue);
 		}
 
 		private void SaveConfigValue<T>(string key, T value)
 		{
-			var config = KSP.IO.PluginConfiguration.CreateForType<AntennaRelay>();
+			this.config.load();
 
-			config.load();
+			this.config.SetValue(key, value);
 
-			config.SetValue(key, value);
-
-			config.save();
+			this.config.save();
 		}
 	}
 }

--- /dev/null
+++ b/ARFlightController.cs
@@ -1,1 +1,125 @@
+// AntennaRange
+//
+// ARFlightController.cs
+//
+// Copyright © 2014, toadicus
+// All rights reserved.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+// 1. Redistributions of source code must retain the above copyright notice,
+//    this list of conditions and the following disclaimer.
+//
+// 2. Redistributions in binary form must reproduce the above copyright notice,
+//    this list of conditions and the following disclaimer in the documentation and/or other
+//    materials provided with the distribution.
+//
+// 3. Neither the name of the copyright holder nor the names of its contributors may be used
+//    to endorse or promote products derived from this software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
+// INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+// WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
+using KSP;
+using System;
+using ToadicusTools;
+using UnityEngine;
+
+namespace AntennaRange
+{
+	[KSPAddon(KSPAddon.Startup.Flight, false)]
+	public class ARFlightController : MonoBehaviour
+	{
+		#region Static Members
+		public static bool requireConnectionForControl;
+		#endregion
+
+		#region Properties
+		public ControlTypes currentControlLock
+		{
+			get
+			{
+				if (this.lockID == string.Empty)
+				{
+					return ControlTypes.None;
+				}
+
+				return InputLockManager.GetControlLock(this.lockID);
+			}
+		}
+
+		public string lockID
+		{
+			get;
+			protected set;
+		}
+
+		public ControlTypes lockSet
+		{
+			get
+			{
+				return ControlTypes.GROUPS_ALL | ControlTypes.STAGING | ControlTypes.SAS | ControlTypes.RCS;
+			}
+		}
+
+		public Vessel vessel
+		{
+			get
+			{
+				if (FlightGlobals.ready && FlightGlobals.ActiveVessel != null)
+				{
+					return FlightGlobals.ActiveVessel;
+				}
+
+				return null;
+			}
+		}
+		#endregion
+
+		#region MonoBehaviour LifeCycle
+		protected void Awake()
+		{
+			this.lockID = "ARConnectionRequired";
+		}
+
+		protected void FixedUpdate()
+		{
+			if (
+				requireConnectionForControl &&
+				this.vessel != null &&
+				!this.vessel.hasCrewCommand() &&
+				this.vessel.IsControllable
+			)
+			{
+				if (this.vessel.HasConnectedRelay())
+				{
+					if (currentControlLock != ControlTypes.None)
+					{
+						InputLockManager.RemoveControlLock(this.lockID);
+					}
+				}
+				else
+				{
+					if (currentControlLock == ControlTypes.None)
+					{
+						InputLockManager.SetControlLock(this.lockSet, this.lockID);
+					}
+				}
+			}
+		}
+
+		protected void Destroy()
+		{
+
+		}
+		#endregion
+	}
+}
+
+

--- a/AntennaRange.csproj
+++ b/AntennaRange.csproj
@@ -9,10 +9,9 @@
     <OutputType>Library</OutputType>
     <RootNamespace>AntennaRange</RootNamespace>
     <AssemblyName>AntennaRange</AssemblyName>
-    <ReleaseVersion>0.6.2</ReleaseVersion>
+    <ReleaseVersion>1.1</ReleaseVersion>
     <SynchReleaseVersion>false</SynchReleaseVersion>
     <TargetFrameworkVersion>v3.5</TargetFrameworkVersion>
-    <UseMSBuildEngine>False</UseMSBuildEngine>
   </PropertyGroup>
   <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug_win|AnyCPU' ">
     <DebugSymbols>true</DebugSymbols>
@@ -77,32 +76,9 @@
     <Compile Include="AntennaRelay.cs" />
     <Compile Include="ProtoAntennaRelay.cs" />
     <Compile Include="RelayDatabase.cs" />
-    <Compile Include="..\ToadicusTools\VesselExtensions.cs">
-      <Link>ToadicusTools\VesselExtensions.cs</Link>
-    </Compile>
-    <Compile Include="..\ToadicusTools\Tools.cs">
-      <Link>ToadicusTools\Tools.cs</Link>
-    </Compile>
-    <Compile Include="..\ToadicusTools\MuMech_Tools.cs">
-      <Link>ToadicusTools\MuMech_Tools.cs</Link>
-    </Compile>
     <Compile Include="RelayExtensions.cs" />
-    <Compile Include="..\ToadicusTools\ModuleDBWrapper.cs">
-      <Link>ToadicusTools\ModuleDBWrapper.cs</Link>
-    </Compile>
-    <Compile Include="..\ToadicusTools\PrefabDBWrapper.cs">
-      <Link>ToadicusTools\PrefabDBWrapper.cs</Link>
-    </Compile>
-    <Compile Include="..\ToadicusTools\IModuleDB.cs">
-      <Link>ToadicusTools\IModuleDB.cs</Link>
-    </Compile>
     <Compile Include="ARConfiguration.cs" />
-    <Compile Include="..\ToadicusTools\Wrapper\ToolbarWrapper.cs">
-      <Link>ToadicusTools\ToolbarWrapper.cs</Link>
-    </Compile>
-    <Compile Include="..\ToadicusTools\WindowTools.cs">
-      <Link>WindowTools.cs</Link>
-    </Compile>
+    <Compile Include="ARFlightController.cs" />
   </ItemGroup>
   <Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" />
   <ItemGroup>
@@ -111,20 +87,20 @@
     </None>
   </ItemGroup>
   <ItemGroup>
-    <Folder Include="ToadicusTools\" />
+    <ProjectReference Include="..\ToadicusTools\ToadicusTools.csproj">
+      <Project>{D48A5542-6655-4149-BC27-B27DF0466F1C}</Project>
+      <Name>ToadicusTools</Name>
+    </ProjectReference>
   </ItemGroup>
   <ItemGroup>
     <Reference Include="Assembly-CSharp">
       <HintPath>..\_KSPAssemblies\Assembly-CSharp.dll</HintPath>
-      <Private>False</Private>
     </Reference>
     <Reference Include="System">
       <HintPath>..\_KSPAssemblies\System.dll</HintPath>
-      <Private>False</Private>
     </Reference>
     <Reference Include="UnityEngine">
       <HintPath>..\_KSPAssemblies\UnityEngine.dll</HintPath>
-      <Private>False</Private>
     </Reference>
   </ItemGroup>
 </Project>

--- a/Properties/AssemblyInfo.cs
+++ b/Properties/AssemblyInfo.cs
@@ -37,7 +37,7 @@
 // The assembly version has the format "{Major}.{Minor}.{Build}.{Revision}".
 // The form "{Major}.{Minor}.*" will automatically update the build and revision,
 // and "{Major}.{Minor}.{Build}.*" will update just the revision.
-[assembly: AssemblyVersion("1.0.0.*")]
+[assembly: AssemblyVersion("1.1.*")]
 // The following attributes are used to specify the signing key for the assembly,
 // if desired. See the Mono documentation for more information about signing.
 //[assembly: AssemblyDelaySign(false)]

--- a/RelayExtensions.cs
+++ b/RelayExtensions.cs
@@ -76,6 +76,24 @@
 		{
 			return RelayDatabase.Instance[vessel].Values.ToList();
 		}
+
+		/// <summary>
+		/// Determines if the specified vessel has a connected relay.
+		/// </summary>
+		/// <returns><c>true</c> if the specified vessel has a connected relay; otherwise, <c>false</c>.</returns>
+		/// <param name="vessel"></param>
+		public static bool HasConnectedRelay(this Vessel vessel)
+		{
+			foreach (IAntennaRelay relay in RelayDatabase.Instance[vessel].Values)
+			{
+				if (relay.CanTransmit())
+				{
+					return true;
+				}
+			}
+
+			return false;
+		}
 	}
 }