From 065940c1f9b81f6b504fa92979bb7390ad0ba64f Mon Sep 17 00:00:00 2001
From: SpoinkyNL <spoinky.nl@gmail.com>
Date: Mon, 7 Nov 2016 14:55:15 +0100
Subject: [PATCH] Fixed Eurotruck Simulator LUA

---
 .../EurotruckSimulator2DataModel.cs             | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/Artemis/Artemis/Modules/Games/EurotruckSimulator2/EurotruckSimulator2DataModel.cs b/Artemis/Artemis/Modules/Games/EurotruckSimulator2/EurotruckSimulator2DataModel.cs
index e1c9cd275..7fca78695 100644
--- a/Artemis/Artemis/Modules/Games/EurotruckSimulator2/EurotruckSimulator2DataModel.cs
+++ b/Artemis/Artemis/Modules/Games/EurotruckSimulator2/EurotruckSimulator2DataModel.cs
@@ -1,23 +1,26 @@
 using Artemis.Models.Interfaces;
 using Artemis.Modules.Games.EurotruckSimulator2.Data;
 using MoonSharp.Interpreter;
-using MoonSharp.Interpreter.Interop;
 
 namespace Artemis.Modules.Games.EurotruckSimulator2
 {
     [MoonSharpUserData]
     public class EurotruckSimulator2DataModel : IDataModel
     {
-        // TODO: Test LUA functionality
-        [MoonSharpVisible(true)]
+        public EurotruckSimulator2DataModel()
+        {
+            // Register types for LUA
+            UserData.RegisterType<IEts2Game>();
+            UserData.RegisterType<IEts2Job>();
+            UserData.RegisterType<IEts2Navigation>();
+            UserData.RegisterType<IEts2Trailer>();
+            UserData.RegisterType<IEts2Truck>();
+        }
+
         public IEts2Game Game { get; set; }
-        [MoonSharpVisible(true)]
         public IEts2Job Job { get; set; }
-        [MoonSharpVisible(true)]
         public IEts2Navigation Navigation { get; set; }
-        [MoonSharpVisible(true)]
         public IEts2Trailer Trailer { get; set; }
-        [MoonSharpVisible(true)]
         public IEts2Truck Truck { get; set; }
     }
 }
\ No newline at end of file