mirror of
https://github.com/hanzoai/runtime.git
synced 2026-10-30 18:23:13 +00:00
zeekay
ea80dc3638
- Add main Dockerfile for production builds - Add Dockerfile.runtime for secure code execution - Add Dockerfile.runtime.fixed with improved configuration - Include .dockerignore for build optimization - Add integration documentation and publishing guides - Update API and CLI components for better runtime support - Enhance sandbox and container management features
88 lines
2.0 KiB
Go
88 lines
2.0 KiB
Go
// Copyright 2025 Hanzo Industries Inc.
|
|
// SPDX-License-Identifier: AGPL-3.0
|
|
|
|
package apiclient
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
|
|
apiclient "github.com/hanzoai/apiclient"
|
|
"github.com/hanzoai/runtime/cli/auth"
|
|
"github.com/hanzoai/runtime/cli/config"
|
|
)
|
|
|
|
var apiClient *apiclient.APIClient
|
|
|
|
const RuntimeSourceHeader = "X-Runtime-Source"
|
|
|
|
func GetApiClient(profile *config.Profile, defaultHeaders map[string]string) (*apiclient.APIClient, error) {
|
|
c, err := config.GetConfig()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var activeProfile config.Profile
|
|
if profile == nil {
|
|
var err error
|
|
activeProfile, err = c.GetActiveProfile()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
} else {
|
|
activeProfile = *profile
|
|
}
|
|
|
|
if apiClient != nil && activeProfile.Api.Key == nil {
|
|
err := auth.RefreshTokenIfNeeded(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return apiClient, nil
|
|
}
|
|
|
|
var newApiClient *apiclient.APIClient
|
|
|
|
serverUrl := activeProfile.Api.Url
|
|
|
|
clientConfig := apiclient.NewConfiguration()
|
|
clientConfig.Servers = apiclient.ServerConfigurations{
|
|
{
|
|
URL: serverUrl,
|
|
},
|
|
}
|
|
|
|
if activeProfile.Api.Key != nil {
|
|
clientConfig.AddDefaultHeader("Authorization", "Bearer "+*activeProfile.Api.Key)
|
|
} else if activeProfile.Api.Token != nil {
|
|
clientConfig.AddDefaultHeader("Authorization", "Bearer "+activeProfile.Api.Token.AccessToken)
|
|
|
|
if activeProfile.ActiveOrganizationId != nil {
|
|
clientConfig.AddDefaultHeader("X-Runtime-Organization-ID", *activeProfile.ActiveOrganizationId)
|
|
}
|
|
}
|
|
|
|
clientConfig.AddDefaultHeader(RuntimeSourceHeader, "cli")
|
|
|
|
for headerKey, headerValue := range defaultHeaders {
|
|
clientConfig.AddDefaultHeader(headerKey, headerValue)
|
|
}
|
|
|
|
newApiClient = apiclient.NewAPIClient(clientConfig)
|
|
|
|
newApiClient.GetConfig().HTTPClient = &http.Client{
|
|
Transport: http.DefaultTransport,
|
|
}
|
|
|
|
if apiClient != nil && activeProfile.Api.Key == nil {
|
|
err = auth.RefreshTokenIfNeeded(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
apiClient = newApiClient
|
|
return apiClient, nil
|
|
}
|