livebook-dev/livebook
1
1
Fork 0
mirror of https://github.com/livebook-dev/livebook.git synced 2025-09-13 08:24:22 +08:00
Code Issues Projects Releases Packages Wiki Activity

Fix billing status bug (#2956)

Browse source
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
1 changed files with 11 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
lib/livebook/hubs/team_client.ex
View file

@ -837,6 +837,14 @@ defmodule Livebook.Hubs.TeamClient do
put_in(state.hub, hub) put_in(state.hub, hub)
end 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 defp put_billing_status(hub, %LivebookProto.BillingStatus{} = status) do
put_in( put_in(
hub.billing_status, hub.billing_status,
@ -844,18 +852,10 @@ defmodule Livebook.Hubs.TeamClient do
) )
end 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( defp put_billing_status(
{:trialing, {:trialing,
%LivebookProto.BillingStatusTrialing{ %LivebookProto.BillingStatusTrialing{
trial_ends_at: %DateTime{} = trial_ends_at trial_ends_at: trial_ends_at
}} }}
) do ) do
%{type: :trialing, trial_ends_at: DateTime.from_unix!(trial_ends_at)} %{type: :trialing, trial_ends_at: DateTime.from_unix!(trial_ends_at)}
@ -864,7 +864,7 @@ defmodule Livebook.Hubs.TeamClient do
defp put_billing_status( defp put_billing_status(
{:trial_ended, {:trial_ended,
%LivebookProto.BillingStatusTrialEnded{ %LivebookProto.BillingStatusTrialEnded{
trial_ends_at: %DateTime{} = trial_ends_at trial_ends_at: trial_ends_at
}} }}
) do ) do
%{type: :trial_ended, trial_ends_at: DateTime.from_unix!(trial_ends_at)} %{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( defp put_billing_status(
{:canceling, {:canceling,
%LivebookProto.BillingStatusCanceling{ %LivebookProto.BillingStatusCanceling{
cancel_at: %DateTime{} = cancel_at cancel_at: cancel_at
}} }}
) do ) do
%{type: :canceling, cancel_at: DateTime.from_unix!(cancel_at)} %{type: :canceling, cancel_at: DateTime.from_unix!(cancel_at)}