# frozen_string_literal: true
require_relative "customer"
require_relative "legacy_customer"
require_relative "polyfill"
class CustomerRepo
def initialize(redis: REDIS, db: DB, braintree: BRAINTREE)
@redis = redis
@db = db
@braintree = braintree
end
def find(customer_id)
@redis.get("jmp_customer_jid-#{customer_id}").then do |jid|
raise "No jid" unless jid
find_inner(customer_id, jid)
end
end
def find_by_jid(jid)
if jid.to_s =~ /\Acustomer_(.+)@jmp.chat\Z/
find($1)
else
@redis.get("jmp_customer_id-#{jid}").then { |customer_id|
raise "No customer id" unless customer_id
find_inner(customer_id, jid)
}.catch do
find_legacy_customer(jid)
end
end
end
def find_by_tel(tel)
@redis.get("catapult_jid-#{tel}").then do |jid|
find_by_jid(jid)
end
end
def create(jid)
@braintree.customer.create.then do |result|
raise "Braintree customer create failed" unless result.success?
cid = result.customer.id
@redis.msetnx(
"jmp_customer_id-#{jid}", cid, "jmp_customer_jid-#{cid}", jid
).then do |redis_result|
raise "Saving new customer to redis failed" unless redis_result == 1
Customer.new(cid, Blather::JID.new(jid))
end
end
end
protected
def find_legacy_customer(jid)
@redis.lindex("catapult_cred-#{jid}", 3).then do |tel|
raise "No customer" unless tel
LegacyCustomer.new(Blather::JID.new(jid), tel)
end
end
def hydrate_plan(customer_id, raw_customer)
raw_customer.dup.tap do |data|
data[:plan] = CustomerPlan.new(
customer_id,
plan: data.delete(:plan_name)&.then(&Plan.method(:for)),
expires_at: data.delete(:expires_at)
)
end
end
def find_inner(customer_id, jid)
result = @db.query_defer(<<~SQL, [customer_id])
SELECT COALESCE(balance,0) AS balance, plan_name, expires_at
FROM customer_plans LEFT JOIN balances USING (customer_id)
WHERE customer_id=$1 LIMIT 1
SQL
result.then do |rows|
data = hydrate_plan(customer_id, rows.first&.transform_keys(&:to_sym) || {})
Customer.new(customer_id, Blather::JID.new(jid), **data)
end
end
end