forked from vieter-v/vieter
				
			Split builds into separate containers; made makepkg parallel
							parent
							
								
									13a2ced6f9
								
							
						
					
					
						commit
						540574b3c3
					
				
							
								
								
									
										99
									
								
								src/build.v
								
								
								
								
							
							
						
						
									
										99
									
								
								src/build.v
								
								
								
								
							| 
						 | 
				
			
			@ -10,18 +10,10 @@ import env
 | 
			
		|||
import net.http
 | 
			
		||||
 | 
			
		||||
const container_build_dir = '/build'
 | 
			
		||||
const build_image_repo = 'vieter-build'
 | 
			
		||||
 | 
			
		||||
fn build() ? {
 | 
			
		||||
	conf := env.load<env.BuildConfig>() ?
 | 
			
		||||
 | 
			
		||||
	// We get the repos list from the Vieter instance
 | 
			
		||||
	mut req := http.new_request(http.Method.get, '$conf.address/api/repos', '') ?
 | 
			
		||||
	req.add_custom_header('X-Api-Key', conf.api_key) ?
 | 
			
		||||
 | 
			
		||||
	res := req.do() ?
 | 
			
		||||
	repos := json.decode([]server.GitRepo, res.text) ?
 | 
			
		||||
 | 
			
		||||
	mut commands := [
 | 
			
		||||
fn create_build_image() ?string {
 | 
			
		||||
	commands := [
 | 
			
		||||
		// Update repos & install required packages
 | 
			
		||||
		'pacman -Syu --needed --noconfirm base-devel git'
 | 
			
		||||
		// Add a non-root user to run makepkg
 | 
			
		||||
| 
						 | 
				
			
			@ -34,31 +26,11 @@ fn build() ? {
 | 
			
		|||
		'mkdir /build',
 | 
			
		||||
		'chown -R builder:builder /build',
 | 
			
		||||
	]
 | 
			
		||||
 | 
			
		||||
	// Each repo gets a unique UUID to avoid naming conflicts when cloning
 | 
			
		||||
	mut uuids := []string{}
 | 
			
		||||
 | 
			
		||||
	for repo in repos {
 | 
			
		||||
		mut uuid := rand.uuid_v4()
 | 
			
		||||
 | 
			
		||||
		// Just to be sure we don't have any collisions
 | 
			
		||||
		for uuids.contains(uuid) {
 | 
			
		||||
			uuid = rand.uuid_v4()
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		uuids << uuid
 | 
			
		||||
 | 
			
		||||
		commands << "su builder -c 'git clone --single-branch --depth 1 --branch $repo.branch $repo.url /build/$uuid'"
 | 
			
		||||
		commands << 'su builder -c \'cd /build/$uuid && makepkg -s --noconfirm --needed && for pkg in \$(ls -1 *.pkg*); do curl -XPOST -T "\${pkg}" -H "X-API-KEY: \$API_KEY" $conf.address/publish; done\''
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// We convert the list of commands into a base64 string, which then gets
 | 
			
		||||
	// passed to the container as an env var
 | 
			
		||||
	cmds_str := base64.encode_str(commands.join('\n'))
 | 
			
		||||
 | 
			
		||||
	c := docker.NewContainer{
 | 
			
		||||
		image: 'archlinux:latest'
 | 
			
		||||
		env: ['BUILD_SCRIPT=$cmds_str', 'API_KEY=$conf.api_key']
 | 
			
		||||
		env: ['BUILD_SCRIPT=$cmds_str']
 | 
			
		||||
		entrypoint: ['/bin/sh', '-c']
 | 
			
		||||
		cmd: ['echo \$BUILD_SCRIPT | base64 -d | /bin/sh -e']
 | 
			
		||||
	}
 | 
			
		||||
| 
						 | 
				
			
			@ -81,5 +53,68 @@ fn build() ? {
 | 
			
		|||
		time.sleep(5000000000)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Finally, we create the image from the container
 | 
			
		||||
	// As the tag, we use the epoch value
 | 
			
		||||
	tag := time.sys_mono_now().str()
 | 
			
		||||
	image := docker.create_image_from_container(id, 'vieter-build', tag) ?
 | 
			
		||||
	docker.remove_container(id) ?
 | 
			
		||||
 | 
			
		||||
	return image.id
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn build() ? {
 | 
			
		||||
	conf := env.load<env.BuildConfig>() ?
 | 
			
		||||
 | 
			
		||||
	// We get the repos list from the Vieter instance
 | 
			
		||||
	mut req := http.new_request(http.Method.get, '$conf.address/api/repos', '') ?
 | 
			
		||||
	req.add_custom_header('X-Api-Key', conf.api_key) ?
 | 
			
		||||
 | 
			
		||||
	res := req.do() ?
 | 
			
		||||
	repos := json.decode([]server.GitRepo, res.text) ?
 | 
			
		||||
 | 
			
		||||
	// No point in doing work if there's no repos present
 | 
			
		||||
	if repos.len == 0 {
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// First, we create a base image which has updated repos n stuff
 | 
			
		||||
	image_id := create_build_image() ?
 | 
			
		||||
 | 
			
		||||
	for repo in repos {
 | 
			
		||||
		commands := [
 | 
			
		||||
			"su builder -c 'git clone --single-branch --depth 1 --branch $repo.branch $repo.url /build/repo'"
 | 
			
		||||
			'su builder -c \'cd /build/repo && MAKEFLAGS="-j\$(nproc)" makepkg -s --noconfirm --needed && for pkg in \$(ls -1 *.pkg*); do curl -XPOST -T "\$pkg" -H "X-API-KEY: \$API_KEY" $conf.address/publish; done\''
 | 
			
		||||
		]
 | 
			
		||||
 | 
			
		||||
		// We convert the list of commands into a base64 string, which then gets
 | 
			
		||||
		// passed to the container as an env var
 | 
			
		||||
		cmds_str := base64.encode_str(commands.join('\n'))
 | 
			
		||||
 | 
			
		||||
		c := docker.NewContainer{
 | 
			
		||||
			image: '$image_id'
 | 
			
		||||
			env: ['BUILD_SCRIPT=$cmds_str', 'API_KEY=$conf.api_key']
 | 
			
		||||
			entrypoint: ['/bin/sh', '-c']
 | 
			
		||||
			cmd: ['echo \$BUILD_SCRIPT | base64 -d | /bin/sh -e']
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		id := docker.create_container(c) ?
 | 
			
		||||
		docker.start_container(id) ?
 | 
			
		||||
 | 
			
		||||
		// This loop waits until the container has stopped, so we can remove it after
 | 
			
		||||
		for {
 | 
			
		||||
			data := docker.inspect_container(id) ?
 | 
			
		||||
 | 
			
		||||
			if !data.state.running {
 | 
			
		||||
				break
 | 
			
		||||
			}
 | 
			
		||||
 | 
			
		||||
			// Wait for 5 seconds
 | 
			
		||||
			time.sleep(5000000000)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		docker.remove_container(id) ?
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// Finally, we remove the builder image
 | 
			
		||||
	docker.remove_image(image_id) ?
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -91,8 +91,3 @@ pub fn request_with_json<T>(method string, url urllib.URL, data &T) ?http.Respon
 | 
			
		|||
 | 
			
		||||
	return request_with_body(method, url, 'application/json', body)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// pull_image pulls tries to pull the image for the given image & tag
 | 
			
		||||
pub fn pull_image(image string, tag string) ?http.Response {
 | 
			
		||||
	return request('POST', urllib.parse('/v1.41/images/create?fromImage=$image&tag=$tag') ?)
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -0,0 +1,31 @@
 | 
			
		|||
module docker
 | 
			
		||||
 | 
			
		||||
import net.http
 | 
			
		||||
import net.urllib
 | 
			
		||||
import json
 | 
			
		||||
 | 
			
		||||
struct Image {
 | 
			
		||||
pub:
 | 
			
		||||
	id string [json: Id]
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// pull_image pulls tries to pull the image for the given image & tag
 | 
			
		||||
pub fn pull_image(image string, tag string) ?http.Response {
 | 
			
		||||
	return request('POST', urllib.parse('/v1.41/images/create?fromImage=$image&tag=$tag') ?)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn create_image_from_container(id string, repo string, tag string) ?Image {
 | 
			
		||||
	res := request('POST', urllib.parse('/v1.41/commit?container=$id&repo=$repo&tag=$tag') ?) ?
 | 
			
		||||
 | 
			
		||||
	if res.status_code != 201 {
 | 
			
		||||
		return error('Failed to create image from container.')
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return json.decode(Image, res.text) or {}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
pub fn remove_image(id string) ?bool {
 | 
			
		||||
	res := request('DELETE', urllib.parse('/v1.41/images/$id') ?) ?
 | 
			
		||||
 | 
			
		||||
	return res.status_code == 200
 | 
			
		||||
}
 | 
			
		||||
		Loading…
	
		Reference in New Issue