Add registry package

Automatic inclusion of source on import

Makefile

@@ -21,6 +21,7 @@ TEST := -v --run OttoError
 TEST := .
 
 test:
+	go test -i
 	go test $(TEST)
 
 assets:

otto.go

@@ -74,6 +74,7 @@ package otto
 
 import (
 	"fmt"
+	"github.com/robertkrimen/otto/registry"
 )
 
 // Otto is the representation of the JavaScript runtime. Each instance of Otto has a self-contained namespace.
@@ -87,6 +88,11 @@ func New() *Otto {
 		runtime: newContext(),
 	}
 	self.Set("console", self.runtime.newConsole())
+
+	registry.Apply(func(entry registry.Entry){
+		self.Run(entry.Source())
+	})
+
 	return self
 }
 

registry/registry.go

@@ -0,0 +1,42 @@
+package registry
+
+var registry []*Entry = make([]*Entry, 0)
+
+type Entry struct {
+	active bool
+	source string
+}
+
+func newEntry(source string) *Entry {
+	return &Entry{
+		active: true,
+		source: source,
+	}
+}
+
+func (self *Entry) Enable() {
+	self.active = true
+}
+
+func (self *Entry) Disable() {
+	self.active = false
+}
+
+func (self Entry) Source() string {
+	return self.source
+}
+
+func Apply(callback func(Entry)) {
+	for _, entry := range registry {
+		if !entry.active {
+			continue
+		}
+		callback(*entry)
+	}
+}
+
+func Register(source string) *Entry {
+	entry := newEntry(source)
+	registry = append(registry, entry)
+	return entry
+}

underscore/underscore.go

@@ -7,6 +7,20 @@ https://github.com/documentcloud/underscore
 */
 package underscore
 
+import (
+	"github.com/robertkrimen/otto/registry"
+)
+
+var entry *registry.Entry = registry.Register(Source())
+
+func Enable() {
+	entry.Enable()
+}
+
+func Disable() {
+	entry.Disable()
+}
+
 // Source returns the underscore source.
 func Source() string {
 	return string(underscore())

underscore_test.go

@@ -6,6 +6,10 @@ import (
     "github.com/robertkrimen/otto/underscore"
 )
 
+func init() {
+	underscore.Disable()
+}
+
 func underscoreTest() func(string, ... interface{}) Value {
 	Otto, test := runTestWithOtto()
 	Otto.Run(underscore.Source())