1# frozen_string_literal: true
   2
   3require "pg/em/connection_pool"
   4require "bandwidth"
   5require "bigdecimal"
   6require "blather/client/dsl"
   7require "date"
   8require "dhall"
   9require "em-hiredis"
  10require "em_promise"
  11require "ougai"
  12require "ruby-bandwidth-iris"
  13require "sentry-ruby"
  14require "statsd-instrument"
  15
  16require_relative "lib/background_log"
  17require_relative "lib/utils"
  18
  19$stdout.sync = true
  20LOG = Ougai::Logger.new(BackgroundLog.new($stdout))
  21LOG.level = ENV.fetch("LOG_LEVEL", "info")
  22LOG.formatter = Ougai::Formatters::Readable.new(
  23	nil,
  24	nil,
  25	plain: !$stdout.isatty
  26)
  27Blather.logger = LOG
  28EM::Hiredis.logger = LOG
  29StatsD.logger = LOG
  30LOG.info "Starting"
  31
  32def log
  33	Thread.current[:log] || LOG
  34end
  35
  36Sentry.init do |config|
  37	config.logger = LOG
  38	config.breadcrumbs_logger = [:sentry_logger]
  39end
  40
  41CONFIG = Dhall::Coder
  42	.new(safe: Dhall::Coder::JSON_LIKE + [Symbol, Proc])
  43	.load(
  44		"(#{ARGV[0]}) : #{__dir__}/config-schema.dhall",
  45		transform_keys: ->(k) { k&.to_sym },
  46		timeout: 30
  47	)
  48WEB_LISTEN =
  49	if CONFIG[:web].is_a?(Hash)
  50		[CONFIG[:web][:interface], CONFIG[:web][:port]]
  51	else
  52		[CONFIG[:web]]
  53	end
  54
  55singleton_class.class_eval do
  56	include Blather::DSL
  57	Blather::DSL.append_features(self)
  58end
  59
  60require_relative "lib/session_manager"
  61
  62IQ_MANAGER = SessionManager.new(self, :id)
  63COMMAND_MANAGER = SessionManager.new(
  64	self,
  65	:sessionid,
  66	timeout: 60 * 60,
  67	error_if: ->(s) { s.cancel? }
  68)
  69
  70require_relative "lib/polyfill"
  71require_relative "lib/alt_top_up_form"
  72require_relative "lib/admin_command"
  73require_relative "lib/backend_sgx"
  74require_relative "lib/bwmsgsv2_repo"
  75require_relative "lib/bandwidth_iris_patch"
  76require_relative "lib/bandwidth_tn_order"
  77require_relative "lib/bandwidth_tn_repo"
  78require_relative "lib/btc_sell_prices"
  79require_relative "lib/buy_account_credit_form"
  80require_relative "lib/configure_calls_form"
  81require_relative "lib/command"
  82require_relative "lib/command_list"
  83require_relative "lib/customer"
  84require_relative "lib/customer_info"
  85require_relative "lib/customer_info_form"
  86require_relative "lib/customer_repo"
  87require_relative "lib/dummy_command"
  88require_relative "lib/db_notification"
  89require_relative "lib/electrum"
  90require_relative "lib/empty_repo"
  91require_relative "lib/expiring_lock"
  92require_relative "lib/em"
  93require_relative "lib/form_to_h"
  94require_relative "lib/low_balance"
  95require_relative "lib/port_in_order"
  96require_relative "lib/patches_for_sentry"
  97require_relative "lib/payment_methods"
  98require_relative "lib/paypal_done"
  99require_relative "lib/postgres"
 100require_relative "lib/reachability_form"
 101require_relative "lib/reachability_repo"
 102require_relative "lib/registration"
 103require_relative "lib/transaction"
 104require_relative "lib/tel_selections"
 105require_relative "lib/sim_repo"
 106require_relative "lib/sim_order"
 107require_relative "lib/edit_sim_nicknames"
 108require_relative "lib/snikket"
 109require_relative "lib/welcome_message"
 110require_relative "web"
 111require_relative "lib/statsd"
 112
 113ELECTRUM = Electrum.new(**CONFIG[:electrum])
 114ELECTRUM_BCH = Electrum.new(**CONFIG[:electrum_bch])
 115
 116LOG.info "Loading scripts from #{__dir__}/redis_lua"
 117EM::Hiredis::Client.load_scripts_from("#{__dir__}/redis_lua")
 118
 119Faraday.default_adapter = :em_synchrony
 120BandwidthIris::Client.global_options = {
 121	account_id: CONFIG[:creds][:account],
 122	username: CONFIG[:creds][:username],
 123	password: CONFIG[:creds][:password]
 124}
 125BANDWIDTH_VOICE = Bandwidth::Client.new(
 126	voice_basic_auth_user_name: CONFIG[:creds][:username],
 127	voice_basic_auth_password: CONFIG[:creds][:password]
 128).voice_client.client
 129
 130class AuthError < StandardError; end
 131
 132require_relative "lib/async_braintree"
 133BRAINTREE = AsyncBraintree.new(**CONFIG[:braintree])
 134
 135def panic(e, hub=nil)
 136	log.fatal(
 137		"Error raised during event loop: #{e.class}",
 138		e
 139	)
 140	if e.is_a?(::Exception)
 141		(hub || Sentry).capture_exception(e, hint: { background: false })
 142	else
 143		(hub || Sentry).capture_message(e.to_s, hint: { background: false })
 144	end
 145	exit 1
 146end
 147
 148EM.error_handler(&method(:panic))
 149
 150require_relative "lib/blather_client"
 151@client = BlatherClient.new
 152
 153setup(
 154	CONFIG[:component][:jid],
 155	CONFIG[:component][:secret],
 156	CONFIG[:server][:host],
 157	CONFIG[:server][:port],
 158	nil,
 159	nil,
 160	async: true
 161)
 162
 163# Infer anything we might have been notified about while we were down
 164def catchup_notify_low_balance(db)
 165	db.query(<<~SQL).each do |c|
 166		SELECT customer_id
 167		FROM balances INNER JOIN customer_plans USING (customer_id)
 168		WHERE balance < 5 AND expires_at > LOCALTIMESTAMP
 169	SQL
 170		db.query("SELECT pg_notify('low_balance', $1)", c.values)
 171	end
 172end
 173
 174def catchup_notify_possible_renewal(db)
 175	db.query(<<~SQL).each do |c|
 176		SELECT customer_id
 177		FROM customer_plans INNER JOIN balances USING (customer_id)
 178		WHERE
 179			expires_at < LOCALTIMESTAMP
 180			AND expires_at >= LOCALTIMESTAMP - INTERVAL '3 months'
 181			AND balance >= 5
 182	SQL
 183		db.query("SELECT pg_notify('possible_renewal', $1)", c.values)
 184	end
 185end
 186
 187def setup_sentry_scope(name)
 188	Sentry.clone_hub_to_current_thread
 189	Sentry.with_scope do |scope|
 190		scope.clear_breadcrumbs
 191		scope.set_transaction_name(name)
 192		Thread.current[:log] = ::LOG.child(transaction: scope.transaction_name)
 193		yield scope
 194	end
 195end
 196
 197def poll_for_notify(db, repo)
 198	db.wait_for_notify_defer.then { |notify|
 199		setup_sentry_scope("DB NOTIFY") do
 200			repo.find(notify[:extra]).then { |customer|
 201				DbNotification.for(notify, customer, repo)
 202			}.then(&:call).catch { |e|
 203				log.error("Error during poll_for_notify", e)
 204				Sentry.capture_exception(e)
 205			}.sync
 206		end
 207	}.then { EM.add_timer(0.5) { poll_for_notify(db, repo) } }
 208end
 209
 210def load_plans_to_db!
 211	DB.transaction do
 212		DB.exec("TRUNCATE plans")
 213		CONFIG[:plans].each do |plan|
 214			DB.exec("INSERT INTO plans VALUES ($1)", [plan.to_json])
 215		end
 216	end
 217end
 218
 219when_ready do
 220	log.info "Ready"
 221	BLATHER = self
 222	REDIS = EM::Hiredis.connect
 223	MEMCACHE = EM::P::Memcache.connect
 224	BTC_SELL_PRICES = BTCSellPrices.new(REDIS, CONFIG[:oxr_app_id])
 225	BCH_SELL_PRICES = BCHSellPrices.new(REDIS, CONFIG[:oxr_app_id])
 226	DB = Postgres.connect(dbname: "jmp", size: 5)
 227	TEL_SELECTIONS = TelSelections.new
 228
 229	EMPromise.resolve(nil).then {
 230		conn = DB.acquire
 231		conn.query("LISTEN low_balance")
 232		conn.query("LISTEN possible_renewal")
 233		catchup_notify_low_balance(conn)
 234		catchup_notify_possible_renewal(conn)
 235
 236		repo = CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 237		poll_for_notify(conn, repo)
 238	}.catch(&method(:panic))
 239
 240	load_plans_to_db!
 241
 242	EM.add_periodic_timer(3600) do
 243		DB.finish # Clear idle connections
 244		ping = Blather::Stanza::Iq::Ping.new(:get, CONFIG[:server][:host])
 245		ping.from = CONFIG[:component][:jid]
 246		self << ping
 247	end
 248
 249	Web.run(LOG.child, *WEB_LISTEN)
 250end
 251
 252message to: /\Aaccount@/, body: /./ do |m|
 253	StatsD.increment("deprecated_account_bot")
 254
 255	self << m.reply.tap { |out|
 256		out.body = "This bot is deprecated. Please talk to xmpp:cheogram.com"
 257	}
 258end
 259
 260FROM_BACKEND =
 261	"(?:#{([CONFIG[:sgx]] + CONFIG[:sgx_creds].keys)
 262		.map(&Regexp.method(:escape)).join('|')})"
 263
 264before(
 265	:iq,
 266	type: [:error, :result],
 267	to: /\Acustomer_/,
 268	from: /(\A|@)#{FROM_BACKEND}(\/|\Z)/
 269) { |iq| halt if IQ_MANAGER.fulfill(iq) }
 270
 271before nil, to: /\Acustomer_/, from: /(\A|@)#{FROM_BACKEND}(\/|\Z)/ do |s|
 272	StatsD.increment("stanza_customer")
 273
 274	Sentry.get_current_scope.set_transaction_name("stanza_customer")
 275	CustomerRepo.new(set_user: Sentry.method(:set_user)).find(
 276		s.to.node.delete_prefix("customer_")
 277	).then do |customer|
 278		ReachabilityRepo::SMS.new
 279			.find(customer, s.from.node, stanza: s).then do |reach|
 280				reach.filter do
 281					customer.stanza_to(s)
 282				end
 283			end
 284	end
 285
 286	halt
 287end
 288
 289ADDRESSES_NS = "http://jabber.org/protocol/address"
 290message(
 291	to: /\A#{CONFIG[:component][:jid]}\Z/,
 292	from: /(\A|@)#{FROM_BACKEND}(\/|\Z)/
 293) do |m|
 294	StatsD.increment("inbound_group_text")
 295	Sentry.get_current_scope.set_transaction_name("inbound_group_text")
 296	log.info "Possible group text #{m.from}"
 297
 298	address = m.find("ns:addresses", ns: ADDRESSES_NS).first
 299		&.find("ns:address", ns: ADDRESSES_NS)
 300		&.find { |el| el["jid"].to_s.start_with?("customer_") }
 301	pass unless address
 302
 303	CustomerRepo
 304		.new(set_user: Sentry.method(:set_user))
 305		.find_by_jid(address["jid"]).then { |customer|
 306			m.from = m.from.with(domain: CONFIG[:component][:jid])
 307			m.to = m.to.with(domain: customer.jid.domain)
 308			address["jid"] = customer.jid.to_s
 309			BLATHER << m
 310		}.catch_only(CustomerRepo::NotFound) { |e|
 311			BLATHER << m.as_error("forbidden", :auth, e.message)
 312		}
 313end
 314
 315# Ignore groupchat messages
 316# Especially if we have the component join MUC for notifications
 317message(type: :groupchat) { true }
 318
 319def billable_message(m)
 320	b = m.body
 321	b && !b.empty? || m.find("ns:x", ns: OOB.registered_ns).first
 322end
 323
 324class OverLimit < StandardError
 325	def initialize(customer, usage)
 326		super("Please contact support")
 327		@customer = customer
 328		@usage = usage
 329	end
 330
 331	def notify_admin
 332		ExpiringLock.new("jmp_usage_notify-#{@customer.customer_id}").with do
 333			BLATHER.join(CONFIG[:notify_admin], "sgx-jmp")
 334			BLATHER.say(
 335				CONFIG[:notify_admin], "#{@customer.customer_id} has used " \
 336				"#{@usage} messages today", :groupchat
 337			)
 338		end
 339	end
 340end
 341
 342class CustomerExpired < StandardError; end
 343
 344CONFIG[:direct_targets].each do |(tel, jid)|
 345	customer_repo = CustomerRepo.new(
 346		sgx_repo: TrivialBackendSgxRepo.new(jid: jid),
 347		set_user: Sentry.method(:set_user)
 348	)
 349
 350	message to: /\A#{Regexp.escape(tel)}@#{CONFIG[:component][:jid]}\/?/ do |m|
 351		customer_repo.find_by_jid(m.from.stripped).then { |customer|
 352			customer.stanza_from(m)
 353		}.catch_only(CustomerRepo::NotFound) {
 354			# This should not happen, but let's still get the message
 355			# to support at least if it does
 356			m.from = ProxiedJID.proxy(m.from, CONFIG[:component][:jid])
 357			m.to = jid
 358			BLATHER << m
 359		}
 360	end
 361end
 362
 363CONFIG[:direct_sources].each do |(jid, tel)|
 364	customer_repo = CustomerRepo.new(
 365		sgx_repo: TrivialBackendSgxRepo.new(jid: jid),
 366		set_user: Sentry.method(:set_user)
 367	)
 368	message to: /\Acustomer_/, from: /\A#{Regexp.escape(jid)}\/?/ do |m|
 369		customer_repo.find(m.to.node.delete_prefix("customer_")).then { |customer|
 370			m.from = "#{tel}@sgx-jmp" # stanza_to will fix domain
 371			customer.stanza_to(m)
 372		}.catch_only(CustomerRepo::NotFound) { |e|
 373			BLATHER << m.as_error("item-not-found", :cancel, e.message)
 374		}
 375	end
 376end
 377
 378def find_from_and_to_customer(from, to)
 379	(
 380		# TODO: group text?
 381		to.node ? CustomerRepo.new.find_by_tel(to.node) : EMPromise.resolve(nil)
 382	).catch_only(CustomerRepo::NotFound) { nil }.then { |target_customer|
 383		sgx_repo = target_customer ? Bwmsgsv2Repo.new : TrivialBackendSgxRepo.new
 384		EMPromise.all([
 385			CustomerRepo.new(set_user: Sentry.method(:set_user), sgx_repo: sgx_repo)
 386				.find_by_jid(from.stripped),
 387			target_customer
 388		])
 389	}
 390end
 391
 392message do |m|
 393	StatsD.increment("message")
 394
 395	today = Time.now.utc.to_date
 396	find_from_and_to_customer(m.from, m.to).then { |(customer, target_customer)|
 397		if target_customer && customer.registered?
 398			m.from = "#{customer.registered?.phone}@sgx-jmp"
 399			next target_customer.stanza_to(m)
 400		end
 401
 402		next customer.stanza_from(m) unless billable_message(m)
 403
 404		if customer.plan_name && !customer.active?
 405			raise CustomerExpired, "Your account is expired, please top up"
 406		end
 407
 408		EMPromise.all([
 409			TrustLevelRepo.new.find(customer),
 410			customer.message_usage((today..today))
 411		]).then { |(tl, usage)|
 412			raise OverLimit.new(customer, usage) unless tl.send_message?(usage)
 413		}.then do
 414			EMPromise.all([customer.incr_message_usage, customer.stanza_from(m)])
 415		end
 416	}.catch_only(OverLimit) { |e|
 417		e.notify_admin
 418		BLATHER << m.as_error("policy-violation", :wait, e.message)
 419	}.catch_only(CustomerRepo::NotFound, CustomerExpired) { |e|
 420		BLATHER << m.as_error("forbidden", :auth, e.message)
 421	}
 422end
 423
 424disco_info to: Blather::JID.new(CONFIG[:component][:jid]) do |iq|
 425	reply = iq.reply
 426	reply.identities = [{
 427		name: "JMP.chat",
 428		type: "sms",
 429		category: "gateway"
 430	}]
 431	reply.features = [
 432		"http://jabber.org/protocol/disco#info",
 433		"http://jabber.org/protocol/commands"
 434	]
 435	form = Blather::Stanza::X.find_or_create(reply.query)
 436	form.type = "result"
 437	form.fields = [
 438		{
 439			var: "FORM_TYPE",
 440			type: "hidden",
 441			value: "http://jabber.org/network/serverinfo"
 442		}
 443	] + CONFIG[:xep0157]
 444	self << reply
 445end
 446
 447disco_info do |iq|
 448	reply = iq.reply
 449	reply.identities = [{
 450		name: "JMP.chat",
 451		type: "sms",
 452		category: "client"
 453	}]
 454	reply.features = [
 455		"urn:xmpp:receipts"
 456	]
 457	self << reply
 458end
 459
 460disco_items(
 461	to: Blather::JID.new(CONFIG[:component][:jid]),
 462	node: "http://jabber.org/protocol/commands"
 463) do |iq|
 464	StatsD.increment("command_list")
 465
 466	reply = iq.reply
 467	reply.node = "http://jabber.org/protocol/commands"
 468
 469	CustomerRepo.new(
 470		sgx_repo: Bwmsgsv2Repo.new,
 471		set_user: Sentry.method(:set_user)
 472	).find_by_jid(
 473		iq.from.stripped
 474	).catch {
 475		nil
 476	}.then { |customer|
 477		CommandList.for(customer, iq.from)
 478	}.then { |list|
 479		reply.items = list.map { |item|
 480			Blather::Stanza::DiscoItems::Item.new(
 481				iq.to,
 482				item[:node],
 483				item[:name]
 484			)
 485		}
 486		self << reply
 487	}
 488end
 489
 490iq "/iq/ns:services", ns: "urn:xmpp:extdisco:2" do |iq|
 491	StatsD.increment("extdisco")
 492
 493	reply = iq.reply
 494	reply << Nokogiri::XML::Builder.new {
 495		services(xmlns: "urn:xmpp:extdisco:2") do
 496			service(
 497				type: "sip",
 498				host: CONFIG[:sip_host]
 499			)
 500		end
 501	}.doc.root
 502
 503	self << reply
 504end
 505
 506Command.new(
 507	"jabber:iq:register",
 508	"Register",
 509	list_for: ->(*) { true },
 510	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 511) {
 512	google_play_userid = if Command.execution.iq.from.domain == "cheogram.com"
 513		Command.execution.iq.command.find(
 514			"./ns:userId", ns: "https://ns.cheogram.com/google-play"
 515		)&.first&.content
 516	end
 517	Command.customer.catch_only(CustomerRepo::NotFound) {
 518		Sentry.add_breadcrumb(Sentry::Breadcrumb.new(message: "Customer.create"))
 519		Command.execution.customer_repo.create(Command.execution.iq.from.stripped)
 520	}.then { |customer|
 521		Sentry.add_breadcrumb(Sentry::Breadcrumb.new(message: "Registration.for"))
 522		Registration.for(customer, google_play_userid, TEL_SELECTIONS).then(&:write)
 523	}.then {
 524		StatsD.increment("registration.completed")
 525	}.catch_only(Command::Execution::FinalStanza) do |e|
 526		StatsD.increment("registration.completed")
 527		EMPromise.reject(e)
 528	end
 529}.register(self).then(&CommandList.method(:register))
 530
 531Command.new(
 532	"info",
 533	"👤 Show Account Info",
 534	list_for: ->(*) { true },
 535	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 536) {
 537	Command.customer.then(&CustomerInfo.method(:for)).then do |info|
 538		Command.finish do |reply|
 539			reply.command << info.form
 540		end
 541	end
 542}.register(self).then(&CommandList.method(:register))
 543
 544Command.new(
 545	"cdrs",
 546	"📲 Show Call Logs"
 547) {
 548	report_for = ((Date.today << 1)..Date.today)
 549
 550	Command.customer.then { |customer|
 551		CDRRepo.new.find_range(customer, report_for)
 552	}.then do |cdrs|
 553		Command.finish do |reply|
 554			reply.command << FormTemplate.render("customer_cdr", cdrs: cdrs)
 555		end
 556	end
 557}.register(self).then(&CommandList.method(:register))
 558
 559Command.new(
 560	"transactions",
 561	"🧾 Show Transactions",
 562	list_for: ->(customer:, **) { !!customer&.currency }
 563) {
 564	Command.customer.then(&:transactions).then do |txs|
 565		Command.finish do |reply|
 566			reply.command << FormTemplate.render("transactions", transactions: txs)
 567		end
 568	end
 569}.register(self).then(&CommandList.method(:register))
 570
 571Command.new(
 572	"configure calls",
 573	"📞 Configure Calls",
 574	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 575) {
 576	Command.customer.then do |customer|
 577		cc_form = ConfigureCallsForm.new(customer)
 578		Command.reply { |reply|
 579			reply.allowed_actions = [:next]
 580			reply.command << cc_form.render
 581		}.then { |iq|
 582			EMPromise.all(cc_form.parse(iq.form).map { |k, v|
 583				Command.execution.customer_repo.public_send("put_#{k}", customer, v)
 584			})
 585		}.then { Command.finish("Configuration saved!") }
 586	end
 587}.register(self).then(&CommandList.method(:register))
 588
 589Command.new(
 590	"ogm",
 591	"⏺️ Record Voicemail Greeting",
 592	list_for: ->(fwd: nil, **) { fwd&.voicemail_enabled? },
 593	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 594) {
 595	Command.customer.then do |customer|
 596		customer.fwd.create_call(CONFIG[:creds][:account]) do |cc|
 597			cc.from = customer.registered?.phone
 598			cc.application_id = CONFIG[:sip][:app]
 599			cc.answer_url = "#{CONFIG[:web_root]}/ogm/start?" \
 600			                "customer_id=#{customer.customer_id}"
 601		end
 602		Command.finish("You will now receive a call.")
 603	end
 604}.register(self).then(&CommandList.method(:register))
 605
 606Command.new(
 607	"migrate billing",
 608	"🏦 Switch to new billing",
 609	list_for: ->(tel:, customer:, **) { tel && !customer&.currency },
 610	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 611) {
 612	EMPromise.all([
 613		Command.customer,
 614		Command.reply do |reply|
 615			reply.allowed_actions = [:next]
 616			reply.command << FormTemplate.render("migrate_billing")
 617		end
 618	]).then do |(customer, iq)|
 619		plan_name = iq.form.field("plan_name").value.to_s
 620		customer = customer.with_plan(plan_name)
 621		customer.save_plan!.then {
 622			Registration::Payment.for(
 623				iq, customer, customer.registered?.phone,
 624				final_message: PaypalDone::MESSAGE,
 625				finish: PaypalDone
 626			)
 627		}.then(&:write).catch_only(Command::Execution::FinalStanza) do |s|
 628			BLATHER.join(CONFIG[:notify_admin], "sgx-jmp")
 629			BLATHER.say(
 630				CONFIG[:notify_admin],
 631				"#{customer.customer_id} migrated to #{customer.currency}",
 632				:groupchat
 633			)
 634			EMPromise.reject(s)
 635		end
 636	end
 637}.register(self).then(&CommandList.method(:register))
 638
 639Command.new(
 640	"credit cards",
 641	"💳 Credit Card Settings and Management",
 642	list_for: ->(customer:, **) { !!customer&.currency }
 643) {
 644	Command.customer.then do |customer|
 645		url = CONFIG[:credit_card_url].call(
 646			customer.jid.to_s.gsub("\\", "%5C"),
 647			customer.customer_id
 648		)
 649		desc = "Manage credits cards and settings"
 650		Command.finish("#{desc}: #{url}") do |reply|
 651			oob = OOB.find_or_create(reply.command)
 652			oob.url = url
 653			oob.desc = desc
 654		end
 655	end
 656}.register(self).then(&CommandList.method(:register))
 657
 658Command.new(
 659	"top up",
 660	"💲 Buy Account Credit by Credit Card",
 661	list_for: ->(payment_methods: [], **) { !payment_methods.empty? },
 662	format_error: ->(e) { "Failed to buy credit, system said: #{e.message}" }
 663) {
 664	Command.customer.then { |customer|
 665		BuyAccountCreditForm.for(customer).then do |credit_form|
 666			Command.reply { |reply|
 667				reply.allowed_actions = [:complete]
 668				reply.command << credit_form.form
 669			}.then do |iq|
 670				CreditCardSale.create(customer, **credit_form.parse(iq.form))
 671			end
 672		end
 673	}.then { |transaction|
 674		Command.finish("#{transaction} added to your account balance.")
 675	}.catch_only(
 676		AmountTooHighError,
 677		AmountTooLowError
 678	) do |e|
 679		Command.finish(e.message, type: :error)
 680	end
 681}.register(self).then(&CommandList.method(:register))
 682
 683Command.new(
 684	"alt top up",
 685	"🪙 Buy Account Credit by Bitcoin, Mail, or Interac e-Transfer",
 686	list_for: ->(customer:, **) { !!customer&.currency }
 687) {
 688	Command.customer.then { |customer|
 689		AltTopUpForm.for(customer)
 690	}.then do |alt_form|
 691		Command.reply { |reply|
 692			reply.allowed_actions = [:complete]
 693			reply.command << alt_form.form
 694		}.then do |iq|
 695			Command.finish { |reply| alt_form.parse(iq.form).action(reply) }
 696		end
 697	end
 698}.register(self).then(&CommandList.method(:register))
 699
 700Command.new(
 701	"plan settings",
 702	"📝 Manage your plan, including overage limits",
 703	list_for: ->(customer:, **) { !!customer&.currency }
 704) {
 705	Command.customer.then { |customer|
 706		EMPromise.all([
 707			REDIS.get("jmp_customer_monthly_data_limit-#{customer.customer_id}"),
 708			SIMRepo.new.owned_by(customer)
 709		]).then { |(limit, sims)| [customer, sims, limit] }
 710	}.then do |(customer, sims, limit)|
 711		Command.reply { |reply|
 712			reply.allowed_actions = [:next]
 713			reply.command << FormTemplate.render(
 714				"plan_settings", customer: customer, sims: sims, data_limit: limit
 715			)
 716		}.then { |iq|
 717			kwargs = {
 718				monthly_overage_limit: iq.form.field("monthly_overage_limit")&.value,
 719				monthly_data_limit: iq.form.field("monthly_data_limit")&.value
 720			}.compact
 721			Command.execution.customer_repo.put_monthly_limits(customer, **kwargs)
 722		}.then { Command.finish("Configuration saved!") }
 723	end
 724}.register(self).then(&CommandList.method(:register))
 725
 726Command.new(
 727	"referral codes",
 728	"👥 Refer a friend for free credit",
 729	list_for: ->(customer:, **) { !!customer&.currency }
 730) {
 731	repo = InvitesRepo.new
 732	Command.customer.then { |customer|
 733		EMPromise.all([
 734			repo.find_or_create_group_code(customer.customer_id),
 735			repo.unused_invites(customer.customer_id)
 736		])
 737	}.then do |(group_code, invites)|
 738		if invites.empty?
 739			Command.finish(
 740				"This code will provide credit equivalent to one month of service " \
 741				"to anyone after they sign up and pay: #{group_code}\n\n" \
 742				"You will receive credit equivalent to one month of service once " \
 743				"their payment clears."
 744			)
 745		else
 746			Command.finish do |reply|
 747				reply.command << FormTemplate.render(
 748					"codes",
 749					invites: invites,
 750					group_code: group_code
 751				)
 752			end
 753		end
 754	end
 755}.register(self).then(&CommandList.method(:register))
 756
 757# Assumes notify_from is a direct target
 758notify_to = CONFIG[:direct_targets].fetch(
 759	Blather::JID.new(CONFIG[:notify_from]).node.to_sym
 760)
 761
 762Command.new(
 763	"sims",
 764	"📶 (e)SIM Details",
 765	list_for: ->(customer:, **) { CONFIG[:keepgo] && !!customer&.currency },
 766	customer_repo: CustomerRepo.new(
 767		sgx_repo: TrivialBackendSgxRepo.new(jid: notify_to)
 768	)
 769) {
 770	Command.customer.then { |customer|
 771		EMPromise.all([customer, SIMRepo.new.owned_by(customer)])
 772	}.then do |(customer, sims)|
 773		Command.reply { |reply|
 774			reply.command << FormTemplate.render("sim_details", sims: sims)
 775		}.then { |iq|
 776			case iq.form.field("http://jabber.org/protocol/commands#actions")&.value
 777			when "order-sim"
 778				SIMOrder.for(customer, **CONFIG.dig(:sims, :sim, customer.currency))
 779			when "order-esim"
 780				SIMOrder::ESIM.for(
 781					customer, **CONFIG.dig(:sims, :esim, customer.currency)
 782				)
 783			when "edit-nicknames"
 784				EditSimNicknames.new(customer, sims)
 785			else
 786				Command.finish
 787			end
 788		}.then { |action|
 789			Command.reply { |reply|
 790				reply.allowed_actions = [:complete]
 791				reply.command << action.form
 792			}.then(&action.method(:complete))
 793		}
 794	end
 795}.register(self).then(&CommandList.method(:register))
 796
 797Command.new(
 798	"subaccount",
 799	"➕️ Create a new phone number linked to this balance",
 800	list_for: lambda do |customer:, **|
 801		!!customer&.currency &&
 802		customer&.billing_customer_id == customer&.customer_id
 803	end
 804) {
 805	cheogram = Command.execution.iq.from.resource =~ /\ACheogram/
 806	Command.customer.then do |customer|
 807		ParentCodeRepo.new.find_or_create(customer.customer_id).then do |code|
 808			Command.finish { |reply|
 809				reply.command << FormTemplate.render(
 810					"subaccount", code: code, cheogram: cheogram
 811				)
 812			}
 813		end
 814	end
 815}.register(self).then(&CommandList.method(:register))
 816
 817Command.new(
 818	"reset sip account",
 819	"☎️ Create or Reset SIP Account",
 820	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 821) {
 822	Command.customer.then do |customer|
 823		sip_account = customer.reset_sip_account
 824		Command.reply { |reply|
 825			reply.allowed_actions = [:next]
 826			form = sip_account.form
 827			form.type = :form
 828			form.fields += [{
 829				type: :boolean, var: "change_fwd",
 830				label: "Should inbound calls forward to this SIP account?"
 831			}]
 832			reply.command << form
 833		}.then do |fwd|
 834			if ["1", "true"].include?(fwd.form.field("change_fwd")&.value.to_s)
 835				Command.execution.customer_repo.put_fwd(
 836					customer,
 837					customer.fwd.with(uri: sip_account.uri)
 838				).then { Command.finish("Inbound calls will now forward to SIP.") }
 839			else
 840				Command.finish
 841			end
 842		end
 843	end
 844}.register(self).then(&CommandList.method(:register))
 845
 846Command.new(
 847	"lnp",
 848	"#️⃣ Port in your number from another carrier",
 849	list_for: ->(**) { true },
 850	customer_repo: CustomerRepo.new(
 851		sgx_repo: TrivialBackendSgxRepo.new(jid: notify_to)
 852	)
 853) {
 854	Command.customer.then do |customer|
 855		Command.reply { |reply|
 856			reply.allowed_actions = [:next]
 857			reply.command << FormTemplate.render("lnp")
 858		}.then { |iq|
 859			PortInOrder.parse(customer, iq.form).complete_with do |form|
 860				Command.reply { |reply|
 861					reply.allowed_actions = [:next]
 862					reply.command << form
 863				}.then(&:form)
 864			end
 865		}.then do |order|
 866			order_id = BandwidthIris::PortIn.create(order.to_h)[:order_id]
 867			customer.stanza_from(Blather::Stanza::Message.new(
 868				"",
 869				order.message(order_id)
 870			))
 871			Command.finish(
 872				"Your port-in request has been accepted, " \
 873				"support will contact you with next steps"
 874			)
 875		end
 876	end
 877}.register(self).then(&CommandList.method(:register))
 878
 879Command.new(
 880	"terminate account",
 881	"❌ Cancel your account and terminate your phone number",
 882	list_for: ->(**) { false },
 883	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
 884) {
 885	Command.reply { |reply|
 886		reply.allowed_actions = [:next]
 887		reply.note_text = "Press next to confirm your account termination."
 888	}.then { Command.customer }.then { |customer|
 889		AdminAction::CancelCustomer.call(
 890			customer,
 891			customer_repo: Command.execution.customer_repo
 892		)
 893	}.then do
 894		Command.finish("Account cancelled")
 895	end
 896}.register(self).then(&CommandList.method(:register))
 897
 898Command.new(
 899	"customer info",
 900	"Show Customer Info",
 901	list_for: ->(customer: nil, **) { customer&.admin? }
 902) {
 903	Command.customer.then do |customer|
 904		raise AuthError, "You are not an admin" unless customer&.admin?
 905
 906		customer_repo = CustomerRepo.new(
 907			sgx_repo: Bwmsgsv2Repo.new,
 908			bandwidth_tn_repo: EmptyRepo.new(BandwidthTnRepo.new) # No CNAM in admin
 909		)
 910
 911		AdminCommand::NoUser.new(customer_repo).start
 912	end
 913}.register(self).then(&CommandList.method(:register))
 914
 915Command.new(
 916	"reachability",
 917	"Test Reachability",
 918	list_for: ->(customer: nil, **) { customer&.admin? }
 919) {
 920	Command.customer.then do |customer|
 921		raise AuthError, "You are not an admin" unless customer&.admin?
 922
 923		form = ReachabilityForm.new(CustomerRepo.new)
 924
 925		Command.reply { |reply|
 926			reply.allowed_actions = [:next]
 927			reply.command << form.render
 928		}.then { |response|
 929			form.parse(response.form)
 930		}.then { |result|
 931			result.repo.get_or_create(result.target).then { |v|
 932				result.target.stanza_from(result.prompt) if result.prompt
 933
 934				Command.finish { |reply|
 935					reply.command << form.render_result(v)
 936				}
 937			}
 938		}.catch_only(RuntimeError) { |e|
 939			Command.finish(e, type: :error)
 940		}
 941	end
 942}.register(self).then(&CommandList.method(:register))
 943
 944Command.new(
 945	"snikket",
 946	"Launch Snikket Instance",
 947	list_for: ->(customer: nil, **) { customer&.admin? }
 948) {
 949	Command.customer.then do |customer|
 950		raise AuthError, "You are not an admin" unless customer&.admin?
 951
 952		Command.reply { |reply|
 953			reply.allowed_actions = [:next]
 954			reply.command << FormTemplate.render("snikket_launch")
 955		}.then { |response|
 956			domain = response.form.field("domain").value.to_s
 957			IQ_MANAGER.write(Snikket::Launch.new(
 958				nil, CONFIG[:snikket_hosting_api],
 959				domain: domain
 960			)).then do |launched|
 961				Snikket::CustomerInstance.for(customer, domain, launched)
 962			end
 963		}.then { |instance|
 964			Command.finish do |reply|
 965				reply.command << FormTemplate.render(
 966					"snikket_launched",
 967					instance: instance
 968				)
 969			end
 970		}
 971	end
 972}.register(self).then(&CommandList.method(:register))
 973
 974Command.new(
 975	"stop snikket",
 976	"STOP Snikket Instance",
 977	list_for: ->(customer: nil, **) { customer&.admin? }
 978) {
 979	Command.customer.then do |customer|
 980		raise AuthError, "You are not an admin" unless customer&.admin?
 981
 982		Command.reply { |reply|
 983			reply.allowed_actions = [:next]
 984			reply.command << FormTemplate.render("snikket_stop")
 985		}.then { |response|
 986			instance_id = response.form.field("instance_id").value.to_s
 987			IQ_MANAGER.write(Snikket::Stop.new(
 988				nil, CONFIG[:snikket_hosting_api],
 989				instance_id: instance_id
 990			))
 991		}.then { |iq|
 992			Command.finish(iq.to_s)
 993		}
 994	end
 995}.register(self).then(&CommandList.method(:register))
 996
 997Command.new(
 998	"delete snikket",
 999	"DELETE Snikket Instance",
1000	list_for: ->(customer: nil, **) { customer&.admin? }
1001) {
1002	Command.customer.then do |customer|
1003		raise AuthError, "You are not an admin" unless customer&.admin?
1004
1005		Command.reply { |reply|
1006			reply.allowed_actions = [:next]
1007			reply.command << FormTemplate.render("snikket_delete")
1008		}.then { |response|
1009			instance_id = response.form.field("instance_id").value.to_s
1010			IQ_MANAGER.write(Snikket::Delete.new(
1011				nil, CONFIG[:snikket_hosting_api],
1012				instance_id: instance_id
1013			))
1014		}.then { |iq|
1015			Command.finish(iq.to_s)
1016		}
1017	end
1018}.register(self).then(&CommandList.method(:register))
1019
1020Command.new(
1021	"find snikket",
1022	"Lookup Snikket Instance",
1023	list_for: ->(customer: nil, **) { customer&.admin? }
1024) {
1025	Command.customer.then do |customer|
1026		raise AuthError, "You are not an admin" unless customer&.admin?
1027
1028		Command.reply { |reply|
1029			reply.allowed_actions = [:next]
1030			reply.command << FormTemplate.render("snikket_launch")
1031		}.then { |response|
1032			domain = response.form.field("domain").value.to_s
1033			IQ_MANAGER.write(Snikket::DomainInfo.new(
1034				nil, CONFIG[:snikket_hosting_api],
1035				domain: domain
1036			))
1037		}.then { |instance|
1038			Command.finish do |reply|
1039				reply.command << FormTemplate.render(
1040					"snikket_result",
1041					instance: instance
1042				)
1043			end
1044		}
1045	end
1046}.register(self).then(&CommandList.method(:register))
1047
1048def reply_with_note(iq, text, type: :info)
1049	reply = iq.reply
1050	reply.status = :completed
1051	reply.note_type = type
1052	reply.note_text = text
1053
1054	self << reply
1055end
1056
1057Command.new(
1058	"https://ns.cheogram.com/sgx/jid-switch",
1059	"Change JID",
1060	list_for: lambda { |customer: nil, from_jid: nil, **|
1061		customer || from_jid.to_s =~ Regexp.new(CONFIG[:onboarding_domain])
1062	},
1063	customer_repo: CustomerRepo.new(sgx_repo: Bwmsgsv2Repo.new)
1064) {
1065	Command.customer.then { |customer|
1066		Command.reply { |reply|
1067			reply.command << FormTemplate.render("jid_switch")
1068		}.then { |response|
1069			new_jid = response.form.field("jid").value
1070			repo = Command.execution.customer_repo
1071			repo.find_by_jid(new_jid).catch_only(CustomerRepo::NotFound) { nil }
1072				.then { |cust|
1073					next EMPromise.reject("Customer Already Exists") if cust
1074
1075					repo.change_jid(customer, new_jid)
1076				}
1077		}.then {
1078			StatsD.increment("changejid.completed")
1079			jid = ProxiedJID.new(customer.jid).unproxied
1080			if jid.domain == CONFIG[:onboarding_domain]
1081				CustomerRepo.new.find(customer.customer_id).then do |cust|
1082					WelcomeMessage.new(cust, customer.registered?.phone).welcome
1083				end
1084			end
1085			Command.finish { |reply|
1086				reply.note_type = :info
1087				reply.note_text = "Customer JID Changed"
1088			}
1089		}
1090	}
1091}.register(self).then(&CommandList.method(:register))
1092
1093Command.new(
1094	"web-register",
1095	"Initiate Register from Web",
1096	list_for: lambda { |from_jid: nil, **|
1097		from_jid&.stripped.to_s == CONFIG[:web_register][:from]
1098	}
1099) {
1100	if Command.execution.iq.from.stripped != CONFIG[:web_register][:from]
1101		next EMPromise.reject(
1102			Command::Execution::FinalStanza.new(iq.as_error("forbidden", :auth))
1103		)
1104	end
1105
1106	Command.reply { |reply|
1107		reply.command << FormTemplate.render("web_register")
1108	}.then do |iq|
1109		jid = iq.form.field("jid")&.value.to_s.strip
1110		tel = iq.form.field("tel")&.value.to_s.strip
1111		if jid !~ /\./ || jid =~ /\s/
1112			Command.finish("The Jabber ID you entered was not valid.", type: :error)
1113		elsif tel !~ /\A\+\d+\Z/
1114			Command.finish("Invalid telephone number", type: :error)
1115		else
1116			IQ_MANAGER.write(Blather::Stanza::Iq::Command.new.tap { |cmd|
1117				cmd.to = CONFIG[:web_register][:to]
1118				cmd.node = "push-register"
1119				cmd.form.fields = [{ var: "to", value: jid }]
1120				cmd.form.type = "submit"
1121			}).then { |result|
1122				TEL_SELECTIONS.set_tel(result.form.field("from")&.value.to_s.strip, tel)
1123			}.then { Command.finish }
1124		end
1125	end
1126}.register(self).then(&CommandList.method(:register))
1127
1128command sessionid: /./ do |iq|
1129	COMMAND_MANAGER.fulfill(iq)
1130	IQ_MANAGER.fulfill(iq)
1131	true
1132end
1133
1134iq type: [:result, :error] do |iq|
1135	IQ_MANAGER.fulfill(iq)
1136	true
1137end
1138
1139iq type: [:get, :set] do |iq|
1140	StatsD.increment("unknown_iq")
1141
1142	self << Blather::StanzaError.new(iq, "feature-not-implemented", :cancel)
1143end
1144
1145trap(:INT) { EM.stop }
1146trap(:TERM) { EM.stop }
1147EM.run { client.run }