feat: add VastAI instance management functions

- Added functions to manage VastAI instances:
- `attach_sshkey_to_instance`
- `stop_instance`
- `destroy_instance`
- `launch_instance`
- `start_instances`
- `start_instance`
- Updated example to demonstrate new functions.

Co-authored-by: mariobassem12 <mariobassem12@gmail.com>

examples/develop/vastai/vastai_example.vsh

@@ -22,3 +22,45 @@ create_instance_res := va.create_instance(
 	}
 )!
 println('create instance res: ${create_instance_res}')
+
+attach_sshkey_to_instance_res := va.attach_sshkey_to_instance(
+	id:     1
+	ssh_key: "ssh-rsa AAAA..."
+)!
+println('attach sshkey to instance res: ${attach_sshkey_to_instance_res}')
+
+stop_instance_res := va.stop_instance(
+	id:     1
+	state: "stopped"
+)!
+println('stop instance res: ${stop_instance_res}')
+
+destroy_instance_res := va.destroy_instance(
+	id:     1
+)!
+println('destroy instance res: ${destroy_instance_res}')
+
+// For some reason this method returns an error from their server, 500 ERROR
+// (request failed with code 500: {"error":"server_error","msg":"Something went wrong on the server"})
+launch_instance_res := va.launch_instance(
+	// Required
+	num_gpus: 1,
+	gpu_name: "RTX_3090",
+	image:    'vastai/tensorflow',
+	disk:     10,
+	region: "us-west",
+
+	// Optional
+	env: "user=7amada, home=/home/7amada",
+)!
+println('destroy instance res: ${launch_instance_res}')
+
+start_instances_res := va.start_instances(
+	ids:     [1, 2, 3]
+)!
+println('start instances res: ${start_instances_res}')
+
+start_instance_res := va.start_instance(
+	id:     1
+)!
+println('start instance res: ${start_instance_res}')

lib/clients/vastai/client.v

@@ -171,3 +171,190 @@ pub fn (mut va VastAI) create_instance(args CreateInstanceArgs) !CreateInstanceR
 
 	return instance_resp
 }
+
+@[params]
+pub struct StopInstanceArgs {
+pub:
+	id    int @[required]
+	state string
+}
+
+pub struct StopInstanceResponse {
+pub:
+	success bool
+	msg     string
+}
+
+// Stops a running container and updates its status to 'stopped'.
+pub fn (mut va VastAI) stop_instance(args StopInstanceArgs) !StopInstanceResponse {
+	// Get HTTP client
+	mut http_client := va.httpclient()!
+	payload := {
+		'state': args.state
+	}
+
+	// Make request
+	resp := http_client.send(
+		method: .put
+		prefix: '/instances/${args.id}/?'
+		data:   json.encode(payload)
+	)!
+
+	if resp.code != 200 {
+		return error('request failed with code ${resp.code}: ${resp.data}')
+	}
+
+	// Parse response
+	instance_resp := json.decode(StopInstanceResponse, resp.data)!
+
+	return instance_resp
+}
+
+@[params]
+pub struct DestroyInstanceArgs {
+pub:
+	id int @[required]
+}
+
+pub struct DestroyInstanceResponse {
+pub:
+	success bool
+	msg     string
+}
+
+// Destroys an instance.
+pub fn (mut va VastAI) destroy_instance(args DestroyInstanceArgs) !DestroyInstanceResponse {
+	// Get HTTP client
+	mut http_client := va.httpclient()!
+
+	// Make request
+	resp := http_client.send(
+		method: .delete
+		prefix: '/instances/${args.id}/?'
+	)!
+
+	if resp.code != 200 {
+		return error('request failed with code ${resp.code}: ${resp.data}')
+	}
+
+	// Parse response
+	instance_resp := json.decode(DestroyInstanceResponse, resp.data)!
+
+	return instance_resp
+}
+
+@[params]
+pub struct AttachSshKeyToInstanceArgs {
+pub:
+	id      int @[required]
+	ssh_key string
+}
+
+pub struct AttachSshKeyToInstanceResponse {
+pub:
+	success bool
+	msg     string
+}
+
+// Attach SSH Key to Instance
+pub fn (mut va VastAI) attach_sshkey_to_instance(args AttachSshKeyToInstanceArgs) !AttachSshKeyToInstanceResponse {
+	// Get HTTP client
+	mut http_client := va.httpclient()!
+	payload := {
+		'ssh_key': args.ssh_key
+	}
+
+	// Make request
+	resp := http_client.send(
+		method: .post
+		prefix: '/instances/${args.id}/ssh/?'
+		data:   json.encode(payload)
+	)!
+
+	if resp.code != 200 {
+		return error('request failed with code ${resp.code}: ${resp.data}')
+	}
+
+	// Parse response
+	instance_resp := json.decode(AttachSshKeyToInstanceResponse, resp.data)!
+
+	return instance_resp
+}
+
+@[params]
+pub struct LaunchInstanceArgs {
+pub:
+	num_gpus int    @[required]
+	gpu_name string @[required]
+	region   string @[required]
+	image    string @[required]
+	disk     int    @[required]
+	env      ?string
+	args     ?[]string
+}
+
+// Launch an instance, This endpoint launches an instance based on the specified parameters, selecting the first available offer that meets the criteria.
+pub fn (mut va VastAI) launch_instance(args LaunchInstanceArgs) !CreateInstanceResponse {
+	// Get HTTP client
+	mut http_client := va.httpclient()!
+
+	// Make request
+	resp := http_client.send(
+		method: .put
+		prefix: '/launch_instance/?'
+		data:   json.encode(args)
+	)!
+
+	if resp.code != 200 {
+		return error('request failed with code ${resp.code}: ${resp.data}')
+	}
+
+	// Parse response
+	instance_resp := json.decode(CreateInstanceResponse, resp.data)!
+
+	return instance_resp
+}
+
+@[params]
+pub struct StartInstancesArgs {
+pub:
+	ids []int @[json: 'IDs'; required]
+}
+
+pub struct StartInstancesResponse {
+pub:
+	success bool
+	msg     string
+}
+
+// Start Instances, Start a list of instances specified by their IDs.
+pub fn (mut va VastAI) start_instances(args StartInstancesArgs) !StartInstancesResponse {
+	// Get HTTP client
+	mut http_client := va.httpclient()!
+	// Make request
+	resp := http_client.send(
+		method: .post
+		prefix: '/instances/start'
+		data:   json.encode(args)
+	)!
+
+	if resp.code != 200 {
+		return error('request failed with code ${resp.code}: ${resp.data}')
+	}
+
+	// Parse response
+	instance_resp := json.decode(StartInstancesResponse, resp.data)!
+
+	return instance_resp
+}
+
+@[params]
+pub struct StartInstanceArgs {
+pub:
+	id int @[required]
+}
+
+// Start Instance, Start an instance specified by its ID.
+pub fn (mut va VastAI) start_instance(args StartInstanceArgs) !StartInstancesResponse {
+	return va.start_instances(StartInstancesArgs{ ids: [args.id] })
+}