forked from vieter-v/vieter
				
			refactor: ran vfmt with new defaults
feat(docker): started work on new implementationhash-on-upload
							parent
							
								
									473c7ec06b
								
							
						
					
					
						commit
						dd9958ea28
					
				| 
						 | 
					@ -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 {
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -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
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -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()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -31,7 +31,7 @@ fn main() {
 | 
				
			||||||
			logs.cmd(),
 | 
								logs.cmd(),
 | 
				
			||||||
		]
 | 
							]
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					 | 
				
			||||||
	app.setup()
 | 
						app.setup()
 | 
				
			||||||
	app.parse(os.args)
 | 
						app.parse(os.args)
 | 
				
			||||||
 | 
						return
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue