Skip to content

feat: add Parseable Cloud API key login#110

Merged
nikhilsinhaparseable merged 7 commits into
parseablehq:mainfrom
pratik50:feat/cloud-api-key-login
Jul 20, 2026
Merged

feat: add Parseable Cloud API key login#110
nikhilsinhaparseable merged 7 commits into
parseablehq:mainfrom
pratik50:feat/cloud-api-key-login

Conversation

@pratik50

@pratik50 pratik50 commented Jun 26, 2026

Copy link
Copy Markdown
Contributor

Summary

Adds Parseable Cloud API key support to pb.

Users can now create a Cloud profile using either:

pb cloud profile add --api-key <key> --name prod --default

or interactive login:

pb login

Then select Parseable Cloud and paste API key.

What Changed

  • Added pb cloud profile add

  • Enabled Parseable Cloud option in interactive pb login

  • Added cloud-aware profile fields:

    • cloud
    • api_key
    • tenant_id
    • ingest_url
    • workspace_id
    • workspace_name
    • orchestrator_url
  • Added API key validation through orchestrator:

    • GET /api/v1/apikey/validate
    • Authorization: Bearer <api_key>
  • Added cloud request auth headers:

    • x-api-key: <api_key>
    • X-P-Tenant: <tenant_id>
  • Kept self-hosted BasicAuth/token profiles backward compatible

  • Config file writes with 0600 permissions because API key is stored locally

Summary by CodeRabbit

  • New Features

    • Added Parseable Cloud profile management via CLI (including device-code OAuth flow) and updated the login wizard for Cloud OAuth/API keys.
    • Introduced global --output/-o (text/json) with JSON-aware help; cloud, status, logout, and tail now support JSON output.
    • Enhanced tail streaming auth for Cloud vs self-hosted and added promql positional-argument overrides.
  • Bug Fixes

    • Standardized authentication handling across query, tail, PromQL, and saved queries, with improved Cloud OAuth session refresh behavior.
    • Improved discovery/statistics output consistency and “no results” messages; refined analytics to avoid private command data.
  • Tests

    • Added/updated tests for Cloud defaults, JSON status/logout, output/flag behavior, and Cloud token refresh/transport handling.

@coderabbitai

coderabbitai Bot commented Jun 26, 2026

Copy link
Copy Markdown

Review Change Stack

Note

Reviews paused

It looks like this branch is under active development. To avoid overwhelming you with review comments due to an influx of new commits, CodeRabbit has automatically paused this review. You can configure this behavior by changing the reviews.auto_review.auto_pause_after_reviewed_commits setting.

Use the following commands to manage reviews:

  • @coderabbitai resume to resume automatic reviews.
  • @coderabbitai review to trigger a single review.

Use the checkboxes below for quick actions:

  • ▶️ Resume reviews
  • ✅ Review completed - (🔄 Check again to review again)
📝 Walkthrough

Walkthrough

Adds Parseable Cloud device authorization and API-key profile management, expands profile authentication modes, centralizes authenticated requests, updates PromQL commands, and adds structured output support across root, profile, status, logout, and tail commands.

Changes

Cloud authentication and profile management

Layer / File(s) Summary
Cloud profile and authentication contracts
pkg/config/config.go, pkg/http/http.go, pkg/http/refresh.go
Profiles now store cloud credentials and auth metadata; shared HTTP helpers select authentication modes, refresh OAuth sessions, and migrate legacy API-key data.
Cloud profile creation and device login
cmd/cloud.go, cmd/login.go, main.go, cmd/cloud_profile_test.go
Cloud commands validate API keys, perform device authorization, construct workspace profiles, persist profiles, and register the cloud command tree.
Interactive cloud login wizard
pkg/model/login/login.go, pkg/model/login/login_test.go
The wizard adds cloud authentication selection, masked API-key entry, device-login state, and cloud-specific profile finalization.
Authentication-aware request paths
pkg/datasets/*, pkg/model/{promql,query,savedQueries}.go, cmd/{promql,queryList,tail}.go
Dataset, query, saved-query, PromQL, and tail requests use shared authentication handling, cloud-aware transport credentials, and gRPC metadata.
PromQL command contracts and rendering
cmd/promql.go, cmd/promql_test.go
PromQL commands accept positional stream-related arguments, render deterministic styled output, and handle empty or envelope-wrapped responses.
Structured CLI output and analytics
main.go, cmd/{profile,status,logout,tail}.go, pkg/analytics/*
Root help, profile, status, logout, and tail commands support structured or non-interactive output, while analytics events omit command arguments, flags, and errors.

Estimated code review effort: 5 (Critical) | ~90 minutes

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name Status Explanation Resolution
Docstring Coverage ⚠️ Warning Docstring coverage is 18.48% which is insufficient. The required threshold is 80.00%. Write docstrings for the functions missing them to satisfy the coverage threshold.
✅ Passed checks (4 passed)
Check name Status Explanation
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title check ✅ Passed The title is clear, concise, and accurately reflects the main cloud API key login feature added in this PR.
Linked Issues check ✅ Passed Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check ✅ Passed Check skipped because no linked issues were found for this pull request.
✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests

Comment @coderabbitai help to get the list of available commands.

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 4

🧹 Nitpick comments (1)
cmd/login.go (1)

60-77: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Duplicated cloud-profile construction.

cloudProfileFromAPIKey builds a config.Profile from the validation result identically to CloudProfileAddCmd.RunE in cmd/cloud.go (lines 96-105). Consider extracting a single helper (e.g. profileFromValidation(apiKey, orchestratorURL, result)) used by both paths so the field mapping stays in sync.

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@cmd/login.go` around lines 60 - 77, The cloud-profile construction logic is
duplicated between cloudProfileFromAPIKey and CloudProfileAddCmd.RunE, so
extract the shared mapping into a single helper such as profileFromValidation
that accepts the apiKey, orchestratorURL, and validation result and returns the
config.Profile. Update both call sites to use that helper so the field
assignments stay consistent in one place.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@cmd/cloud.go`:
- Around line 184-200: saveCloudProfile currently treats every
ReadConfigFromFile error as an empty config and then writes it back, which can
wipe existing profiles, and it also overwrites an existing profile when the
derived or provided name already exists. Update saveCloudProfile to distinguish
a missing config file from other read errors so non-fatal parse/read failures
are surfaced instead of replacing the config, and add an overwrite check before
assigning fileConfig.Profiles[name]. Use the existing stepConfirmReplace pattern
from the interactive login flow to warn or confirm when a profile name collision
would replace an existing entry, then only call WriteConfigToFile after the user
has explicitly accepted the overwrite.
- Around line 33-37: The default Cloud orchestrator URL is currently pointing at
the staging endpoint, so `defaultCloudOrchestratorURL` in `cmd/cloud.go` should
be changed to the production orchestrator unless staging is explicitly intended.
Update the default used by the Cloud validation flow, and make sure the
`envCloudOrchestratorURL` override still works so users can opt into a different
endpoint if needed.

In `@cmd/tail.go`:
- Around line 120-132: The tailAuthMetadata helper is missing bearer-token
handling and currently falls back to Basic auth for token-only profiles. Update
tailAuthMetadata to mirror the same auth precedence used by AddAuthHeaders:
handle profile.Token before the Basic fallback, returning an Authorization
header with Bearer <token> when a token is present, while keeping the existing
cloud x-api-key and tenant behavior unchanged.

In `@pkg/config/config.go`:
- Line 101: The config write path in the file-opening logic does not harden
permissions for already-existing files, so broader modes can persist after
writing sensitive values like api_key. Update the config save flow around
os.OpenFile in the config write function to explicitly set restrictive
permissions on the opened file after it is created/opened, using
file.Chmod(0600) or equivalent, so both new and existing config files end up
with 0600.

---

Nitpick comments:
In `@cmd/login.go`:
- Around line 60-77: The cloud-profile construction logic is duplicated between
cloudProfileFromAPIKey and CloudProfileAddCmd.RunE, so extract the shared
mapping into a single helper such as profileFromValidation that accepts the
apiKey, orchestratorURL, and validation result and returns the config.Profile.
Update both call sites to use that helper so the field assignments stay
consistent in one place.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Run ID: 19834b75-b4c7-47e5-8e5b-cdc99993cf39

📥 Commits

Reviewing files that changed from the base of the PR and between e9d8b1e and fcb4795.

📒 Files selected for processing (13)
  • cmd/cloud.go
  • cmd/login.go
  • cmd/promql.go
  • cmd/queryList.go
  • cmd/tail.go
  • main.go
  • pkg/config/config.go
  • pkg/datasets/datasets.go
  • pkg/http/http.go
  • pkg/model/login/login.go
  • pkg/model/promql.go
  • pkg/model/query.go
  • pkg/model/savedQueries.go

Comment thread cmd/cloud.go
Comment thread cmd/cloud.go
Comment thread cmd/tail.go Outdated
Comment thread pkg/config/config.go Outdated

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 11

Caution

Some comments are outside the diff and can’t be posted inline due to platform limitations.

⚠️ Outside diff range comments (2)
pkg/model/login/login.go (1)

579-593: 🎯 Functional Correctness | 🟡 Minor | ⚡ Quick win

Render the actual Cloud authentication mode.

OAuth browser profiles are currently labeled as Cloud API-key profiles, while Cloud API-key profiles render two AUTH rows.

Proposed fix
 		if m.Profile.Cloud {
+			auth := "Cloud API key (stored after validation)"
+			if m.CloudBrowserLogin {
+				auth = "Cloud OAuth (browser)"
+			}
 			b.WriteString("  " + labelStyle.Render("AUTH "))
-			b.WriteString(normalStyle.Render("Cloud API key (stored after validation)"))
+			b.WriteString(normalStyle.Render(auth))
 			b.WriteString("\n")
 		}
-		if m.Profile.APIKey != "" {
+		if !m.Profile.Cloud && m.Profile.APIKey != "" {
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@pkg/model/login/login.go` around lines 579 - 593, Update the profile
rendering logic around the Cloud and APIKey checks to distinguish OAuth browser
profiles from Cloud API-key profiles: render the Cloud authentication label
according to the actual authentication mode, and avoid emitting a second AUTH
row for Cloud API-key profiles. Preserve the existing username rendering and
stored-key messaging for non-Cloud profiles.
cmd/queryList.go (1)

53-69: 🎯 Functional Correctness | 🟠 Major | ⚡ Quick win

Propagate saved-query request failures.

When NewRequest rejects an invalid profile, fetchFilters prints the error to stdout and returns nil. JSON mode then emits [] and exits successfully, producing invalid mixed output and masking authentication failures.

Change fetchFilters to return ([]Item, error) and let RunE return the error without printing inside the data layer.

Also applies to: 181-191

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@cmd/queryList.go` around lines 53 - 69, The fetchFilters data path currently
suppresses request errors, causing JSON mode to output an empty array and
succeed. Update fetchFilters to return ([]Item, error), propagate NewRequest
failures without printing, and update RunE and all callers to handle the
returned error by returning it before formatting output; preserve successful
JSON and non-JSON behavior.
🧹 Nitpick comments (3)
cmd/dataset_test.go (1)

22-22: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Assert the authentication header in at least one handler.

Supplying APIKey only satisfies AuthMode; these tests still pass if x-api-key is never sent.

Proposed assertion
 server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+	if got := r.Header.Get("x-api-key"); got != "test-api-key" {
+		t.Fatalf("x-api-key mismatch: got %q", got)
+	}

Also applies to: 66-66

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@cmd/dataset_test.go` at line 22, Update the handlers tested in
cmd/dataset_test.go to assert that requests include the expected x-api-key
authentication header, using the configured "test-api-key" value. Add this
assertion in at least one handler, including the additionally referenced
handler, while preserving the existing test behavior.
cmd/promql.go (2)

384-393: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Repeated flag/positional-arg override boilerplate across ~7 subcommands.

The same pattern (val, _ := cmd.Flags().GetString(...); if len(args) > idx { val = args[idx] }) is duplicated in promqlLabelsCmd, promqlLabelValuesCmd, promqlSeriesCmd, promqlCardinalityLabelNamesCmd, promqlCardinalityLabelValuesCmd, promqlCardinalityActiveSeriesCmd, and promqlTSDBCmd. Consider extracting a small helper to centralize this.

♻️ Proposed helper and example usage
// argOrFlag returns args[idx] if present, otherwise the current value of the given string flag.
func argOrFlag(cmd *cobra.Command, args []string, idx int, flagName string) string {
	val, _ := cmd.Flags().GetString(flagName)
	if len(args) > idx {
		return args[idx]
	}
	return val
}
-		stream, _ := cmd.Flags().GetString("dataset")
-		if len(args) > 0 {
-			stream = args[0]
-		}
+		stream := argOrFlag(cmd, args, 0, "dataset")

Apply the same substitution at the other six call sites.

Also applies to: 438-448, 494-503, 569-578, 620-633, 717-732, 843-852

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@cmd/promql.go` around lines 384 - 393, The positional-argument-overrides-flag
logic is duplicated across the PromQL subcommands. Add an argOrFlag helper near
the command implementations, then update promqlLabelsCmd and the corresponding
promqlLabelValuesCmd, promqlSeriesCmd, promqlCardinalityLabelNamesCmd,
promqlCardinalityLabelValuesCmd, promqlCardinalityActiveSeriesCmd, and
promqlTSDBCmd call sites to use it with the appropriate argument index and flag
name.

620-633: 🎯 Functional Correctness | 🔵 Trivial | ⚡ Quick win

Positional order for label-values is reversed between sibling commands.

promql label-values takes [label_name] [stream] (Line 438), while promql cardinality label-values takes [stream] [label] (Line 620) — same subcommand name, swapped order, no type validation on either positional. A user transposing the args between the two contexts would silently query the wrong stream/label instead of getting an error.

Worth a deliberate decision (and doc note) on whether to align ordering with the top-level command or keep it consistent with the other cardinality subcommands (which are all stream-first).

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@cmd/promql.go` around lines 620 - 633, Deliberately resolve the positional
argument order mismatch between the top-level label-values command and the
cardinality label-values command. Update the cardinality label-values
definition, including its Use, Example, and RunE argument mapping around stream
and labelName, to follow the chosen ordering consistently; document the
supported order and preserve compatibility expectations where applicable.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@cmd/cloud.go`:
- Around line 216-218: Update the direct exchange call in
cloudProfileFromDirectCodeExchange’s caller to pass the explicit clerk session
token from the --clerk-session-token option instead of the browser token
currently supplied twice. Apply the same token-source correction to the
corresponding direct-exchange paths at the other referenced locations,
preserving browser-token behavior only when no explicit flag is provided.
- Line 41: Update cloudDefaultOrchestratorAuthToken to an empty value so the
access-token fallback is used instead of sending the placeholder bearer token.
Preserve the existing fallback behavior in the default browser login and
interactive pb login Cloud flow, allowing organization resolution without an
explicit token override.
- Around line 858-864: Update cloudClerkScriptURL to replace the moving `@latest`
Clerk SDK tag with one fixed, tested version in both the CDN fallback URL and
the host-based URL, keeping the existing path selection behavior unchanged.

In `@cmd/logout.go`:
- Around line 61-64: Ensure the logout flow remains non-interactive when --yes
is set: before calling selectLogoutProfile, detect the
no-default/multiple-profile case and return an ambiguity error or require an
explicit profile. Preserve normal selector behavior for interactive invocations
and update the existing condition around outputFormat accordingly.

In `@cmd/promql.go`:
- Around line 384-393: Add tests in cmd/promql_test.go covering optional
positional-argument parsing for promqlLabelsCmd, promqlLabelValuesCmd,
promqlSeriesCmd, promqlCardinalityLabelValuesCmd,
promqlCardinalityActiveSeriesCmd, and promqlTSDBCmd, verifying each command uses
the positional argument when provided and preserves the dataset flag/default
behavior otherwise.

In `@cmd/status.go`:
- Around line 35-39: Update the status command’s RunE flow to route preflight
failures such as missing configuration or an active profile through statusOutput
when the output format is JSON, while still returning an error so the command
exits nonzero. Preserve the existing human-readable error behavior for other
formats and use the established statusOutput rendering path.

In `@cmd/tail.go`:
- Around line 154-161: Update the Flight client credential setup in the tail
command to use TLS for config.AuthCloudAPIKey and config.AuthCloudOAuth
profiles, ensuring API keys and session cookies are transmitted over encrypted
gRPC. Preserve insecure.NewCredentials() only for the intentional self-hosted
path, and keep the existing metadata construction unchanged.

In `@pkg/config/config.go`:
- Around line 85-137: The authentication model currently conflates migrated
legacy token values with API keys. Update Profile, AuthMode, and AddAuthHeaders
to retain a distinct legacy token field and authentication mode, classify token
profiles separately from API-key profiles, and continue emitting Authorization:
Bearer for them. Preserve existing cloud, basic, and API-key validation behavior
while rejecting ambiguous combinations consistently.

In `@pkg/datasets/datasets.go`:
- Around line 116-123: Update the dataset-building loop around fetchDatasetType
to avoid serial HTTP requests for every name. Fetch dataset types with bounded
concurrency, ensuring all results are safely associated with their corresponding
Dataset and the client’s request limits are respected, while preserving the
existing behavior when a type lookup fails.

In `@pkg/http/http.go`:
- Around line 61-62: Update NewRequest to validate client.Profile before
constructing the request URL or invoking baseAPIURL, returning the existing
“profile is nil” error through the normal error path. Ensure
DefaultClient(nil).NewRequest(...) does not dereference the nil profile, while
preserving the existing AddAuthHeaders behavior for valid profiles.

In `@pkg/model/login/login.go`:
- Line 110: Configure apiKeyInput to use masked/no-echo input, matching the
existing password input behavior, while preserving its prompt and length limit.
Update the newInput invocation or input configuration associated with
apiKeyInput so pasted API keys are not displayed during entry.

---

Outside diff comments:
In `@cmd/queryList.go`:
- Around line 53-69: The fetchFilters data path currently suppresses request
errors, causing JSON mode to output an empty array and succeed. Update
fetchFilters to return ([]Item, error), propagate NewRequest failures without
printing, and update RunE and all callers to handle the returned error by
returning it before formatting output; preserve successful JSON and non-JSON
behavior.

In `@pkg/model/login/login.go`:
- Around line 579-593: Update the profile rendering logic around the Cloud and
APIKey checks to distinguish OAuth browser profiles from Cloud API-key profiles:
render the Cloud authentication label according to the actual authentication
mode, and avoid emitting a second AUTH row for Cloud API-key profiles. Preserve
the existing username rendering and stored-key messaging for non-Cloud profiles.

---

Nitpick comments:
In `@cmd/dataset_test.go`:
- Line 22: Update the handlers tested in cmd/dataset_test.go to assert that
requests include the expected x-api-key authentication header, using the
configured "test-api-key" value. Add this assertion in at least one handler,
including the additionally referenced handler, while preserving the existing
test behavior.

In `@cmd/promql.go`:
- Around line 384-393: The positional-argument-overrides-flag logic is
duplicated across the PromQL subcommands. Add an argOrFlag helper near the
command implementations, then update promqlLabelsCmd and the corresponding
promqlLabelValuesCmd, promqlSeriesCmd, promqlCardinalityLabelNamesCmd,
promqlCardinalityLabelValuesCmd, promqlCardinalityActiveSeriesCmd, and
promqlTSDBCmd call sites to use it with the appropriate argument index and flag
name.
- Around line 620-633: Deliberately resolve the positional argument order
mismatch between the top-level label-values command and the cardinality
label-values command. Update the cardinality label-values definition, including
its Use, Example, and RunE argument mapping around stream and labelName, to
follow the chosen ordering consistently; document the supported order and
preserve compatibility expectations where applicable.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Run ID: bf1c6043-814d-4416-9f93-e62991edae58

📥 Commits

Reviewing files that changed from the base of the PR and between fcb4795 and f0635f6.

📒 Files selected for processing (17)
  • cmd/cloud.go
  • cmd/dataset_test.go
  • cmd/login.go
  • cmd/logout.go
  • cmd/profile.go
  • cmd/promql.go
  • cmd/queryList.go
  • cmd/status.go
  • cmd/tail.go
  • main.go
  • pkg/config/config.go
  • pkg/datasets/datasets.go
  • pkg/http/http.go
  • pkg/model/login/login.go
  • pkg/model/promql.go
  • pkg/model/query.go
  • pkg/model/savedQueries.go
🚧 Files skipped from review as they are similar to previous changes (2)
  • cmd/login.go
  • pkg/model/promql.go

Comment thread cmd/cloud.go Outdated
Comment thread cmd/cloud.go Outdated
Comment thread cmd/cloud.go Outdated
Comment thread cmd/logout.go Outdated
Comment thread cmd/promql.go
Comment thread cmd/tail.go
Comment thread pkg/config/config.go
Comment thread pkg/datasets/datasets.go Outdated
Comment thread pkg/http/http.go
Comment thread pkg/model/login/login.go

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 7

Caution

Some comments are outside the diff and can’t be posted inline due to platform limitations.

⚠️ Outside diff range comments (1)
pkg/config/config.go (1)

183-194: 🗄️ Data Integrity & Integration | 🟠 Major | 🏗️ Heavy lift

Write the config atomically instead of truncating it in place.

A chmod failure after Line 183 already leaves the existing config empty, while a short write can leave partial TOML. Write a 0600 temporary file in the same directory, close/sync it, then atomically rename it over the destination.

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@pkg/config/config.go` around lines 183 - 194, Update the config-writing flow
around os.OpenFile and file.Write to use a 0600 temporary file in the
destination directory, write the complete TOML, handle short writes, sync and
close it, then atomically rename it over filePath. Apply permissions before
replacing the destination, and remove the in-place truncation path while
cleaning up the temporary file on failure.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@cmd/cloud.go`:
- Line 253: Restore optional default-profile handling: in cmd/cloud.go at lines
253-253, bind the cloud profile add command’s --default boolean and pass its
value to saveCloudProfile instead of hard-coding true; in
cmd/cloud_profile_test.go at lines 9-15, replace the absence assertion with
coverage for both preserving the existing default without the flag and updating
it when --default is provided.
- Around line 401-408: Update the device polling switch on oauthErr.Error to
handle "slow_down" by increasing interval by 5 seconds and continuing the
polling loop, while preserving the existing authorization_pending,
expired_token, and default failure behavior.
- Around line 68-72: Add a client ID field to cloudDeviceTokenRequest,
serialized as client_id, and populate it with the existing pb-cli client ID
wherever the device token exchange request is constructed. Keep the value
consistent with the client ID used during device-code issuance.

In `@cmd/profile.go`:
- Line 185: Update the profile-add flow around the fileConfig.DefaultProfile
assignment so adding a profile only sets DefaultProfile when no current default
exists. Preserve the existing active default for secondary profiles, and retain
explicit --default handling separately.

In `@cmd/promql_test.go`:
- Around line 85-96: Update the cardinality label-values and active-series test
cases for promqlCardinalityLabelValuesCmd and promqlCardinalityActiveSeriesCmd
to set conflicting --label and active-series selector flag values through
flagSetup, while keeping the expected query values sourced from the positional
arguments.

In `@pkg/analytics/analytics.go`:
- Line 157: Update the analytics delivery error handling around the existing
fmt.Println call to write the diagnostic to stderr instead of stdout, preserving
the current error message and behavior while keeping command JSON output clean.

In `@pkg/http/refresh.go`:
- Around line 43-45: Update the refresh flow around transport.refresh in the
request orchestrator to close the original response and return a wrapped error
when refresh or token persistence fails, rather than returning the response with
a nil error. Preserve the successful refresh path and include the underlying
failure for diagnosis.

---

Outside diff comments:
In `@pkg/config/config.go`:
- Around line 183-194: Update the config-writing flow around os.OpenFile and
file.Write to use a 0600 temporary file in the destination directory, write the
complete TOML, handle short writes, sync and close it, then atomically rename it
over filePath. Apply permissions before replacing the destination, and remove
the in-place truncation path while cleaning up the temporary file on failure.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Run ID: e2cf1f6b-de56-48ef-9b2b-72b4c20fc256

📥 Commits

Reviewing files that changed from the base of the PR and between f0635f6 and 19047df.

📒 Files selected for processing (23)
  • cmd/cloud.go
  • cmd/cloud_profile_test.go
  • cmd/login.go
  • cmd/logout.go
  • cmd/logout_test.go
  • cmd/profile.go
  • cmd/promql.go
  • cmd/promql_test.go
  • cmd/status.go
  • cmd/status_test.go
  • cmd/tail.go
  • cmd/tail_test.go
  • pkg/analytics/analytics.go
  • pkg/analytics/analytics_test.go
  • pkg/config/config.go
  • pkg/datasets/datasets.go
  • pkg/datasets/datasets_test.go
  • pkg/http/http.go
  • pkg/http/http_test.go
  • pkg/http/refresh.go
  • pkg/model/login/login.go
  • pkg/model/login/login_test.go
  • pkg/model/promql.go
🚧 Files skipped from review as they are similar to previous changes (6)
  • pkg/model/promql.go
  • cmd/logout.go
  • cmd/status.go
  • pkg/model/login/login.go
  • pkg/http/http.go
  • cmd/promql.go

Comment thread cmd/cloud.go
Comment thread cmd/cloud.go
Comment thread cmd/cloud.go
Comment thread cmd/profile.go
Comment thread cmd/promql_test.go Outdated
Comment thread pkg/analytics/analytics.go Outdated
Comment thread pkg/http/refresh.go

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

Caution

Some comments are outside the diff and can’t be posted inline due to platform limitations.

⚠️ Outside diff range comments (3)
main.go (3)

55-66: 🎯 Functional Correctness | 🟡 Minor | ⚡ Quick win

Honor --output when printing the version.

The version branch returns before validating rootOutputFormat, so pb --version --output=json emits plain text and pb --version --output=xml succeeds. Serialize the version in JSON or reject incompatible formats before printing.

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@main.go` around lines 55 - 66, Update the version branch in the RunE handler
to validate rootOutputFormat and honor --output: serialize Version and Commit as
JSON for json output, allow text output to use pb.PrintVersion, and return the
existing unsupported-format error for other values before printing.

256-265: 🎯 Functional Correctness | 🟠 Major | ⚡ Quick win

Do not shadow the global output format with per-command state.

configureParentOutput and newHelpCommand bind --output to separate local variables instead of resolving the root value. As a result, pb --output=json cloud and pb --output=json help cloud can still render text, while placing the flag after the command behaves differently. Use one shared output resolver or bind all paths to the same state.

Also applies to: 302-303, 314-405

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@main.go` around lines 256 - 265, Update configureParentOutput and
newHelpCommand to resolve --output through the shared root output state instead
of separate per-command variables. Ensure root flags, subcommand flags, and help
subcommands all use the same resolver so values such as JSON remain effective
regardless of flag placement.

314-405: 🎯 Functional Correctness | 🟠 Major | ⚡ Quick win

Skip profile setup for parent help/JSON output

pb sql --output=json still goes through combinedPreRun, so PreRunDefaultProfile can fail before the command renders help/JSON. Bypass the profile/auth pre-run for these help-only parent commands.

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@main.go` around lines 314 - 405, Update the parent-command execution flow
around configureParentOutput and the help command’s JSON/text handling so
help-only output does not invoke combinedPreRun or PreRunDefaultProfile. Ensure
pb sql --output=json and equivalent parent help requests render their output
without profile/auth setup, while preserving normal pre-run behavior for actual
command execution.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@main.go`:
- Around line 47-54: Update the profile command’s pre-run hook to use
analyticsPreRun instead of directly invoking analytics.CheckAndCreateULID, or
remove the hook if unnecessary. Ensure every command-level pre-run path respects
the analyticsEnabled gate defined alongside the root cli command.

---

Outside diff comments:
In `@main.go`:
- Around line 55-66: Update the version branch in the RunE handler to validate
rootOutputFormat and honor --output: serialize Version and Commit as JSON for
json output, allow text output to use pb.PrintVersion, and return the existing
unsupported-format error for other values before printing.
- Around line 256-265: Update configureParentOutput and newHelpCommand to
resolve --output through the shared root output state instead of separate
per-command variables. Ensure root flags, subcommand flags, and help subcommands
all use the same resolver so values such as JSON remain effective regardless of
flag placement.
- Around line 314-405: Update the parent-command execution flow around
configureParentOutput and the help command’s JSON/text handling so help-only
output does not invoke combinedPreRun or PreRunDefaultProfile. Ensure pb sql
--output=json and equivalent parent help requests render their output without
profile/auth setup, while preserving normal pre-run behavior for actual command
execution.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Run ID: 3c5a3188-838e-4cdc-b1d2-337924aa8736

📥 Commits

Reviewing files that changed from the base of the PR and between 19047df and 36fbedc.

📒 Files selected for processing (7)
  • cmd/cloud.go
  • cmd/cloud_profile_test.go
  • cmd/promql_test.go
  • main.go
  • pkg/analytics/analytics.go
  • pkg/http/http_test.go
  • pkg/http/refresh.go
💤 Files with no reviewable changes (2)
  • cmd/cloud_profile_test.go
  • cmd/cloud.go
🚧 Files skipped from review as they are similar to previous changes (4)
  • cmd/promql_test.go
  • pkg/http/refresh.go
  • pkg/http/http_test.go
  • pkg/analytics/analytics.go

Comment thread main.go
@pratik50 pratik50 self-assigned this Jul 20, 2026

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@pkg/config/defaults.go`:
- Around line 12-14: Replace the placeholder value in CloudOrchestratorURL with
the real staging orchestrator URL used by local/development builds. In
.github/workflows/release.yaml, validate PB_CLOUD_ORCHESTRATOR_URL as a
well-formed URL before proceeding, and preserve .goreleaser.yml’s linker
injection only for validated values.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: defaults

Review profile: CHILL

Plan: Pro

Run ID: 853ec77a-b812-4dd5-9dea-5b7a8e034bb3

📥 Commits

Reviewing files that changed from the base of the PR and between 36fbedc and 08e12aa.

📒 Files selected for processing (12)
  • .github/workflows/release.yaml
  • .goreleaser.yml
  • buildscripts/gen-ldflags.go
  • cmd/cloud.go
  • cmd/cluster.go
  • cmd/login.go
  • main.go
  • pkg/analytics/analytics.go
  • pkg/config/defaults.go
  • pkg/installer/installer.go
  • pkg/installer/uninstaller.go
  • pkg/model/login/login.go
🚧 Files skipped from review as they are similar to previous changes (5)
  • pkg/analytics/analytics.go
  • cmd/cloud.go
  • cmd/login.go
  • main.go
  • pkg/model/login/login.go

Comment thread pkg/config/defaults.go
@nikhilsinhaparseable
nikhilsinhaparseable merged commit 7f7218e into parseablehq:main Jul 20, 2026
3 checks passed
@github-actions github-actions Bot locked and limited conversation to collaborators Jul 20, 2026
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants