Extending objects in NamedDomainObjectContainers (GRADLE)

build.gradle

import org.gradle.internal.reflect.Instantiator

class Host {
	String name
	
	Host(String name){
            this.name = name
	}
}

class Foo {
	def host
	def someFooVar = "default"
	
	public Foo(Host fooHost){
		this.host = fooHost
	}
}

project.extensions.hosts = project.container(Host, new NamedDomainObjectFactory<Host>(){
	Host create(String name){
		def instantiator = project.services.get(Instantiator.class)
		return instantiator.newInstance(Host, name)
	}
})

hosts.all{ currentHost ->
	currentHost.extensions.create('foo', Foo, currentHost)	
}

hosts {
	host1 {
		foo {
			someFooVar = "bar"
		}
	}
}

task assertHostConfig << {
	assert hosts.host1.foo.someFooVar == "bar"
}