diff options
Diffstat (limited to 'src/subcommand.zig')
| -rw-r--r-- | src/subcommand.zig | 14 |
1 files changed, 12 insertions, 2 deletions
diff --git a/src/subcommand.zig b/src/subcommand.zig index b145d4b..bb46941 100644 --- a/src/subcommand.zig +++ b/src/subcommand.zig @@ -491,12 +491,22 @@ fn authLogin( defer panto.deinit(); const client = panto.httpClient(); + // The auth HTTP calls carry the identity headers of a provider that uses + // this session (e.g. Copilot's editor headers). 1:1 in practice; if no + // provider references it yet, send none. + const headers: []const panto.Header = blk: { + for (cfg.providers) |p| { + if (std.mem.eql(u8, p.auth_name, name)) break :blk p.extra_headers; + } + break :blk &.{}; + }; + var arena = std.heap.ArenaAllocator.init(allocator); defer arena.deinit(); const aa = arena.allocator(); var presenter = CliPresenter{ .io = io }; - const toks = panto.oauthLogin(aa, io, client, oauth, presenter.presenter()) catch |err| { + const toks = panto.oauthLogin(aa, io, client, oauth, presenter.presenter(), headers) catch |err| { try out.print("login failed: {t}\n", .{err}); return; }; @@ -508,7 +518,7 @@ fn authLogin( // immediately usable and we surface any exchange error during login. if (oauth.exchange) |exchange| { if (ts.access_token) |access| { - ts.exchange = panto.runExchange(aa, client, exchange, access) catch |err| blk: { + ts.exchange = panto.runExchange(aa, client, exchange, access, headers) catch |err| blk: { try out.print("note: token exchange failed ({t}); will retry on first use\n", .{err}); break :blk null; }; |