whatsapp-nodejs 0.0.1-security → 0.0.1

package/src/schema/message.js

@@ -0,0 +1,44 @@
+module.exports = {
+  mobile: {
+    type: String,
+    index: true,
+  },
+  to: {
+    type: String,
+    name: '接收者',
+    index: true,
+  },
+  mediaType: {
+    type: String,
+    default: 'text',
+  },
+  from: {
+    type: String,
+    index: true,
+  },
+  id: {
+    type: String,
+    name: '消息 id',
+    index: true,
+  },
+  attrs: {
+    type: Object,
+  },
+  record: {
+    type: String,
+    name: '消息内容',
+  },
+  timestamp: {
+    type: Number,
+  },
+  isRead: {
+    type: Boolean,
+    default: false,
+    index: true,
+  },
+  retry: {
+    type: Number,
+    name: '重试次数',
+    default: 0,
+  },
+};

package/src/schema/prekey.js

@@ -0,0 +1,51 @@
+module.exports = {
+  mobile: {
+    type: String,
+    index: true,
+  },
+  keyId: {
+    type: Number,
+    index: true,
+  },
+  sent_to_server: {
+    type: Boolean,
+    default: false,
+  },
+  preKey: {
+    type: Object,
+    set: val => {
+      const { keyPair } = val;
+      return {
+        ...val,
+        keyPair: {
+          pubKey: Buffer.from(keyPair.pubKey),
+          privKey: Buffer.from(keyPair.privKey),
+        },
+      };
+    },
+    get: val => {
+      const { keyPair } = val;
+      const identityKeyPair = {
+        pubKey: new ArrayBuffer(keyPair.pubKey.buffer.length),
+        privKey: new ArrayBuffer(keyPair.privKey.buffer.length),
+      };
+      const pubKey = new Uint8Array(identityKeyPair.pubKey);
+      const privKey = new Uint8Array(identityKeyPair.privKey);
+      for (let i = 0; i < keyPair.pubKey.buffer.length; i++) {
+        const item = keyPair.pubKey.buffer[i];
+        pubKey[i] = item;
+      }
+      for (let i = 0; i < keyPair.privKey.buffer.length; i++) {
+        const item = keyPair.privKey.buffer[i];
+        privKey[i] = item;
+      }
+      return {
+        ...val,
+        keyPair: identityKeyPair,
+      };
+    },
+  },
+  timestamp: {
+    type: Number,
+  },
+};

package/src/schema/recvmessage.js

@@ -0,0 +1,33 @@
+module.exports = {
+  mobile: {
+    type: String,
+    index: true,
+  },
+  isRead: {
+    type: Boolean,
+    index: true,
+    default: false,
+  },
+  jid: {
+    type: String,
+    name: '来自',
+    index: true,
+  },
+  gid: {
+    type: String,
+    name: '群 id',
+    default: '',
+  },
+  id: {
+    type: String,
+    name: '消息 id',
+    index: true,
+  },
+  message: {
+    type: Object,
+  },
+  t: {
+    title: '接收消息时间',
+    type: Number,
+  },
+};

package/src/schema/senderkey.js

@@ -0,0 +1,20 @@
+module.exports = {
+  mobile: {
+    type: String,
+    index: true,
+  },
+  groupId: {
+    type: String,
+    index: true,
+  },
+  senderId: {
+    type: String,
+    index: true,
+  },
+  record: {
+    type: String,
+  },
+  timestamp: {
+    type: Number,
+  },
+};

package/src/schema/session.js

@@ -0,0 +1,22 @@
+module.exports = {
+  mobile: {
+    type: String,
+    index: true,
+  },
+  address: {
+    type: String,
+    index: true,
+  },
+  jid: {
+    type: Number,
+  },
+  deviceId: {
+    type: String,
+  },
+  record: {
+    type: String,
+  },
+  timestamp: {
+    type: Number,
+  },
+};

package/src/schema/signedprekeys.js

@@ -0,0 +1,56 @@
+module.exports = {
+  mobile: {
+    type: String,
+    index: true,
+  },
+  keyId: {
+    type: Number,
+  },
+  preKey: {
+    type: Object,
+    set: val => {
+      const { keyPair } = val;
+      return {
+        ...val,
+        keyPair: {
+          pubKey: Buffer.from(keyPair.pubKey),
+          privKey: Buffer.from(keyPair.privKey),
+        },
+        signature: Buffer.from(val.signature),
+      };
+    },
+    get: val => {
+      const { keyPair } = val;
+      const identityKeyPair = {
+        pubKey: new ArrayBuffer(keyPair.pubKey.buffer.length),
+        privKey: new ArrayBuffer(keyPair.privKey.buffer.length),
+      };
+      const signature = new ArrayBuffer(val.signature.buffer.length);
+
+      const pubKey = new Uint8Array(identityKeyPair.pubKey);
+      const privKey = new Uint8Array(identityKeyPair.privKey);
+      const sign = new Uint8Array(signature);
+
+      for (let i = 0; i < keyPair.pubKey.buffer.length; i++) {
+        const item = keyPair.pubKey.buffer[i];
+        pubKey[i] = item;
+      }
+      for (let i = 0; i < keyPair.privKey.buffer.length; i++) {
+        const item = keyPair.privKey.buffer[i];
+        privKey[i] = item;
+      }
+      for (let i = 0; i < val.signature.buffer.length; i++) {
+        const item = val.signature.buffer[i];
+        sign[i] = item;
+      }
+      return {
+        ...val,
+        keyPair: identityKeyPair,
+        signature,
+      };
+    },
+  },
+  timestamp: {
+    type: Number,
+  },
+};

package/test/WASocketClient.test.js

@@ -0,0 +1,23 @@
+const logger = require('../src/logger');
+const WASocketClient = require('../src/WASocketClient');
+
+const main = async () => {
+  //
+
+  const waSocketClient = new WASocketClient({
+    proxy: {
+      host: 'localhost',
+      port: 1080,
+    },
+    mobile: '233244040964',
+    socketName: 'Socket_1',
+    endpoint: {
+      host: 'e1.whatsapp.net',
+      port: 443,
+    },
+  });
+
+  await waSocketClient.init();
+};
+
+main();

package/test/handshake.test.js

@@ -0,0 +1,38 @@
+const logger = require('../src/logger');
+const WASocketClient = require('../src/WASocketClient');
+
+const HandShake = require('../src/protocol/HandShake');
+const db = require('../src/db');
+const libsignal = require('../src/lib/libsignal-protocol');
+const config = require('../src/config');
+
+const main = async () => {
+  //
+  libsignal.curve = (await libsignal.default()).Curve;
+  await db.init();
+  const waSocketClient = new WASocketClient({
+    proxy: {
+      host: '127.0.0.1',
+      port: 1086,
+      type: 'socks5',
+    },
+    mobile: '233244040964',
+    socketName: 'Socket_1',
+    endpoint: config.getEndPoint(),
+  });
+
+  await waSocketClient.init();
+
+  const account = await db.findAccount(233244040964);
+  console.log('account', account);
+
+  const handShake = new HandShake(waSocketClient);
+
+  await handShake.start(account, account.serverStaticPublic);
+
+  handShake.waSocketClient.on('data', data => {
+    // console.log('data', data);
+  });
+};
+
+main();

package/test/logger.test.js

@@ -0,0 +1,17 @@
+const logger = require('../src/logger');
+
+console.log('log,info');
+console.trace('trace');
+console.debug('debug');
+console.log('log');
+console.info('info');
+console.warn('warn');
+console.error('error');
+console.fatal('fatal');
+
+// const e = new Error('失败');
+// console.error(e, e.message, e.stack);
+
+const childLogger = logger.child({ reqId: '1' }, false);
+
+childLogger.info('hi', 'a');

package/test/whatsapp.test.js

@@ -0,0 +1,17 @@
+const Whatsapp = require('../src/whatsapp');
+
+const main = async () => {
+  const whatsapp = new Whatsapp();
+  await whatsapp.init({
+    mobile: '233244040964',
+    proxy: {
+      host: '127.0.0.1',
+      port: 1080,
+    },
+  });
+
+  const res = await whatsapp.login();
+  console.log('res', res);
+};
+
+main();