add Kube controller #12

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

ilackarms merged 13 commits into main from kube-controller

Feb 11, 2025

controller/README.md

This file was deleted.

controller/go.mod

This file was deleted.

go/autogen/api/client.go

-Original file line number
+Diff line change
@@ -0,0 +1,116 @@
+    package api
+    import (
+    	"bytes"
+    	"encoding/json"
+    	"fmt"
+    	"net/http"
+    	"strings"
+    	"time"
+    )
+    type Client struct {
+    	BaseURL    string
+    	WSURL      string
+    	HTTPClient *http.Client
+    }
+    func NewClient(baseURL, wsURL string) *Client {
+    	// Ensure baseURL doesn't end with a slash
+    	baseURL = strings.TrimRight(baseURL, "/")
+    	return &Client{
+    		BaseURL: baseURL,
+    		WSURL:   wsURL,
+    		HTTPClient: &http.Client{
+    			Timeout: time.Second * 30,
+    		},
+    	}
+    }
+    func (c *Client) GetVersion() (string, error) {
+    	var result struct {
+    		Status  bool   `json:"status"`
+    		Message string `json:"message"`
+    		Data    struct {
+    			Version string `json:"version"`
+    		} `json:"data"`
+    	}
+    	err := c.doRequest("GET", "/version", nil, &result)
+    	if err != nil {
+    		return "", err
+    	}
+    	if !result.Status {
+    		return "", fmt.Errorf("api error: %s", result.Message)
+    	}
+    	return result.Data.Version, nil
+    }
+    func (c *Client) doRequest(method, path string, body interface{}, result interface{}) error {
+    	var bodyReader *bytes.Reader
+    	if body != nil {
+    		bodyBytes, err := json.Marshal(body)
+    		if err != nil {
+    			return fmt.Errorf("error marshaling request body: %w", err)
+    		}
+    		bodyReader = bytes.NewReader(bodyBytes)
+    	}
+    	// Ensure path starts with a slash
+    	if !strings.HasPrefix(path, "/") {
+    		path = "/" + path
+    	}
+    	url := c.BaseURL + path
+    	var req *http.Request
+    	var err error
+    	if bodyReader != nil {
+    		req, err = http.NewRequest(method, url, bodyReader)
+    	} else {
+    		req, err = http.NewRequest(method, url, nil)
+    	}
+    	if err != nil {
+    		return fmt.Errorf("error creating request: %w", err)
+    	}
+    	req.Header.Set("Content-Type", "application/json")
+    	resp, err := c.HTTPClient.Do(req)
+    	if err != nil {
+    		return fmt.Errorf("error making request: %w", err)
+    	}
+    	defer resp.Body.Close()
+    	if resp.StatusCode >= 400 {
+    		return fmt.Errorf("request failed with status: %s", resp.Status)
+    	}
+    	// Decode into APIResponse first
+    	var apiResp APIResponse
+    	if err := json.NewDecoder(resp.Body).Decode(&apiResp); err != nil {
+    		return fmt.Errorf("error decoding response: %w", err)
+    	}
+    	// Check response status
+    	if !apiResp.Status {
+    		return fmt.Errorf("api error: %s", apiResp.Message)
+    	}
+    	// If caller wants the result, marshal the Data field into their result type
+    	if result != nil {
+    		dataBytes, err := json.Marshal(apiResp.Data)
+    		if err != nil {
+    			return fmt.Errorf("error re-marshaling data: %w", err)
+    		}
+    		if err := json.Unmarshal(dataBytes, result); err != nil {
+    			return fmt.Errorf("error unmarshaling into result: %w", err)
+    		}
+    	}
+    	return nil
+    }

go/autogen/api/run.go

-Original file line number
+Diff line change
@@ -0,0 +1,37 @@
+    package api
+    import "fmt"
+    func (c *Client) CreateRun(req *CreateRunRequest) (*CreateRunResult, error) {
+    	var run CreateRunResult
+    	err := c.doRequest("POST", "/runs", req, &run)
+    	return &run, err
+    }
+    func (c *Client) GetRun(runID string) (*Run, error) {
+    	var run Run
+    	err := c.doRequest("GET", fmt.Sprintf("/runs/%s", runID), nil, &run)
+    	return &run, err
+    }
+    func (c *Client) ListRuns(userID string) ([]Run, error) {
+    	// Go through all sessions and then retrieve all runs for each session
+    	var sessions []Session
+    	err := c.doRequest("GET", fmt.Sprintf("/sessions/?user_id=%s", userID), nil, &sessions)
+    	if err != nil {
+    		return nil, err
+    	}
+    	// For each session, get the run information
+    	var runs []Run
+    	for _, session := range sessions {
+    		var sessionRuns SessionRuns
+    		err := c.doRequest("GET", fmt.Sprintf("/sessions/%d/runs/?user_id=%s", session.ID, userID), nil, &sessionRuns)
+    		if err != nil {
+    			fmt.Println("Error getting runs for session")
+    			return nil, err
+    		}
+    		runs = append(runs, sessionRuns.Runs...)
+    	}
+    	return runs, nil
+    }

go/autogen/api/session.go

-Original file line number
+Diff line change
@@ -0,0 +1,25 @@
+    package api
+    import "fmt"
+    func (c *Client) ListSessions(userID string) ([]Session, error) {
+    	var sessions []Session
+    	err := c.doRequest("GET", fmt.Sprintf("/sessions/?user_id=%s", userID), nil, &sessions)
+    	return sessions, err
+    }
+    func (c *Client) CreateSession(session *CreateSession) (*Session, error) {
+    	var result Session
+    	err := c.doRequest("POST", "/sessions/", session, &result)
+    	return &result, err
+    }
+    func (c *Client) GetSession(sessionID int, userID string) (*Session, error) {
+    	var session Session
+    	err := c.doRequest("GET", fmt.Sprintf("/sessions/%d?user_id=%s", sessionID, userID), nil, &session)
+    	return &session, err
+    }
+    func (c *Client) DeleteSession(sessionID int, userID string) error {
+    	return c.doRequest("DELETE", fmt.Sprintf("/sessions/%d?user_id=%s", sessionID, userID), nil, nil)
+    }

go/autogen/api/team.go

-Original file line number
+Diff line change
@@ -0,0 +1,32 @@
+    package api
+    import "fmt"
+    func (c *Client) ListTeams(userID string) ([]TeamResponse, error) {
+    	var teams []TeamResponse
+    	err := c.doRequest("GET", fmt.Sprintf("/teams/?user_id=%s", userID), nil, &teams)
+    	return teams, err
+    }
+    func (c *Client) CreateTeam(team *TeamResponse) error {
+    	return c.doRequest("POST", "/teams/", team, team)
+    }
+    func (c *Client) GetTeam(teamLabel string, userID string) (*TeamResponse, error) {
+    	allTeams, err := c.ListTeams(userID)
+    	if err != nil {
+    		return nil, err
+    	}
+    	for _, team := range allTeams {
+    		if team.Component.Label == teamLabel {
+    			return &team, nil
+    		}
+    	}
+    	return nil, nil
+    }
+    func (c *Client) DeleteTeam(teamID int, userID string) error {
+    	return c.doRequest("DELETE", fmt.Sprintf("/teams/%d?user_id=%s", teamID, userID), nil, nil)
+    }

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

add Kube controller #12

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!

Uh oh!

Uh oh!