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 | import net.http | ||||||
| 
 | 
 | ||||||
| const container_build_dir = '/build' | const container_build_dir = '/build' | ||||||
|  | const build_image_repo = 'vieter-build' | ||||||
| 
 | 
 | ||||||
| fn build() ? { | fn create_build_image() ?string { | ||||||
| 	conf := env.load<env.BuildConfig>() ? | 	commands := [ | ||||||
| 
 |  | ||||||
| 	// 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 := [ |  | ||||||
| 		// Update repos & install required packages | 		// Update repos & install required packages | ||||||
| 		'pacman -Syu --needed --noconfirm base-devel git' | 		'pacman -Syu --needed --noconfirm base-devel git' | ||||||
| 		// Add a non-root user to run makepkg | 		// Add a non-root user to run makepkg | ||||||
|  | @ -34,31 +26,11 @@ fn build() ? { | ||||||
| 		'mkdir /build', | 		'mkdir /build', | ||||||
| 		'chown -R builder:builder /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')) | 	cmds_str := base64.encode_str(commands.join('\n')) | ||||||
| 
 | 
 | ||||||
| 	c := docker.NewContainer{ | 	c := docker.NewContainer{ | ||||||
| 		image: 'archlinux:latest' | 		image: 'archlinux:latest' | ||||||
| 		env: ['BUILD_SCRIPT=$cmds_str', 'API_KEY=$conf.api_key'] | 		env: ['BUILD_SCRIPT=$cmds_str'] | ||||||
| 		entrypoint: ['/bin/sh', '-c'] | 		entrypoint: ['/bin/sh', '-c'] | ||||||
| 		cmd: ['echo \$BUILD_SCRIPT | base64 -d | /bin/sh -e'] | 		cmd: ['echo \$BUILD_SCRIPT | base64 -d | /bin/sh -e'] | ||||||
| 	} | 	} | ||||||
|  | @ -81,5 +53,68 @@ fn build() ? { | ||||||
| 		time.sleep(5000000000) | 		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) ? | 	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) | 	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