Fixes

Author: ndiezel0

.gitignore

@@ -14,3 +14,5 @@ tags
 /.image.*
 Makefile.env
 *.iml
+CLAUDE.md
+.claude/

CLAUDE.md

@@ -23,8 +23,7 @@
     chargebacks = #{} :: #{hg_invoice_payment_chargeback:id() => hg_invoice_payment_chargeback:state()},
     adjustments = [] :: [hg_invoice_payment:adjustment()],
     recurrent_token :: undefined | dmsl_domain_thrift:'Token'(),
-    cascade_recurrent_tokens ::
-        undefined | #{dmsl_customer_thrift:'ProviderTerminalKey'() => dmsl_domain_thrift:'Token'()},
+    cascade_recurrent_tokens :: undefined | hg_customer_client:cascade_tokens(),
     opts :: undefined | hg_invoice_payment:opts(),
     repair_scenario :: undefined | hg_invoice_repair:scenario(),
     capture_data :: undefined | hg_invoice_payment:capture_data(),

apps/hellgate/include/hg_invoice_payment.hrl

@@ -94,6 +94,10 @@
     {invoice_payment_rec_token_acquired, #payproc_InvoicePaymentRecTokenAcquired{token = Token}}
 ).
 
+-define(cascade_tokens_loaded(Tokens),
+    {invoice_payment_cascade_tokens_loaded, #payproc_InvoicePaymentCascadeTokensLoaded{tokens = Tokens}}
+).
+
 -define(cash_changed(OldCash, NewCash),
     {invoice_payment_cash_changed, #payproc_InvoicePaymentCashChanged{old_cash = OldCash, new_cash = NewCash}}
 ).

apps/hellgate/include/payment_events.hrl

@@ -1,44 +1,110 @@
 -module(hg_customer_client).
 
 -include_lib("damsel/include/dmsl_customer_thrift.hrl").
+-include_lib("damsel/include/dmsl_domain_thrift.hrl").
 
--export([get_cascade_tokens/2]).
+-export([create_customer/1]).
+-export([get_recurrent_tokens/2]).
+-export([tokens_to_map/1]).
+-export([save_recurrent_token/6]).
+
+-export_type([cascade_tokens/0]).
 
 -type invoice_id() :: dmsl_domain_thrift:'InvoiceID'().
 -type payment_id() :: dmsl_domain_thrift:'InvoicePaymentID'().
 -type provider_terminal_key() :: dmsl_customer_thrift:'ProviderTerminalKey'().
 -type token() :: dmsl_domain_thrift:'Token'().
+-type recurrent_token() :: dmsl_customer_thrift:'RecurrentToken'().
 -type cascade_tokens() :: #{provider_terminal_key() => token()}.
 
 %%
 
--spec get_cascade_tokens(invoice_id(), payment_id()) -> cascade_tokens().
-get_cascade_tokens(InvoiceID, PaymentID) ->
-    try
-        get_cascade_tokens_(InvoiceID, PaymentID)
-    catch
-        error:_ -> #{}
-    end.
+-spec create_customer(dmsl_domain_thrift:'PartyConfigRef'()) -> dmsl_customer_thrift:'Customer'().
+create_customer(PartyConfigRef) ->
+    {ok, Customer} = call(customer_management, 'Create', {#customer_CustomerParams{party_ref = PartyConfigRef}}),
+    Customer.
 
-get_cascade_tokens_(InvoiceID, PaymentID) ->
-    case hg_woody_wrapper:call(customer_management, 'GetByParentPayment', {InvoiceID, PaymentID}) of
+-spec get_recurrent_tokens(invoice_id(), payment_id()) -> [recurrent_token()].
+get_recurrent_tokens(InvoiceID, PaymentID) ->
+    case call(customer_management, 'GetByParentPayment', {InvoiceID, PaymentID}) of
         {ok, #customer_CustomerState{bank_card_refs = BankCardRefs}} ->
-            lists:foldl(fun collect_bank_card_tokens/2, #{}, BankCardRefs);
+            lists:flatmap(fun collect_bank_card_tokens/1, BankCardRefs);
         {exception, #customer_CustomerNotFound{}} ->
-            #{};
+            [];
         {exception, #customer_InvalidRecurrentParent{}} ->
-            #{}
+            []
     end.
 
-collect_bank_card_tokens(#customer_BankCardRef{id = BankCardID}, Acc) ->
-    case hg_woody_wrapper:call(bank_card_storage, 'GetRecurrentTokens', {BankCardID}) of
-        {ok, Tokens} ->
-            lists:foldl(fun collect_recurrent_token/2, Acc, Tokens);
-        {exception, _} ->
-            Acc
-    end.
+-spec tokens_to_map([recurrent_token()]) -> cascade_tokens().
+tokens_to_map(Tokens) ->
+    lists:foldl(fun token_to_map_entry/2, #{}, Tokens).
+
+-spec save_recurrent_token(
+    dmsl_customer_thrift:'CustomerID'(),
+    token(),
+    dmsl_domain_thrift:'PaymentRoute'(),
+    token(),
+    invoice_id(),
+    payment_id()
+) -> ok.
+save_recurrent_token(
+    CustomerID,
+    BankCardToken,
+    #domain_PaymentRoute{provider = ProviderRef, terminal = TerminalRef},
+    RecToken,
+    InvoiceID,
+    PaymentID
+) ->
+    {ok, #customer_BankCard{id = BankCardID}} = call(
+        customer_management,
+        'AddBankCard',
+        {CustomerID, #customer_BankCardParams{bank_card_token = BankCardToken}}
+    ),
+    {ok, _} = call(
+        bank_card_storage,
+        'AddRecurrentToken',
+        {#customer_RecurrentTokenParams{
+            bank_card_id = BankCardID,
+            provider_ref = ProviderRef,
+            terminal_ref = TerminalRef,
+            token = RecToken
+        }}
+    ),
+    {ok, ok} = call(
+        customer_management,
+        'AddPayment',
+        {CustomerID, InvoiceID, PaymentID}
+    ),
+    ok.
+
+%%
+
+call(ServiceName, Function, Args) ->
+    Service = hg_proto:get_service(ServiceName),
+    Opts = hg_woody_wrapper:get_service_options(ServiceName),
+    WoodyContext =
+        try
+            hg_context:get_woody_context(hg_context:load())
+        catch
+            error:badarg -> woody_context:new()
+        end,
+    Request = {Service, Function, Args},
+    woody_client:call(
+        Request,
+        Opts#{
+            event_handler => {
+                scoper_woody_event_handler,
+                genlib_app:env(hellgate, scoper_event_handler_options, #{})
+            }
+        },
+        WoodyContext
+    ).
+
+collect_bank_card_tokens(#customer_BankCardRef{id = BankCardID}) ->
+    {ok, Tokens} = call(bank_card_storage, 'GetRecurrentTokens', {BankCardID}),
+    Tokens.
 
-collect_recurrent_token(
+token_to_map_entry(
     #customer_RecurrentToken{
         provider_ref = ProviderRef,
         terminal_ref = TerminalRef,