refactor: ran vfmt with new defaults

feat(docker): started work on new implementation
hash-on-upload
Jef Roosens 2022-05-13 21:28:24 +02:00
parent 473c7ec06b
commit dd9958ea28
Signed by: Jef Roosens
GPG Key ID: B75D4F293C7052DB
4 changed files with 88 additions and 30 deletions

View File

@ -3,17 +3,22 @@ module docker
import json import json
import net.urllib import net.urllib
import time import time
import net.http
struct Container { struct Container {
id string [json: Id] id string [json: Id]
names []string [json: Names] names []string [json: Names]
} }
// containers returns a list of all currently running containers pub fn (mut d DockerDaemon) containers() ?[]Container {
pub fn containers() ?[]Container { d.send_request('GET', urllib.parse('/v1.41/containers/json')?)?
res := request('GET', urllib.parse('/v1.41/containers/json')?)? res_header := d.read_response_head()?
content_length := res_header.header.get(http.CommonHeader.content_length)?.int()
res := d.read_response_body(content_length)?
return json.decode([]Container, res.text) or {} data := json.decode([]Container, res)?
return data
} }
pub struct NewContainer { pub struct NewContainer {

View File

@ -9,8 +9,8 @@ import json
// it. // it.
fn send(req &string) ?http.Response { fn send(req &string) ?http.Response {
// Open a connection to the socket // Open a connection to the socket
mut s := unix.connect_stream(docker.socket) or { mut s := unix.connect_stream(socket) or {
return error('Failed to connect to socket ${docker.socket}.') return error('Failed to connect to socket ${socket}.')
} }
defer { defer {
@ -21,19 +21,19 @@ fn send(req &string) ?http.Response {
} }
// Write the request to the socket // 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()? s.wait_for_write()?
mut c := 0 mut c := 0
mut buf := []u8{len: docker.buf_len} mut buf := []u8{len: buf_len}
mut res := []u8{} mut res := []u8{}
for { 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] res << buf[..c]
if c < docker.buf_len { if c < buf_len {
break break
} }
} }
@ -41,7 +41,7 @@ fn send(req &string) ?http.Response {
// After reading the first part of the response, we parse it into an HTTP // 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. // response. If it isn't chunked, we return early with the data.
parsed := http.parse_response(res.bytestr()) or { 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' { if parsed.header.get(http.CommonHeader.transfer_encoding) or { '' } != 'chunked' {
@ -55,12 +55,10 @@ fn send(req &string) ?http.Response {
s.wait_for_write()? s.wait_for_write()?
for { for {
c = s.read(mut buf) or { c = s.read(mut buf) or { return error('Failed to read data from socket ${socket}.') }
return error('Failed to read data from socket ${docker.socket}.')
}
res << buf[..c] res << buf[..c]
if c < docker.buf_len { if c < buf_len {
break break
} }
} }

View File

@ -3,10 +3,15 @@ module docker
import net.unix import net.unix
import io import io
import net.http import net.http
import strings
import net.urllib
import json
const socket = '/var/run/docker.sock' const (
socket = '/var/run/docker.sock'
const buf_len = 10 * 1024 buf_len = 10 * 1024
http_separator = [u8(`\r`), `\n`, `\r`, `\n`]
)
pub struct DockerDaemon { pub struct DockerDaemon {
mut: mut:
@ -14,34 +19,84 @@ mut:
reader &io.BufferedReader reader &io.BufferedReader
} }
pub fn new_conn() ?DockerDaemon { pub fn new_conn() ?&DockerDaemon {
s := unix.connect_stream(socket) ? s := unix.connect_stream(docker.socket)?
d := DockerDaemon{socket: s, reader: io.new_buffered_reader(reader: s)} d := &DockerDaemon{
socket: s
reader: io.new_buffered_reader(reader: s)
}
return d return d
} }
fn (mut d DockerDaemon) send_request(req &string) ? { pub fn (mut d DockerDaemon) send_request(method string, url urllib.URL) ? {
d.socket.write_string(req) ? req := '$method $url.request_uri() HTTP/1.1\nHost: localhost\n\n'
d.socket.wait_for_write() ?
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) request_with_json<T>(method string, url urllib.URL, data &T) ? {
body := json.encode(data)
return request_with_body(method, url, 'application/json', body)
} }
// read_response_head consumes the socket's contents until it encounters // read_response_head consumes the socket's contents until it encounters
// '\n\n', after which it parses the response as an HTTP response. // '\r\n\r\n', after which it parses the response as an HTTP response.
fn (mut d DockerDaemon) read_response_head() ?http.Response { pub fn (mut d DockerDaemon) read_response_head() ?http.Response {
mut c := 0 mut c := 0
mut buf := [buf_len]u8{len: docker.buf_len} mut buf := []u8{len: 4}
mut res := []u8{} mut res := []u8{}
for { for {
c = d.socket.read(mut buf) ? c = d.reader.read(mut buf)?
res << buf[..c] res << buf[..c]
if res#[-2..] == [u8(`\n`), `\n`] { 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 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()) return http.parse_response(res.bytestr())
} }
pub fn (mut d DockerDaemon) read_response_body(length int) ?string {
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()
}

View File

@ -31,7 +31,7 @@ fn main() {
logs.cmd(), logs.cmd(),
] ]
} }
app.setup() app.setup()
app.parse(os.args) app.parse(os.args)
return
} }