index.js 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455
  1. "use strict";
  2. var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
  3. return new (P || (P = Promise))(function (resolve, reject) {
  4. function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
  5. function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
  6. function step(result) { result.done ? resolve(result.value) : new P(function (resolve) { resolve(result.value); }).then(fulfilled, rejected); }
  7. step((generator = generator.apply(thisArg, _arguments || [])).next());
  8. });
  9. };
  10. var __importDefault = (this && this.__importDefault) || function (mod) {
  11. return (mod && mod.__esModule) ? mod : { "default": mod };
  12. };
  13. Object.defineProperty(exports, "__esModule", { value: true });
  14. const contact_book_1 = require("./contact-book");
  15. const crypto = require("libp2p-crypto");
  16. const ws_1 = __importDefault(require("ws"));
  17. const contact_item_1 = require("./contact-item");
  18. const content_item_1 = require("./content-item");
  19. const util_1 = require("./util");
  20. const contact_address_1 = require("./contact-address");
  21. class BankClient {
  22. constructor(urlBase, ipfsUrlBase, storage, webClient) {
  23. this.urlBase = urlBase;
  24. this.ipfsUrlBase = ipfsUrlBase;
  25. this.storage = storage;
  26. this.webClient = webClient;
  27. this.wsUrlBase = urlBase.replace(/^http/i, 'ws');
  28. }
  29. static parseBankLink(bankLink) {
  30. if (!bankLink.startsWith('bank:')) {
  31. throw new Error('address must start with bank:');
  32. }
  33. const deprefixed = bankLink.substring(5);
  34. let host;
  35. let address;
  36. let topic;
  37. if (deprefixed[0] === '/' && deprefixed[1] === '/') {
  38. [host, address, topic] = deprefixed.substring(2).split('/');
  39. }
  40. else {
  41. [address, topic] = deprefixed.split('/');
  42. }
  43. if (!address || !topic) {
  44. throw new Error('cannot parse address and topic');
  45. }
  46. return { host, address, topic };
  47. }
  48. getPub() {
  49. return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
  50. yield this.bootstrap();
  51. this.getPriv().id((idErr, pubHash) => {
  52. if (idErr) {
  53. return reject(idErr);
  54. }
  55. resolve(pubHash);
  56. });
  57. }));
  58. }
  59. bootstrap() {
  60. if (this.bootstrapResult) {
  61. return Promise.resolve(this.bootstrapResult);
  62. }
  63. if (this.bootstrapPromise) {
  64. return this.bootstrapPromise;
  65. }
  66. return this.bootstrapPromise = new Promise((resolve, reject) => {
  67. this.storage.get('notaprivatekey').then(privateKeyFromStorage => {
  68. if (privateKeyFromStorage == null) {
  69. console.log('no private key in storage. generating new');
  70. crypto.keys.generateKeyPair('RSA', 2048, (generateErr, privateKey) => {
  71. if (generateErr) {
  72. return reject(generateErr);
  73. }
  74. privateKey.export('password', (exportErr, exportResult) => {
  75. if (exportErr) {
  76. return reject(exportErr);
  77. }
  78. this.storage.set('notaprivatekey', exportResult).then(err => {
  79. // whatever
  80. }).catch(reject);
  81. this.privateKey = privateKey;
  82. resolve(true);
  83. });
  84. });
  85. }
  86. else {
  87. // console.log('importing privatekey');
  88. crypto.keys.import(privateKeyFromStorage, 'password', (err, importedPrivateKey) => {
  89. if (err) {
  90. return reject(err);
  91. }
  92. this.privateKey = importedPrivateKey;
  93. // console.log(this.getPublicKeyString());
  94. // console.log(privateKeyFromStorage);
  95. resolve(true);
  96. });
  97. }
  98. }).catch(reject);
  99. });
  100. }
  101. getNonce() {
  102. return __awaiter(this, void 0, void 0, function* () {
  103. const nonce = yield this.webClient.request({
  104. method: 'GET',
  105. url: this.urlBase + '/bank/nonce'
  106. });
  107. return Number(nonce);
  108. });
  109. }
  110. getBalance() {
  111. return __awaiter(this, void 0, void 0, function* () {
  112. const nonce = yield this.getNonce();
  113. const retrieveRequest = yield this.makePlaintextPayload(JSON.stringify({
  114. _date: new Date().toISOString(),
  115. _nonce: nonce
  116. }));
  117. const topicURL = this.urlBase + '/bank/getbalance';
  118. const postResponse = yield this.webClient.requestJSON({
  119. body: retrieveRequest,
  120. method: 'POST',
  121. url: topicURL
  122. });
  123. return postResponse.balance;
  124. });
  125. }
  126. upload(params) {
  127. return __awaiter(this, void 0, void 0, function* () {
  128. const url = this.urlBase + '/bank/upload';
  129. const formData = {};
  130. formData.creator = yield this.getPub();
  131. if (params.fileData) {
  132. formData.file = {
  133. value: params.fileData,
  134. options: {
  135. filename: params.fileName
  136. }
  137. };
  138. }
  139. if (params.thumbFileData) {
  140. formData.thumb = {
  141. value: params.thumbFileData,
  142. options: {
  143. filename: params.thumbFileName
  144. }
  145. };
  146. }
  147. if (params.links) {
  148. formData.links = JSON.stringify(params.links);
  149. }
  150. for (const attr of ['title', 'text', 'type']) {
  151. if (params[attr] != null) {
  152. formData[attr] = params[attr];
  153. }
  154. }
  155. // console.log('formData', formData);
  156. const uploadResponse = yield this.webClient.requestJSON({
  157. formData,
  158. method: 'POST',
  159. url
  160. });
  161. // console.log('uploadResponse', uploadResponse);
  162. return uploadResponse.hash;
  163. });
  164. }
  165. uploadSlimJSON(item) {
  166. return __awaiter(this, void 0, void 0, function* () {
  167. const url = this.urlBase + '/bank/upload/slim';
  168. const uploadResponse = yield this.webClient.requestJSON({
  169. body: item,
  170. method: 'POST',
  171. url
  172. });
  173. // console.log('uploadResponse', uploadResponse);
  174. return uploadResponse.hash;
  175. });
  176. }
  177. uploadSlimText(item) {
  178. return __awaiter(this, void 0, void 0, function* () {
  179. const url = this.urlBase + '/bank/upload/slim';
  180. const uploadResponse = JSON.parse(yield this.webClient.request({
  181. body: item,
  182. headers: {
  183. 'content-type': 'text/plain'
  184. },
  185. method: 'POST',
  186. url
  187. }));
  188. // console.log('uploadResponse', uploadResponse);
  189. return uploadResponse.hash;
  190. });
  191. }
  192. appendPrivate(peerAddr, topic, hash, replaceHash, deleteHash) {
  193. return __awaiter(this, void 0, void 0, function* () {
  194. const nonce = yield this.getNonce();
  195. const payload = yield this.makePlaintextPayload(JSON.stringify({
  196. _date: new Date().toISOString(),
  197. _nonce: nonce,
  198. deleteHash,
  199. hash,
  200. replaceHash,
  201. }));
  202. const topicURL = this.urlBase + '/bank/private/' + encodeURIComponent(peerAddr) + '/' + encodeURIComponent(topic);
  203. const result = yield this.webClient.request({
  204. body: JSON.stringify(payload),
  205. headers: {
  206. 'content-type': 'application/json'
  207. },
  208. method: 'PUT',
  209. url: topicURL
  210. });
  211. });
  212. }
  213. retrievePrivate(peerAddr, topic) {
  214. return __awaiter(this, void 0, void 0, function* () {
  215. const nonce = yield this.getNonce();
  216. const retrieveRequest = yield this.makePlaintextPayload(JSON.stringify({
  217. _date: new Date().toISOString(),
  218. _nonce: nonce
  219. }));
  220. const topicURL = this.urlBase + '/bank/private/' + encodeURIComponent(peerAddr) + '/' + encodeURIComponent(topic);
  221. const result = yield this.webClient.request({
  222. body: JSON.stringify(retrieveRequest),
  223. headers: {
  224. 'content-type': 'application/json'
  225. },
  226. method: 'POST',
  227. url: topicURL
  228. });
  229. return result;
  230. });
  231. }
  232. subscribePrivate(peerAddr, topic, connectCallback, messageCallback) {
  233. return __awaiter(this, void 0, void 0, function* () {
  234. yield this.connectWebsocket(peerAddr, topic, connectCallback, messageCallback);
  235. });
  236. }
  237. getOrCreateContact(peerId, addressType, addressValue) {
  238. return __awaiter(this, void 0, void 0, function* () {
  239. const contactBook = yield this.getContactBook(peerId);
  240. const existing = contactBook.lookupByAddress(addressType, addressValue);
  241. if (existing != null) {
  242. return existing;
  243. }
  244. return yield this.createContact(peerId, addressType, addressValue);
  245. });
  246. }
  247. createContact(peerId, addressType, addressValue) {
  248. return __awaiter(this, void 0, void 0, function* () {
  249. const contactId = util_1.uuid();
  250. const newItem = {
  251. addrs: [],
  252. id: contactId
  253. };
  254. if (addressType != null && addressValue != null) {
  255. newItem.addrs.push(new contact_address_1.ContactAddress(addressType, addressValue).toPrefixedString());
  256. }
  257. const newItemHash = yield this.uploadSlimJSON(newItem);
  258. yield this.appendPrivate(peerId, '📇', newItemHash);
  259. const contactBook2 = yield this.getContactBook(peerId);
  260. return (yield contactBook2.lookupById(contactId));
  261. });
  262. }
  263. getAllContacts(peerId) {
  264. return __awaiter(this, void 0, void 0, function* () {
  265. const contactList = yield this.retrievePrivate(peerId, '📇');
  266. const items = yield this.getItemsForCommaList(contactList);
  267. return items.map(data => new contact_item_1.ContactItem(data));
  268. });
  269. }
  270. getContactBook(peerId) {
  271. return __awaiter(this, void 0, void 0, function* () {
  272. if (peerId == null) {
  273. throw new Error('Missing peerId');
  274. }
  275. return new contact_book_1.ContactBook(yield this.getAllContacts(peerId));
  276. });
  277. }
  278. updateContact(peerId, contactId, newProperties) {
  279. return __awaiter(this, void 0, void 0, function* () {
  280. const contactBook = yield this.getContactBook(peerId);
  281. const existing = yield contactBook.lookupById(contactId);
  282. if (!existing) {
  283. throw new Error('missing contact with id ' + contactId);
  284. }
  285. const existingData = existing.getData();
  286. const newProps = util_1.mergeDeep({}, newProperties);
  287. delete newProps.id;
  288. const newItem = util_1.mergeDeep(existingData, newProps);
  289. delete newItem.hash;
  290. newItem.lastChanged = new Date().toISOString();
  291. const newItemHash = yield this.uploadSlimJSON(newItem);
  292. yield this.appendPrivate(peerId, '📇', newItemHash, existing.hash);
  293. const contactBook2 = yield this.getContactBook(peerId);
  294. return (yield contactBook2.lookupById(contactId));
  295. });
  296. }
  297. getContentItemByHash(hash) {
  298. return __awaiter(this, void 0, void 0, function* () {
  299. hash = this.parseItemHash(hash).hash;
  300. const contentParams = (yield this.webClient.requestJSON({
  301. method: 'get',
  302. url: this.ipfsUrlBase + '/ipfs/' + hash + '/content.json'
  303. }));
  304. return new content_item_1.ContentItem(hash, contentParams);
  305. });
  306. }
  307. getItemsForCommaList(commaList) {
  308. return __awaiter(this, void 0, void 0, function* () {
  309. const itemHashes = commaList.split(',').filter(x => x.trim() !== '');
  310. const items = yield Promise.all(itemHashes.map(itemId => {
  311. const itemHash = this.parseItemHash(itemId).hash;
  312. return this.webClient.requestJSON({
  313. method: 'get',
  314. url: this.ipfsUrlBase + '/ipfs/' + itemHash,
  315. });
  316. }));
  317. for (const item of items) {
  318. item.hash = itemHashes.shift();
  319. }
  320. return items;
  321. });
  322. }
  323. parseItemHash(itemHash) {
  324. let type = null;
  325. let timestamp = null;
  326. let hash = null;
  327. if (itemHash.startsWith('/ipfs/')) {
  328. itemHash = itemHash.substring(6);
  329. }
  330. const matched = itemHash.match(/^([0-9]*)_(..)_(.*)$/);
  331. if (matched) {
  332. timestamp = matched[1];
  333. type = matched[2];
  334. hash = matched[3];
  335. }
  336. if (!type) {
  337. type = 'CO';
  338. }
  339. if (!hash) {
  340. hash = itemHash;
  341. }
  342. return { type, timestamp, hash };
  343. }
  344. runAgent(address, topic, storage, itemProcessCallback) {
  345. return __awaiter(this, void 0, void 0, function* () {
  346. yield this.subscribePrivate(address, topic, () => {
  347. // console.log('websocket connected');
  348. }, (msg) => __awaiter(this, void 0, void 0, function* () {
  349. console.log('Got new item notification', msg);
  350. yield agentUpdate();
  351. }));
  352. setTimeout(() => __awaiter(this, void 0, void 0, function* () {
  353. yield agentUpdate();
  354. }), 0);
  355. const agentUpdate = () => __awaiter(this, void 0, void 0, function* () {
  356. const agentConfig = (yield storage.get('config')) || {};
  357. const items = yield this.retrievePrivate(address, topic);
  358. const itemsList = items.split(',').filter((x) => x.trim() !== '');
  359. for (const itemId of itemsList) {
  360. const processed = agentConfig.processed || [];
  361. if (processed.includes(itemId)) {
  362. continue;
  363. }
  364. try {
  365. const item = yield this.getContentItemByHash(itemId);
  366. yield itemProcessCallback(item);
  367. processed.push(itemId);
  368. agentConfig.processed = processed;
  369. yield storage.set('config', agentConfig);
  370. }
  371. catch (e) {
  372. console.error('error processing item', itemId, e);
  373. }
  374. }
  375. console.log('Finished checking for items');
  376. });
  377. });
  378. }
  379. connectWebsocket(peerAddr, topic, connectCallback, messageCallback) {
  380. return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
  381. const nonce = yield this.getNonce();
  382. const retrieveRequest = yield this.makePlaintextPayload(JSON.stringify({
  383. _date: new Date().toISOString(),
  384. _nonce: nonce,
  385. addr: peerAddr,
  386. topic: topic
  387. }));
  388. const jsonOutput = JSON.stringify(retrieveRequest);
  389. const base64ed = Buffer.from(jsonOutput).toString('base64');
  390. const encoded = encodeURIComponent(base64ed);
  391. const ws = new ws_1.default(this.wsUrlBase + '/bank/ws?arg=' + encoded);
  392. ws.on('open', () => {
  393. connectCallback();
  394. });
  395. ws.on('message', data => {
  396. messageCallback(data);
  397. });
  398. const reconnect = () => {
  399. // console.log('reconnect');
  400. try {
  401. ws.terminate();
  402. }
  403. finally {
  404. console.log('reconnecting in 5s');
  405. setTimeout(() => {
  406. this.connectWebsocket(peerAddr, topic, connectCallback, messageCallback);
  407. }, 5000);
  408. }
  409. };
  410. ws.on('error', err => {
  411. console.error('websocket error', err);
  412. });
  413. ws.on('close', err => {
  414. reconnect();
  415. });
  416. resolve();
  417. }));
  418. }
  419. getPriv() {
  420. if (!this.privateKey) {
  421. throw new Error('missing private key');
  422. }
  423. return this.privateKey;
  424. }
  425. makePlaintextPayload(message) {
  426. const messageBytes = Buffer.from(message, 'utf-8');
  427. return new Promise((resolve, reject) => __awaiter(this, void 0, void 0, function* () {
  428. yield this.bootstrap();
  429. this.privateKey.sign(messageBytes, (signErr, signatureBytes) => __awaiter(this, void 0, void 0, function* () {
  430. if (signErr) {
  431. reject(signErr);
  432. return;
  433. }
  434. const publicDERBytes = this.privateKey.public.bytes;
  435. this.privateKey.id((idErr, pubHash) => {
  436. if (idErr) {
  437. reject(idErr);
  438. return;
  439. }
  440. const result = {
  441. date: new Date().toISOString(),
  442. msg: util_1.encodeHex(messageBytes),
  443. pub: util_1.encodeHex(publicDERBytes),
  444. pubHash,
  445. sig: util_1.encodeHex(signatureBytes),
  446. };
  447. // console.log('result', result, signatureBytes);
  448. resolve(result);
  449. });
  450. }));
  451. }));
  452. }
  453. }
  454. exports.BankClient = BankClient;
  455. //# sourceMappingURL=index.js.map