From d0e0c2fce2c1e5b7f8b7a0afc5b73e2411f4777d Mon Sep 17 00:00:00 2001
From: MinaciousGrace <minaciousgrace@gmail.com>
Date: Fri, 28 Apr 2017 15:13:57 -0400
Subject: [PATCH] allow lua access to get song/steps pointers by chartkey

---
 src/SongManager.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/SongManager.cpp b/src/SongManager.cpp
index 34de3d5a85..d40356fba2 100644
--- a/src/SongManager.cpp
+++ b/src/SongManager.cpp
@@ -2291,6 +2291,18 @@ class LunaSongManager: public Luna<SongManager>
 		lua_pushboolean(L, p->WasLoadedFromAdditionalCourses(pCourse));
 		return 1;
 	}
+	static int GetSongByChartKey(T* p, lua_State *L)
+	{
+		Song* pSong = p->GetSongByChartkey(SArg(1));
+		pSong->PushSelf(L);
+		return 1;
+	}
+	static int GetStepsByChartKey(T* p, lua_State *L)
+	{
+		Steps* pSteps = p->GetStepsByChartkey(SArg(1));
+		pSteps->PushSelf(L);
+		return 1;
+	}
 
 	LunaSongManager()
 	{
@@ -2333,6 +2345,8 @@ class LunaSongManager: public Luna<SongManager>
 		ADD_METHOD( SongToPreferredSortSectionName );
 		ADD_METHOD( WasLoadedFromAdditionalSongs );
 		ADD_METHOD( WasLoadedFromAdditionalCourses );
+		ADD_METHOD( GetSongByChartKey );
+		ADD_METHOD( GetStepsByChartKey );
 	}
 };