From c9e28a2bff78f9cdd4d0c287bd6e27eb9d32ba13 Mon Sep 17 00:00:00 2001
From: Eryn Wells <eryn@erynwells.me>
Date: Fri, 16 Oct 2015 23:50:48 -0700
Subject: [PATCH] Move lib environment methods to the lib tool

---
 site_scons/site_init.py      | 12 ------------
 site_scons/site_tools/lib.py | 12 +++++++++++-
 2 files changed, 11 insertions(+), 13 deletions(-)

diff --git a/site_scons/site_init.py b/site_scons/site_init.py
index c0e207d..881bd6d 100644
--- a/site_scons/site_init.py
+++ b/site_scons/site_init.py
@@ -154,10 +154,6 @@ class Environment(SCons.Environment.Environment):
                 continue
             yield (lib, lib_dir)
 
-    #
-    # Library processing
-    #
-
     def process_lib_dirs(self):
         self.log('Processing libs in #{} ...'.format(self.lib_root.path))
         for name, lib in self.lib_dirs:
@@ -170,14 +166,6 @@ class Environment(SCons.Environment.Environment):
         # do_sconscript(env, env.source_root, src_out_dir)
         self.Append(CPPPATH=[self.src_root])
 
-    def lib(self, name):
-        return self['LIBS'].get(name)
-
-    def register_lib(self, name, lib):
-        if name in self['LIBS']:
-            self.log_error('Library has already been built: {}'.format(name))
-        self['LIBS'][name] = lib
-
     #
     # Test processing
     #
diff --git a/site_scons/site_tools/lib.py b/site_scons/site_tools/lib.py
index 212bd6b..2e01526 100644
--- a/site_scons/site_tools/lib.py
+++ b/site_scons/site_tools/lib.py
@@ -8,6 +8,15 @@ SCons builder for a lib directory.
 import SCons.Errors
 import SCons.Script
 
+def _lib(env, name):
+    return env['LIBS'].get(name)
+
+
+def _register_lib(env, name, lib):
+    if name in env['LIBS']:
+        env.log_error('Library has already been built: {}'.format(name))
+    env['LIBS'][name] = lib
+
 
 def _process_lib_dir(env, lib, src_dir=None, out_dir=None, inc_dir=None):
     if not src_dir:
@@ -39,7 +48,7 @@ def _build_library(env, lib_func):
 
     def builder(env, lib_name, sources, *args, **kwargs):
         lib = original_builder(lib_name, sources, *args, **kwargs)
-        env.register_lib(lib_name, lib)
+        _register_lib(env, lib_name, lib)
         return lib
 
     return builder
@@ -54,6 +63,7 @@ def generate(env):
     env.AddMethod(_build_library(env, env.Library), 'Library')
     env.AddMethod(_build_library(env, env.StaticLibrary), 'StaticLibrary')
     env.AddMethod(_build_library(env, env.SharedLibrary), 'SharedLibrary')
+    env.AddMethod(_lib, 'lib')
 
 def exists(env):
     return True