Compare commits

...

4 Commits

Author SHA1 Message Date
Jef Roosens 5991b09528
some better error handling 2022-05-13 22:15:30 +02:00
Jef Roosens 7c451d04c6
WIP 2022-05-13 22:03:06 +02:00
Jef Roosens 341c94669c
refactor: ran vfmt with new defaults
feat(docker): started work on new implementation
2022-05-13 21:28:24 +02:00
Jef Roosens 1caccceda2
wip 2022-05-11 22:14:20 +02:00
26 changed files with 349 additions and 162 deletions

View File

@ -17,6 +17,8 @@ const build_image_repo = 'vieter-build'
// makepkg with. The base image should be some Linux distribution that uses
// Pacman as its package manager.
pub fn create_build_image(base_image string) ?string {
mut dd := docker.new_conn() ?
commands := [
// Update repos & install required packages
'pacman -Syu --needed --noconfirm base-devel git'
@ -48,12 +50,14 @@ pub fn create_build_image(base_image string) ?string {
// We pull the provided image
docker.pull_image(image_name, image_tag)?
id := docker.create_container(c) ?
docker.start_container(id) ?
id := dd.create_container(c)?.id
/* id := docker.create_container(c)? */
dd.start_container(id)?
// This loop waits until the container has stopped, so we can remove it after
for {
data := docker.inspect_container(id) ?
println('wot')
data := dd.inspect_container(id)?
if !data.state.running {
break
@ -68,7 +72,7 @@ pub fn create_build_image(base_image string) ?string {
// conflicts.
tag := time.sys_mono_now().str()
image := docker.create_image_from_container(id, 'vieter-build', tag)?
docker.remove_container(id) ?
dd.remove_container(id)?
return image.id
}

View File

@ -3,19 +3,33 @@ module docker
import json
import net.urllib
import time
import net.http
struct DockerError {
message string
}
struct Container {
id string [json: Id]
names []string [json: Names]
}
// containers returns a list of all currently running containers
pub fn containers() ?[]Container {
res := request('GET', urllib.parse('/v1.41/containers/json') ?) ?
pub fn (mut d DockerDaemon) containers() ?[]Container {
d.send_request('GET', urllib.parse('/v1.41/containers/json')?)?
head, res := d.read_response()?
return json.decode([]Container, res.text) or {}
if head.status_code != 200 {
data := json.decode(DockerError, res)?
return error(data.message)
}
data := json.decode([]Container, res)?
return data
}
[params]
pub struct NewContainer {
image string [json: Image]
entrypoint []string [json: Entrypoint]
@ -26,7 +40,35 @@ pub struct NewContainer {
}
struct CreatedContainer {
pub:
id string [json: Id]
warnings []string [json: Warnings]
}
pub fn (mut d DockerDaemon) create_container(c NewContainer) ?CreatedContainer {
d.send_request_with_json('POST', urllib.parse('/v1.41/containers/create')?, c)?
head, res := d.read_response()?
if head.status_code != 201 {
data := json.decode(DockerError, res)?
return error(data.message)
}
data := json.decode(CreatedContainer, res)?
return data
}
pub fn (mut d DockerDaemon) start_container(id string) ? {
d.send_request('POST', urllib.parse('/v1.41/containers/$id/start')?)?
head, body := d.read_response() ?
if head.status_code != 204 {
data := json.decode(DockerError, body)?
return error(data.message)
}
}
// create_container creates a container defined by the given configuration. If
@ -67,6 +109,27 @@ pub mut:
end_time time.Time [skip]
}
pub fn (mut d DockerDaemon) inspect_container(id string) ?ContainerInspect {
d.send_request('GET', urllib.parse('/v1.41/containers/$id/json')?)?
head, body := d.read_response()?
if head.status_code != 200 {
data := json.decode(DockerError, body)?
return error(data.message)
}
mut data := json.decode(ContainerInspect, body)?
data.state.start_time = time.parse_rfc3339(data.state.start_time_str)?
if data.state.status == 'exited' {
data.state.end_time = time.parse_rfc3339(data.state.end_time_str)?
}
return data
}
// inspect_container returns the result of inspecting a container with a given
// ID.
pub fn inspect_container(id string) ?ContainerInspect {
@ -87,6 +150,17 @@ pub fn inspect_container(id string) ?ContainerInspect {
return data
}
pub fn (mut d DockerDaemon) remove_container(id string) ? {
d.send_request('DELETE', urllib.parse('/v1.41/containers/$id')?)?
head, body := d.read_response() ?
if head.status_code != 204 {
data := json.decode(DockerError, body)?
return error(data.message)
}
}
// remove_container removes a container with a given ID.
pub fn remove_container(id string) ?bool {
res := request('DELETE', urllib.parse('/v1.41/containers/$id')?)?

View File

@ -5,16 +5,12 @@ import net.urllib
import net.http
import json
const socket = '/var/run/docker.sock'
const buf_len = 1024
// send writes a request to the Docker socket, waits for a response & returns
// it.
fn send(req &string) ?http.Response {
// Open a connection to the socket
mut s := unix.connect_stream(docker.socket) or {
return error('Failed to connect to socket ${docker.socket}.')
mut s := unix.connect_stream(socket) or {
return error('Failed to connect to socket ${socket}.')
}
defer {
@ -25,19 +21,19 @@ fn send(req &string) ?http.Response {
}
// Write the request to the socket
s.write_string(req) or { return error('Failed to write request to socket ${docker.socket}.') }
s.write_string(req) or { return error('Failed to write request to socket ${socket}.') }
s.wait_for_write()?
mut c := 0
mut buf := []u8{len: docker.buf_len}
mut buf := []u8{len: buf_len}
mut res := []u8{}
for {
c = s.read(mut buf) or { return error('Failed to read data from socket ${docker.socket}.') }
c = s.read(mut buf) or { return error('Failed to read data from socket ${socket}.') }
res << buf[..c]
if c < docker.buf_len {
if c < buf_len {
break
}
}
@ -45,7 +41,7 @@ fn send(req &string) ?http.Response {
// After reading the first part of the response, we parse it into an HTTP
// response. If it isn't chunked, we return early with the data.
parsed := http.parse_response(res.bytestr()) or {
return error('Failed to parse HTTP response from socket ${docker.socket}.')
return error('Failed to parse HTTP response from socket ${socket}.')
}
if parsed.header.get(http.CommonHeader.transfer_encoding) or { '' } != 'chunked' {
@ -59,12 +55,10 @@ fn send(req &string) ?http.Response {
s.wait_for_write()?
for {
c = s.read(mut buf) or {
return error('Failed to read data from socket ${docker.socket}.')
}
c = s.read(mut buf) or { return error('Failed to read data from socket ${socket}.') }
res << buf[..c]
if c < docker.buf_len {
if c < buf_len {
break
}
}

114
src/docker/socket.v 100644
View File

@ -0,0 +1,114 @@
module docker
import net.unix
import io
import net.http
import strings
import net.urllib
import json
const (
socket = '/var/run/docker.sock'
buf_len = 10 * 1024
http_separator = [u8(`\r`), `\n`, `\r`, `\n`]
)
pub struct DockerDaemon {
mut:
socket &unix.StreamConn
reader &io.BufferedReader
}
pub fn new_conn() ?&DockerDaemon {
s := unix.connect_stream(docker.socket)?
d := &DockerDaemon{
socket: s
reader: io.new_buffered_reader(reader: s)
}
return d
}
pub fn (mut d DockerDaemon) send_request(method string, url urllib.URL) ? {
req := '$method $url.request_uri() HTTP/1.1\nHost: localhost\n\n'
d.socket.write_string(req)?
}
pub fn (mut d DockerDaemon) send_request_with_body(method string, url urllib.URL, content_type string, body string) ? {
req := '$method $url.request_uri() HTTP/1.1\nHost: localhost\nContent-Type: $content_type\nContent-Length: $body.len\n\n$body\n\n'
d.socket.write_string(req)?
}
pub fn (mut d DockerDaemon) send_request_with_json<T>(method string, url urllib.URL, data &T) ? {
body := json.encode(data)
return d.send_request_with_body(method, url, 'application/json', body)
}
// read_response_head consumes the socket's contents until it encounters
// '\r\n\r\n', after which it parses the response as an HTTP response.
pub fn (mut d DockerDaemon) read_response_head() ?http.Response {
mut c := 0
mut buf := []u8{len: 4}
mut res := []u8{}
for {
c = d.reader.read(mut buf)?
res << buf[..c]
mut i := 0
mut match_len := 0
for i + match_len < c {
if buf[i + match_len] == docker.http_separator[match_len] {
match_len += 1
} else {
i += match_len + 1
match_len = 0
}
}
if match_len == 4 {
break
} else if match_len > 0 {
mut buf2 := []u8{len: 4 - match_len}
c2 := d.reader.read(mut buf2)?
res << buf2[..c2]
if buf2 == docker.http_separator[match_len..] {
break
}
}
}
return http.parse_response(res.bytestr())
}
pub fn (mut d DockerDaemon) read_response_body(length int) ?string {
if length == 0 {
return ''
}
mut buf := []u8{len: docker.buf_len}
mut c := 0
mut builder := strings.new_builder(docker.buf_len)
for builder.len < length {
c = d.reader.read(mut buf) or { break }
builder.write(buf[..c])?
}
return builder.str()
}
pub fn (mut d DockerDaemon) read_response() ?(http.Response, string) {
head := d.read_response_head()?
content_length := head.header.get(http.CommonHeader.content_length)?.int()
res := d.read_response_body(content_length)?
return head, res
}

View File

@ -7,6 +7,7 @@ import build
import console.git
import console.logs
import cron
import docker
fn main() {
mut app := cli.Command{
@ -31,7 +32,7 @@ fn main() {
logs.cmd(),
]
}
app.setup()
app.parse(os.args)
return
}