Fix billing status bug (#2956)

This commit is contained in:
Hugo Baraúna 2025-03-11 21:09:55 -03:00 committed by GitHub
parent 091118d6b3
commit 0c58516a4e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -837,6 +837,14 @@ defmodule Livebook.Hubs.TeamClient do
put_in(state.hub, hub)
end
# TODO: Remove when Billing is public
defp put_billing_status(hub, nil = _status) do
put_in(
hub.billing_status,
%{disabled: false, type: nil}
)
end
defp put_billing_status(hub, %LivebookProto.BillingStatus{} = status) do
put_in(
hub.billing_status,
@ -844,18 +852,10 @@ defmodule Livebook.Hubs.TeamClient do
)
end
# TODO: Remove when Billing is public
defp put_billing_status(hub, nil) do
put_in(
hub.billing_status,
%{disabled: false, type: nil}
)
end
defp put_billing_status(
{:trialing,
%LivebookProto.BillingStatusTrialing{
trial_ends_at: %DateTime{} = trial_ends_at
trial_ends_at: trial_ends_at
}}
) do
%{type: :trialing, trial_ends_at: DateTime.from_unix!(trial_ends_at)}
@ -864,7 +864,7 @@ defmodule Livebook.Hubs.TeamClient do
defp put_billing_status(
{:trial_ended,
%LivebookProto.BillingStatusTrialEnded{
trial_ends_at: %DateTime{} = trial_ends_at
trial_ends_at: trial_ends_at
}}
) do
%{type: :trial_ended, trial_ends_at: DateTime.from_unix!(trial_ends_at)}
@ -873,7 +873,7 @@ defmodule Livebook.Hubs.TeamClient do
defp put_billing_status(
{:canceling,
%LivebookProto.BillingStatusCanceling{
cancel_at: %DateTime{} = cancel_at
cancel_at: cancel_at
}}
) do
%{type: :canceling, cancel_at: DateTime.from_unix!(cancel_at)}