moralis_sdk.js 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047
  1. /* import moralis */
  2. const Moralis = require("moralis/node");
  3. var utils = require('./utils.js');
  4. // var config = require('../config/config.js')(db_config,
  5. // moralis_config)
  6. var { moralis_config, reids_token_config, account_config } = require('../config/config.js')
  7. const redis = require("./db/redis_db") //导入 db.js
  8. const mysql = require("./db/mysql_db")
  9. const logger = require('./logger')
  10. const report = require("./report") //导入 db.js
  11. const BigNumber = require('bignumber.js')
  12. /* Moralis init code */
  13. var serverUrl = moralis_config.SERVER_URL;
  14. var appId = moralis_config.APP_ID;
  15. var masterKey = moralis_config.MASTER_KEY;
  16. var moralisSecret = moralis_config.MORALIS_SECRET;
  17. // 内部异常
  18. const ERROR_CODE_001 = -1;
  19. const SUCCEED_CODE = 0;
  20. // 归集队列
  21. var collectCoinsArrays = [];
  22. // 是否执行归集
  23. var isExecCollect = false;
  24. var masterInit = false
  25. var secretInit = false
  26. // 对 shib 做限制
  27. const SHIB_AMOUNT_LIMIT = 20000
  28. const SHIB_AMOUNT_FEE_LIMIT = 10000
  29. const SHIB_ADDRESS = '0x2859e4544C4bB03966803b044A93563Bd2D0DD4D'
  30. // logger.log('Moralis.settings',Moralis)
  31. // Moralis.settings.setAPIRateLimit({
  32. // anonymous:10, authenticated:20, windowMs:60000
  33. // })
  34. /**
  35. * 初始化 moralis
  36. * https://st94nif1cq.feishu.cn/docs/doccnNxG2UwHPCdZXbywgbdy13f#
  37. */
  38. async function initMasterSDK() {
  39. logger.debug('initMasterSDK start')
  40. if (masterInit) {
  41. logger.log('initMasterSDK', masterInit)
  42. return
  43. }
  44. await Moralis.start({ serverUrl, appId, masterKey });
  45. logger.debug('initMasterSDK start ok')
  46. // masterInit = true
  47. }
  48. async function initMoralisSecretSDK() {
  49. if (secretInit) {
  50. logger.log('initMoralisSecretSDK', secretInit)
  51. return
  52. }
  53. logger.debug('initMoralisSecretSDK start')
  54. await Moralis.start({ serverUrl, appId, moralisSecret });
  55. logger.debug('initMoralisSecretSDK start ok')
  56. // secretInit = true
  57. }
  58. function toJson(code_, obj_, errMsg_) {
  59. return utils.toJson(code_, obj_, errMsg_);
  60. }
  61. /**
  62. * 通过交易记录获取 gas 费用
  63. * @param {} type
  64. * @param {*} obj
  65. * @param {*} address
  66. */
  67. function getTransferRecordGasFree(type, obj, address) {
  68. logger.log('getTransferRecordGasFree fun in', type, obj, address)
  69. if (obj && obj.data && obj.data.result && Array.isArray(obj.data.result) && obj.data.result.length > 0) {
  70. logger.log('getTransferRecordGasFree fun in', 1)
  71. for (let index = 0; index < obj.data.result.length; index++) {
  72. const element = obj.data.result[index];
  73. logger.log('getTransferRecordGasFree fun in 2', element)
  74. if (element.type == type && element.to_address.toLowerCase() == address.toLowerCase()) {
  75. if (element.gas_price && element.gas && type == 'native') {//native
  76. return {
  77. totalGasFree: parseInt(element.gas_price) * parseInt(element.gas),
  78. gas_price: parseInt(element.gas_price)
  79. }
  80. }
  81. }
  82. }
  83. }
  84. return null;
  85. }
  86. /**
  87. * 获取转账的 gas 费
  88. * @param {*} type
  89. * @param {*} json
  90. * @returns
  91. */
  92. function getTransferGasFree(type, json) {
  93. var curGasPrice = account_config.BNB_GAS_PRICE
  94. var curGasLimit = account_config.TOKEN_GAS_LIMIT
  95. var nativeValue = 0;
  96. var tokenValue = 0;
  97. var code = -1;
  98. if (type == 'native') {
  99. curGasLimit = account_config.BNB_GAS_LIMIT
  100. }
  101. var totalGasFree = parseInt(curGasLimit) * parseInt(curGasPrice);
  102. logger.log('getTransferGasFree json=', json)
  103. if (json) {
  104. var obj = JSON.parse(json)
  105. code = obj.code;
  106. logger.log('getTransferGasFree obj=', obj)
  107. if (obj.data && obj.data.gasPrice && obj.data.gasLimit && obj.data.gasPrice.hex && obj.data.gasLimit.hex) {
  108. var curGasPrice = BigNumber(obj.data.gasPrice.hex).toNumber()
  109. var curGasLimit = BigNumber(obj.data.gasLimit.hex).toNumber()
  110. if (curGasPrice > 0 && curGasLimit > 0) {
  111. totalGasFree = curGasPrice * curGasLimit;
  112. }
  113. logger.log('getTransferGasFree totalGasFree=', curGasPrice, curGasLimit, totalGasFree)
  114. if (obj.data.value && obj.data.value.hex) {
  115. var value = BigNumber(obj.data.value.hex).toNumber()
  116. logger.log('getTransferGasFree native value=', value)
  117. if (value != 0) {
  118. nativeValue = value
  119. }
  120. }
  121. }
  122. }
  123. return {
  124. code: code,
  125. totalGasFree: totalGasFree,
  126. nativeValue: nativeValue,
  127. tokenValue: tokenValue,
  128. }
  129. }
  130. /**
  131. *
  132. * @param {转账是否成功} json
  133. */
  134. function isTransferSucceed(json) {
  135. if (json) {
  136. var obj = JSON.parse(json)
  137. return obj.code == 0
  138. }
  139. return false
  140. }
  141. const withdraw = async (obj) => {
  142. obj.withdraw = 1;
  143. return await transfer_handle(obj);
  144. }
  145. async function getAccountBalances(options) {
  146. await initMasterSDK();
  147. if (options.chain) {
  148. options.chain = utils.getChainName(options.chain)
  149. }
  150. logger.log('getAccountBalances :', options)
  151. //做 3次 重试,每次间隔 1s+=
  152. //Error: Request failed with status code 524
  153. //Error: Request failed with status code 500
  154. var tryCount = 4;
  155. var delay = 1000
  156. var interval = 500
  157. var result;
  158. do {
  159. try {
  160. if (options.type == 'native') {
  161. result = await Moralis.Web3API.account.getNativeBalance(options);
  162. logger.log('getNativeBalance=', result);
  163. } else {
  164. result = await Moralis.Web3API.account.getTokenBalances(options);
  165. logger.log('getTokenBalances=', result);
  166. }
  167. break
  168. } catch (error) {
  169. if (tryCount == 1) {
  170. logger.error('getBalances error:', error.toString(), JSON.stringify(options))
  171. result = null;
  172. }
  173. tryCount -= 1;
  174. await utils.sleep(delay)
  175. delay += interval
  176. logger.error('getBalances error:', JSON.stringify(options), "重试:" + tryCount +"次")
  177. }
  178. } while (tryCount >= 1);
  179. return result
  180. }
  181. /**
  182. * 获取当前账户下所有不同种类的币 主流币 + 20币
  183. *
  184. * 必填项
  185. * obj.chain = ?
  186. * @param {*} obj
  187. */
  188. async function getAccountAllCoins(obj) {
  189. var temp_chain = obj.chain;
  190. //拿到我当前所有的币种
  191. obj.type = 'native'
  192. obj.chain = temp_chain
  193. var native_balance = await getAccountBalances(obj);
  194. //拿到我当前所有的 20 币
  195. obj.type = '20'
  196. obj.chain = temp_chain
  197. var others_balances = await getAccountBalances(obj);
  198. return {
  199. native: native_balance,
  200. other: others_balances,
  201. }
  202. }
  203. /**
  204. * 判断是否转 gas 费
  205. * @param {*} my_account_all_coins
  206. */
  207. async function computeTransferGasFree(obj, my_account_all_coins, tokenPrices) {
  208. var totalCount = 0;
  209. var tokenCount = 0;
  210. var tokenGasPrice = 0;
  211. var nativeGasPrice = 0;
  212. var ret_total_gas_price = 0;
  213. var ret_total_count = 0;
  214. var ret_a_gas = 0;
  215. //需要转账的数组对象
  216. var transfer_obj = [];
  217. var receiver_info = await queryCompanyInfoFromId(0);
  218. //得到 20 币 满足 1美刀的 count
  219. if (Array.isArray(my_account_all_coins.other) && my_account_all_coins.other.length > 0) {
  220. for (let index = 0; index < my_account_all_coins.other.length; index++) {
  221. const element = my_account_all_coins.other[index];
  222. // my_account_all_coins.other.forEach(element => {
  223. logger.log('20 element=', element);
  224. var find_transfer_item = findTokenPriceItem(element.token_address, tokenPrices);
  225. if (element.token_address == SHIB_ADDRESS) {
  226. if (parseInt(element.balance) < SHIB_AMOUNT_LIMIT) {
  227. logger.warn('不满足归集条件', element);
  228. continue
  229. }
  230. }
  231. if (find_transfer_item) {
  232. var total_all_usdprice = calculate_total_usdprice(element.balance, element.decimals, find_transfer_item.usdPrice);
  233. logger.log('findTokenPriceItem ret=', element.token_address, find_transfer_item, total_all_usdprice);
  234. if (find_transfer_item && total_all_usdprice > 1.0) {
  235. tokenCount += 1;
  236. logger.log('token > 1.0', tokenCount, element.token_address);
  237. var obj_20 = {
  238. chain: obj.chain,
  239. contractAddress: element.token_address,
  240. amount: element.balance,
  241. receiver: receiver_info.user_address,
  242. type: 'erc20',
  243. address: obj.address,
  244. }
  245. transfer_obj.push(obj_20)
  246. }
  247. } else {
  248. logger.log('findTokenPriceItem not fount =', element);
  249. }
  250. // });
  251. }
  252. logger.log('account_config.TOKEN_GAS_LIMIT=', account_config.TOKEN_GAS_LIMIT);
  253. var lastTokenFree = await redis.readRedis(reids_token_config.LAST_TOTAL_TOKEN_FREE)
  254. logger.log('LAST_TOTAL_TOKEN_FREE=', lastTokenFree);
  255. if (lastTokenFree && reids_token_config.LAST_TOTAL_TOKEN_FREE && parseInt(lastTokenFree) > 0) {
  256. tokenGasPrice = parseInt(tokenCount) * parseInt(lastTokenFree);
  257. } else {
  258. tokenGasPrice = parseInt(tokenCount) * parseInt(account_config.TOKEN_GAS_LIMIT) * parseInt(account_config.BNB_GAS_PRICE);
  259. }
  260. ret_a_gas = account_config.BNB_GAS_PRICE;
  261. logger.log('tokenGasPrice=', tokenGasPrice);
  262. }
  263. //计算 native 是否满足 1美刀
  264. logger.log('isTransferGasFree token count:', tokenCount);
  265. var nativeAllBalance = 0;
  266. if (my_account_all_coins.native && my_account_all_coins.native.balance)
  267. nativeAllBalance = my_account_all_coins.native.balance
  268. var nativeCount = 0
  269. var nativePriceItem = findTokenPriceItem('0x0000000000000000000000000000000000000000', tokenPrices);//todo 线上环境需要换
  270. logger.log('native nativePriceItem=', nativePriceItem, nativeAllBalance);
  271. if (nativePriceItem) {
  272. var total_all_usdprice = calculate_total_usdprice(nativeAllBalance, '18', nativePriceItem.usdPrice);
  273. logger.log('native total_all_usdprice=', total_all_usdprice);
  274. if (total_all_usdprice > 1.0) {
  275. nativeCount = 1;
  276. logger.log('native > 1.0', tokenCount);
  277. var lastBnbFree = await redis.readRedis(reids_token_config.LAST_TOTAL_BNB_FREE)
  278. logger.log('LAST_TOTAL_BNB_FREE=', lastBnbFree);
  279. if (lastTokenFree && reids_token_config.LAST_TOTAL_TOKEN_FREE && parseInt(lastTokenFree) > 0) {
  280. nativeGasPrice = parseInt(nativeCount) * parseInt(lastBnbFree);
  281. } else {
  282. nativeGasPrice = parseInt(nativeCount) * parseInt(account_config.BNB_GAS_LIMIT) * parseInt(account_config.BNB_GAS_PRICE);
  283. }
  284. ret_a_gas = account_config.BNB_GAS_PRICE;
  285. // var real_native_amount = BigInt(nativeAllBalance) - BigInt(nativeGasPrice) - BigInt(tokenGasPrice);
  286. var real_native_amount = BigInt(nativeAllBalance);
  287. var obj_native = {
  288. chain: obj.chain,
  289. amount: real_native_amount.toString(),
  290. receiver: receiver_info.user_address,
  291. type: 'native',
  292. address: obj.address,
  293. }
  294. transfer_obj.push(obj_native)
  295. }
  296. }
  297. logger.log('transfer obj=', transfer_obj)
  298. //计算所有币转账所需要的 gas
  299. totalCount = nativeCount + tokenCount;
  300. // var gasPrice = await redis.readRedis(reids_token_config.GASPRICE);
  301. logger.log('nativeAllBalance', nativeAllBalance);
  302. logger.log('totalCount', totalCount);
  303. var total2Gas = nativeGasPrice + tokenGasPrice;
  304. var service_charge = 0;
  305. logger.log('total2Gas', total2Gas);
  306. //需要转账的 obj
  307. my_account_all_coins.transfer_arrays = transfer_obj
  308. //如果当前的钱不够 gas
  309. if (nativeAllBalance < total2Gas) {
  310. if (tokenCount > 0) {//出现 token 需要转移手续费
  311. service_charge = 1;
  312. // total2Gas = (total2Gas - nativeAllBalance);//充手续费
  313. logger.log('需要转账=', total2Gas);
  314. }
  315. }
  316. ret_total_gas_price = total2Gas.toString();
  317. ret_total_count = totalCount;
  318. //返回结果
  319. return {
  320. gasPrice: ret_total_gas_price, //需要 归集到用户地址的 gas 费转移
  321. totalCount: ret_total_count, //一共归集次数
  322. aGasPrice: ret_a_gas, //单个 gas 费用
  323. get_service_charge: service_charge,//是否需要服务费
  324. transfer_obj: transfer_obj,
  325. };
  326. }
  327. function findTokenPriceItem(token_address, tokenPrices) {
  328. return tokenPrices.tokenPrice.find(element => {
  329. // logger.log('findTokenPriceItem find=', element.contract, token_address)
  330. return element.contract.toLowerCase() == token_address.toLowerCase();
  331. })
  332. }
  333. // function calculate_total_usdprice(amount, decimals, usdprice) {
  334. // return parseInt(amount) / (10**parseInt(decimals)) * parseInt(usdprice) ;
  335. function calculate_total_usdprice(amount, decimals, usdprice) {
  336. // if(usdprice.toString().indexOf("e") != -1 ){
  337. // return parseInt(amount) / (10 ** parseInt(decimals)) * parseFloat(usdprice);
  338. // }else {
  339. // return parseInt(amount) / (10 ** parseInt(decimals)) * parseFloat(usdprice);
  340. // }
  341. return parseInt(amount) / (10 ** parseInt(decimals)) * parseFloat(usdprice);
  342. }
  343. function addNativeValue(nativeValue, aValue) {
  344. return BigInt(nativeValue) + BigInt(aValue)
  345. }
  346. function reduceNativeValue(nativeValue, rValue) {
  347. return BigInt(nativeValue) - BigInt(rValue)
  348. }
  349. /**
  350. * todo --> 计算 gas
  351. * @param {*} nativeBalance
  352. * @param {*} obj
  353. * @returns
  354. */
  355. async function updateNativeBalance(nativeBalance, obj) {
  356. var temp = { ...obj }
  357. console.log('updateNativeBalance before', temp, nativeBalance)
  358. var retryCount = 15;
  359. if (temp.chain) {
  360. temp.chain = utils.getChainName(temp.chain)
  361. }
  362. console.log('updateNativeBalance after', temp, nativeBalance)
  363. do {
  364. //上面转账完 BNB 会减去,这里再获取一次
  365. var native_ret = await Moralis.Web3API.account.getNativeBalance(temp);
  366. logger.log('更新余额 :', nativeBalance, native_ret, retryCount)
  367. if (nativeBalance != native_ret.balance && BigInt(nativeBalance) < BigInt(native_ret.balance)) {
  368. return native_ret.balance;
  369. }
  370. await utils.sleep(3000);
  371. retryCount--;
  372. } while (native_ret.balance == nativeBalance && retryCount > 0);
  373. return null;
  374. }
  375. //20 and native 归集
  376. async function transfers(obj, my_account_all_coins, logParams) {
  377. // 优化后的归集
  378. if (my_account_all_coins && my_account_all_coins.transfer_arrays && Array.isArray(my_account_all_coins.transfer_arrays) && my_account_all_coins.transfer_arrays.length > 0) {
  379. var t_i = 0;
  380. var totalGasFee = 0;
  381. var collects = []
  382. for (let index = 0; index < my_account_all_coins.transfer_arrays.length; index++) {
  383. var ti = my_account_all_coins.transfer_arrays[index]
  384. logger.tlog('ti=', ti)
  385. //fix 归集失败的问题
  386. //线上环境和测试环境由于访问快,导致余额未来得及更新,这里做一个 延迟判断
  387. if (ti.type == 'native') {
  388. await updateNativeBalance(ti.amount, ti);
  389. }
  390. var ret = await start_collectCoins(ti)
  391. //更新 native 金额
  392. if (isTransferSucceed(ret)) {
  393. logger.tlog('start_collectCoins respose...', ret);
  394. t_i += 1
  395. collects.push(ret);
  396. //缓存当前交易的 gas 费用
  397. if (ret && ti.contractAddress) {
  398. var tr = getTransferGasFree('token', ret)
  399. // tr.totalGasFree
  400. totalGasFee += parseInt(tr.totalGasFree);
  401. } else {
  402. var tr = getTransferGasFree('native', ret)
  403. // tr.totalGasFree
  404. totalGasFee += parseInt(tr.totalGasFree);
  405. }
  406. } else {
  407. logger.tlog('start_collectCoins error=', ret);
  408. logger.error('start_collectCoins 归集 error=', ret, ti.toString());
  409. };
  410. }
  411. if (t_i == my_account_all_coins.transfer_arrays.length) {
  412. //日志埋点-归集全部所花费的 gas 费用
  413. logParams.collectCoinsTotalGasFee = totalGasFee.toString();
  414. logger.log('归集日志上报:', logParams);
  415. logParams.type = report.REPORT_TYPE.collect_coins
  416. //日志埋点-归集的响应
  417. logParams.results = collects
  418. var info = await queryCompanyInfoFromId(0);
  419. logParams.receiver_address_total_balance_after = await queryCollectBalance(info.user_address, obj.chain)
  420. //日志埋点-日志上报
  421. report.logReport(logParams)
  422. return toJson(0, null, '所有币归集成功.');
  423. } else {
  424. return toJson(-1, null, '归集失败.');
  425. }
  426. } else {
  427. return toJson(-1, null, 'transfer conditions are not met.');
  428. }
  429. }
  430. const start_collectCoins = async (obj) => {
  431. obj.withdraw = 0;
  432. return await transfer_handle(obj);
  433. }
  434. const transfer_handle = async (obj) => {
  435. //提币
  436. if (obj.withdraw) {
  437. var id = 0
  438. if (obj.privateKeyId)
  439. id = obj.privateKeyId
  440. //读取用户充币地址对应的私钥
  441. var info = await queryCompanyInfoFromId(id);
  442. logger.log('transfer_handle queryCompanyInfoFromId=', info);
  443. //提币公司
  444. obj.privateKey = info.user_private_key;
  445. // if (process.env.NODE_ENV != 'dev') {
  446. logger.log('readCompanyPriveteKeyFromMysql=', obj.privateKey)
  447. // }
  448. } else {
  449. //读取用户充币地址对应的私钥
  450. obj.privateKey = await readPriveteKeyFromMysql(obj.address);
  451. }
  452. if (obj.privateKey && obj.privateKey.results) {
  453. obj.privateKey = obj.privateKey.results;
  454. }
  455. if (!obj.privateKey) {
  456. logger.error('readPriveteKeyFromMysql error.', obj);
  457. //提币公司
  458. return toJson(-1, null, "readPriveteKeyFromMysql error.");
  459. }
  460. try {
  461. //解密
  462. obj.privateKey = utils.decryptPrivityKey(obj.privateKey);
  463. if (!obj.privateKey) {
  464. logger.error('decryptPrivityKey error', obj);
  465. return toJson(-1, null, "decryptPrivityKey error.");
  466. }
  467. } catch (error) {
  468. logger.error('decryptPrivityKey error', error.toString(), JSON.stringify(obj));
  469. if (!obj.privateKey) {
  470. return toJson(-1, null, "decryptPrivityKey error.", error.toString());
  471. }
  472. }
  473. var ret = await transfer(obj);
  474. if (isTransferSucceed(ret)) {
  475. //缓存当前交易的 gas 费用
  476. if (ret && obj.contractAddress) {
  477. var tr = getTransferGasFree('token', ret)
  478. logger.debug('cache setkey token LAST_TOTAL_TOKEN_FREE getTransferGasFree', tr)
  479. redis.redis_set(reids_token_config.LAST_TOTAL_TOKEN_FREE, tr.totalGasFree);
  480. } else {
  481. var tr = getTransferGasFree('native', ret)
  482. logger.debug('cache setkey LAST_TOTAL_BNB_FREE getTransferGasFree', tr)
  483. redis.redis_set(reids_token_config.LAST_TOTAL_BNB_FREE, tr.totalGasFree);
  484. }
  485. } else {
  486. logger.error('transfer_handle transfer error', JSON.stringify(obj));
  487. }
  488. return ret;
  489. }
  490. async function queryCollectBalance(address, chain) {
  491. // var t_chain = utils.getChainName(chain)
  492. try {
  493. var queryCollectBalance = {
  494. address: address,
  495. chain: chain
  496. }
  497. logger.log('queryCollectBalance', queryCollectBalance)
  498. // 查询归集地址余额
  499. // return await getAccountBalances(queryCollectBalance);
  500. return await getAccountAllCoins(queryCollectBalance);
  501. } catch (error) {
  502. logger.error('queryCollectBalance error', error.toString(), address, chain);
  503. return 0;
  504. }
  505. }
  506. /**
  507. * 用户充币地址的币转移到归集地址
  508. * 1、检查当前账户的 主流币或者 20 币,是否满足一美刀,并且查看是否满足转账费用 n 如果不满足,先从归集地址 -> 用户提币地址(0.*2 来回2次转移)
  509. * 2、发起归集 用户账户 -> 从 mysql 拿到私钥进行解密 -> 转移到归集地址
  510. *
  511. *
  512. * @param {*} obj
  513. */
  514. const collectCoins = async (obj) => {
  515. logger.log('fun collectCoins in =', obj)
  516. var chain = obj.chain;
  517. //1、拿到当前账户所有的币
  518. //2、是否满足交易费 如果不满足则 归集地址转移 币count * 手续费 到充币地址
  519. //3、遍历所有币,开始转移到归集地址
  520. // 日志上报的参数
  521. var logParams = {};
  522. var my_account_all_coins = await getAccountAllCoins(obj);
  523. logger.log('collectCoins getAccountAllCoins=', my_account_all_coins)
  524. if (!my_account_all_coins.native && !my_account_all_coins.other) {
  525. return 'getAccountAllCoins error.' + my_account_all_coins
  526. }
  527. //埋点日志-转账之前的充币地址余额
  528. logParams.addressBalances = { ...my_account_all_coins };
  529. //得到币价格
  530. if (process.env.NODE_ENV == 'dev') {
  531. var test_json = '{"tokenPrice": [{"contract": "0x0000000000000000000000000000000000000000", "usdPrice": 400}, {"contract": "0x03716F32f72c692a0B355fa04639669E3341B94e", "usdPrice": 0.1}]}'
  532. obj.tokenPrices = JSON.parse(test_json);
  533. } else {
  534. obj.tokenPrices = await redis.readRedis(reids_token_config.TOKENPRICE)
  535. if (!obj.tokenPrices) {
  536. logger.error('readRedis TOKENPRICE error')
  537. return 'readRedis error'
  538. }
  539. if (typeof obj.tokenPrices == 'string')
  540. obj.tokenPrices = JSON.parse(obj.tokenPrices);
  541. }
  542. logger.log('tokenPrices=', obj.tokenPrices)
  543. obj.chain = chain;
  544. //计算 gas 费用 是否需要归集
  545. var transFerGasFree = await computeTransferGasFree(obj, my_account_all_coins, obj.tokenPrices);
  546. //埋点日志- 需要归集的 native token
  547. if (transFerGasFree.transfer_obj && Array.isArray(transFerGasFree.transfer_obj)) {
  548. logParams.transfers = []
  549. for (let index = 0; index < transFerGasFree.transfer_obj.length; index++) {
  550. const element = transFerGasFree.transfer_obj[index];
  551. logParams.transfers.push({ ...element })
  552. }
  553. }
  554. logger.log('computeTransferGasFree=', transFerGasFree)
  555. logger.log(' logParams.transfers=', logParams.transfers)
  556. //是否需要归集
  557. if (transFerGasFree.totalCount > 0) {
  558. //需要转移 gas 费
  559. //每次都需要充值 gas 费
  560. if (account_config.TRANSFER_GAS || (parseInt(transFerGasFree.gasPrice) > 0 && transFerGasFree.get_service_charge == 1)) {
  561. var info = await queryCompanyInfoFromId(0);
  562. var obj_wd = {
  563. chain: chain,
  564. amount: transFerGasFree.gasPrice,
  565. receiver: obj.address,
  566. type: 'native',
  567. // address: account_config.WELLET_PUBLIC_KEY, //todo 正式环境需要替换从 mysql read
  568. address: info.user_address, //todo 正式环境需要替换从 mysql read
  569. }
  570. //埋点日志-预存 gas 付费
  571. logParams.prestore_gas_fee = { ...obj_wd };
  572. logParams.receiver_address_total_balance_before = await queryCollectBalance(info.user_address, chain)
  573. logParams.company_public_key = info.user_address
  574. logger.log('开始充值 gas ', obj_wd)
  575. var ret = await withdraw(obj_wd)
  576. logger.log('充值完成 gas ', ret)
  577. if (!isTransferSucceed(ret)) return ret;
  578. var transfer = getTransferGasFree('native', ret)
  579. logger.log('getTransferGasFree transfer =', transfer)
  580. if (transfer && transfer.nativeValue > 0) {
  581. logger.log('tempNativeValue=', my_account_all_coins.native.balance)
  582. var tempNativeValue = addNativeValue(my_account_all_coins.native.balance, transfer.nativeValue)
  583. my_account_all_coins.native.balance = tempNativeValue.toString();
  584. logger.log('udpateNativeValue=', tempNativeValue);
  585. } else return "get native value error."
  586. }
  587. obj.chain = chain;
  588. obj.transFerGasFree = transFerGasFree;
  589. logger.log('transfers--->', obj);
  590. var ret = await transfers(info, my_account_all_coins, logParams);
  591. logger.log('归集结果=', ret);
  592. return ret;
  593. }
  594. return obj.address + ':不满足归集条件';
  595. }
  596. async function execCollectCoinsTask() {
  597. logger.log('execCollectCoinsTask in', collectCoinsArrays.length)
  598. if (isExecCollect) return
  599. isExecCollect = true;
  600. while (collectCoinsArrays.length > 0) {
  601. var obj = collectCoinsArrays.pop();
  602. //开始收集用户地址里面的币到归集地址
  603. var ret = await collectCoins(obj);
  604. // await utils.sleep(3000)
  605. logger.log('execCollectCoinsTask=', collectCoinsArrays.length, ret)
  606. }
  607. isExecCollect = false;
  608. logger.log('execCollectCoinsTask out', collectCoinsArrays.length)
  609. }
  610. function pushCollectConisObj(obj) {
  611. logger.log('collectCoinsArrays length=', collectCoinsArrays.length, collectCoinsArrays)
  612. if (collectCoinsArrays.length > 0) {
  613. var findItem = collectCoinsArrays.find(element => {
  614. return (obj.address == element.address)
  615. })
  616. if (findItem) {
  617. logger.log('当前任务正在处理中...', obj.address)
  618. return;
  619. }
  620. }
  621. collectCoinsArrays.push(obj)
  622. execCollectCoinsTask();
  623. }
  624. async function readPriveteKeyFromMysql(address) {
  625. return new Promise(resolve => {
  626. mysql.queryUserPrivateKeyFromUserAddress(address).then(ret => {
  627. logger.log('readPriveteKeyFromMysql=', ret);
  628. resolve(ret);
  629. })
  630. })
  631. }
  632. async function queryCompanyInfoFromId(id) {
  633. return new Promise(resolve => {
  634. mysql.queryCompanyInfoFromId(id).then(ret => {
  635. logger.log('queryCompanyInfoFromId=', ret, ret.results, ret.results.user_address, ret.results.user_private_key);
  636. resolve(ret.results);
  637. })
  638. })
  639. }
  640. const transfer = async (obj) => {
  641. logger.debug("fun transfer serverUrl ", serverUrl);
  642. logger.debug("fun transfer appId ", serverUrl);
  643. logger.debug("fun transfer moralisSecret ", moralisSecret);
  644. await initMoralisSecretSDK();
  645. // initSDK(moralisSecret);
  646. logger.debug("fun transfer start ok ");
  647. const opts = {};
  648. opts.chainId = 'bsc_testnet';
  649. opts.privateKey = moralis_config.DEFAULT_PRIVATE_KEY;
  650. opts.type = "erc20"; //native erc20
  651. if (!obj.receiver || !obj.amount || parseInt(obj.amount) <= 0) {
  652. logger.error("transfer fun transfer parameter error.", obj.receiver, obj.amount, obj.amount);
  653. return toJson(ERROR_CODE_001, null, "please check receiver or amount parameter is ok ?");
  654. }
  655. if (obj.chain != null) {
  656. opts.chainId = utils.getChainId(obj.chain);
  657. console.log("chainId:", opts.chainId);
  658. }
  659. if (obj.type != null) {
  660. opts.type = obj.type;
  661. }
  662. if (obj.from_block != null) {
  663. opts.from_block = obj.from_block;
  664. }
  665. if (obj.to_block != null) {
  666. opts.to_block = obj.to_block;
  667. }
  668. opts.contractAddress = obj.contractAddress;
  669. opts.receiver = obj.receiver;
  670. //调用者传入
  671. // opts.amount = Moralis.Units.Token(obj.amount, 18);
  672. opts.amount = obj.amount;
  673. if (obj.privateKey != null) {
  674. opts.privateKey = obj.privateKey;
  675. }
  676. var options;
  677. try {
  678. // sending 0.5 DAI tokens with 18 decimals on BSC testnet
  679. if (opts.contractAddress) { //如果存在就是代币
  680. // if (opts.contractAddress == SHIB_ADDRESS) {
  681. // //提币金额必须大于 20000 ,手续费 10000
  682. // if (parseInt(opts.amount) >= SHIB_AMOUNT_LIMIT) {
  683. // opts.amount = (parseInt(opts.amount) - SHIB_AMOUNT_FEE_LIMIT).toString()
  684. // } else {
  685. // return toJson(-1, null, "最低限额为 20000 个token.");
  686. // }
  687. // }
  688. options = Moralis.TransferOptions = {
  689. type: opts.type,
  690. amount: opts.amount,
  691. receiver: opts.receiver, //接收钱包地址
  692. contractAddress: opts.contractAddress //用户合约地址
  693. };
  694. logger.tlog("options 20 =", options);
  695. } else { //ETH or BNB
  696. options = Moralis.TransferOptions = {
  697. type: opts.type,
  698. amount: opts.amount,
  699. receiver: opts.receiver, //接收钱包地址
  700. };
  701. logger.tlog("options native =", options, opts.chainId);
  702. }
  703. // Enable web3
  704. await Moralis.enableWeb3({
  705. //BSC mainnet = 0x38-56 testnet:0x61-97
  706. chainId: opts.chainId,
  707. privateKey: opts.privateKey,
  708. });
  709. logger.tlog("options id =", opts.chainId);
  710. var ret = await Moralis.transfer(options);
  711. logger.tlog("transfer 结果 =", ret);
  712. return toJson(SUCCEED_CODE, ret, "");
  713. } catch (error) {
  714. logger.tlog('transfer error:', error);
  715. logger.error('transfer error:', error.toString(), JSON.stringify(options))
  716. if (error.reason != null) {
  717. return toJson(ERROR_CODE_001, null, error.toString());
  718. } else {
  719. return toJson(ERROR_CODE_001, null, error.toString());;
  720. }
  721. }
  722. };
  723. const getAllTokenWithdrawInfoLists = async (obj) => {
  724. try {
  725. var key = reids_token_config.TOKENWITHDRAW;
  726. var ret = await redis.readRedis(key);
  727. return toJson(SUCCEED_CODE, ret, null);
  728. } catch (error) {
  729. console.error("getAllTokenWithdrawInfoLists=", error);
  730. return toJson(ERROR_CODE_001, null, error.toString());
  731. }
  732. }
  733. /**
  734. * 获取代币价格 -> usdPrice
  735. */
  736. const getAllTotkenPrice = async () => {
  737. try {
  738. logger.log('当前环境:', process.env.NODE_ENV);
  739. logger.log("getAllTotkenPrice in", reids_token_config); // Prints "value"
  740. var token_price_key = reids_token_config.TOKENPRICE;
  741. logger.log("getAllTotkenPrice token_price_key=", token_price_key);
  742. return await redis.readRedis(token_price_key)
  743. } catch (error) {
  744. logger.error("getTotkenPrice=", error);
  745. return toJson(ERROR_CODE_001, null, error.toString());
  746. }
  747. }
  748. function setTransfersDataType(type, ret) {
  749. if (ret && Array.isArray(ret) && ret.length > 0) {
  750. ret.forEach(element => {
  751. element.type = type;
  752. });
  753. }
  754. }
  755. //获取交易记录
  756. //hash 0xe09ba3a4c9f7a8902e01af68d0f1f91906f3f7db1195227e61c45c0e86b2630a
  757. async function getTokenTransfers(opt) {
  758. await initMasterSDK();
  759. logger.log("fun getTokenTransfers in ", opt);
  760. const options = {};
  761. options.type = 'all';
  762. options.chain = 'bsc_mainnet';
  763. if (opt.chain != null) {
  764. options.chain = utils.getChainName(opt.chain);
  765. logger.log('getTokenTransfers getChainName =', options.chain);
  766. }
  767. if (opt.order != null) {
  768. options.order = opt.order;
  769. }
  770. if (opt.startTime != null) {
  771. options.from_date = opt.startTime;
  772. }
  773. if (opt.endTime != null) {
  774. options.to_date = opt.endTime;
  775. }
  776. if (opt.from_block != null) {
  777. options.from_block = opt.from_block;
  778. }
  779. if (opt.to_block != null) {
  780. options.to_block = opt.to_block;
  781. }
  782. if (opt.transaction_hash) {
  783. options.transaction_hash = opt.transaction_hash;
  784. options.type = 'transaction_hash';
  785. }
  786. logger.log('getTokenTransfers >>>>>', options);
  787. if (options.type == 'all') {//查询主流币和 20 币所有的交易
  788. if (opt.address != null) {
  789. options.address = opt.address;
  790. } else {
  791. logger.error('getTokenTransfers error please check address parameter is ok ?', options);
  792. return toJson(ERROR_CODE_001, null, "please check address parameter is ok ?");
  793. }
  794. var t_1
  795. var t_2
  796. // Error: Request failed with status code 524
  797. // 做一次重试
  798. var tryCount = 4;
  799. var delay = 1000
  800. var interval = 500
  801. do {
  802. //主流币
  803. try {
  804. logger.log('getTokenTransfers account getTransactions>>>>>', options);
  805. t_1 = await Moralis.Web3API.account.getTransactions(options);
  806. logger.log('getTokenTransfers native ret -->>> t_1', t_1);
  807. setTransfersDataType('native', t_1.result)
  808. break
  809. } catch (error) {
  810. if (tryCount == 1) {
  811. logger.error("getTransactions error:", '主流币:', error.toString(), JSON.stringify(options))
  812. return toJson(ERROR_CODE_001, null, error.toString());
  813. }
  814. tryCount -= 1;
  815. await utils.sleep(delay)
  816. delay += interval
  817. }
  818. } while (tryCount >= 1);
  819. // Error: Request failed with status code 524
  820. tryCount = 4
  821. delay = 1000
  822. interval = 500
  823. do {
  824. //20币
  825. try {
  826. //token 获取交易记录如果没有时间有些地址会失败
  827. if (!options.to_block) {
  828. options.to_block = '10000000000'
  829. }
  830. logger.log('getTokenTransfers account getTokenTransfers>>>>>', options);
  831. t_2 = await Moralis.Web3API.account.getTokenTransfers(options);
  832. logger.log('getTokenTransfers token ret -->>> t_2', t_2);
  833. setTransfersDataType('token', t_2.result)
  834. break
  835. } catch (error) {
  836. if (tryCount == 1) {
  837. logger.error("getTransactions error:", 'token币:', error.toString(), JSON.stringify(options))
  838. return toJson(ERROR_CODE_001, null, error.toString());
  839. }
  840. tryCount -= 1;
  841. await utils.sleep(delay)
  842. delay += interval
  843. }
  844. } while (tryCount >= 1);
  845. //异常
  846. if (t_2 && t_2.total > 0 && Array.isArray(t_2.result) && t_2.result.length <= 0) {
  847. logger.error('getTokenTransfers token 数据异常 -->>>', t_2.toString(), JSON.stringify(options));
  848. return toJson(ERROR_CODE_001, null, 'token 数据异常.');
  849. }
  850. //排序组合
  851. try {
  852. let arr = t_1.result;
  853. let arr1 = t_2.result;
  854. if (Array.isArray(arr1) && Array.isArray(arr)) {
  855. let arr2 = arr.concat(arr1);
  856. t_1.result = arr2;
  857. logger.log('getTokenTransfers-->>> concat t_1', t_1);
  858. if (t_1.total) {
  859. t_1.total = t_1.result.length
  860. }
  861. }
  862. //将结果排序
  863. t_1.result.sort((a, b) => {
  864. let t1 = new Date(Date.parse(a.block_timestamp))
  865. let t2 = new Date(Date.parse(b.block_timestamp))
  866. return t2.getTime() - t1.getTime()
  867. })
  868. logger.log('getTokenTransfers-->>> sort t_1', t_1);
  869. return toJson(SUCCEED_CODE, t_1, null);
  870. } catch (error) {
  871. logger.error("getTransactions 排序组合 error :", error.toString(), JSON.stringify(options), JSON.stringify(t_1), JSON.stringify(t_2))
  872. return toJson(ERROR_CODE_001, null, error.toString());
  873. }
  874. } else if (options.type == 'transaction_hash') {//根据哈希查询
  875. var tryCount = 2
  876. do {
  877. try {
  878. logger.log('transaction_hash getTransaction options-->>> ', options);
  879. //native
  880. const transaction = await Moralis.Web3API.native.getTransaction(options);
  881. var arr = [];
  882. if (transaction)
  883. arr.push(transaction)
  884. var obj = { result: arr }
  885. logger.log('transaction_hash getTransaction ret-->>> ', transaction, obj);
  886. return toJson(SUCCEED_CODE, obj, null);
  887. } catch (error) {
  888. if (tryCount == 1) {
  889. logger.error("native getTransaction error:", error.toString(), JSON.stringify(options))
  890. return toJson(ERROR_CODE_001, null, error.toString());
  891. }
  892. tryCount -= 1;
  893. }
  894. } while (tryCount >= 1);
  895. } else {
  896. return toJson(ERROR_CODE_001, null, "This type is not supported.");;
  897. }
  898. }
  899. module.exports = {
  900. transfer,
  901. getTokenTransfers,
  902. toJson,
  903. getAllTokenWithdrawInfoLists,
  904. getAllTotkenPrice,
  905. withdraw,
  906. collectCoins,
  907. isTransferSucceed,
  908. getTransferRecordGasFree,
  909. pushCollectConisObj,
  910. getTransferGasFree,
  911. queryCollectBalance,
  912. queryCompanyInfoFromId,
  913. }