fix: read current password when constructing v3 wallet (#261)

* fix: read current password when constructing v3 wallet

* refactor: add readWalletPasswordOrThrow helper

jest.config.js

@@ -0,0 +1,6 @@
+module.exports = {
+  transform: { '^.+\\.ts?$': 'ts-jest' },
+  testEnvironment: 'node',
+  testRegex: '/test/.*\\.spec\\.ts$',
+  moduleFileExtensions: ['ts', 'tsx', 'js', 'jsx', 'json', 'node'],
+}

package.json

@@ -29,7 +29,7 @@
     "lint": "eslint --fix \"src/**/*.ts\" && cd ui && npm run lint",
     "lint:check": "eslint \"src/**/*.ts\" && cd ui && npm run lint:check",
     "check:types": "tsc --project tsconfig.test.json && cd ui && tsc --project tsconfig.test.json",
-    "test:unit": "jest --verbose -config=jest.config.ts "
+    "test:unit": "jest --verbose --config=jest.config.js"
   },
   "keywords": [
     "bee",

src/config-yaml.ts

@@ -22,3 +22,16 @@ export function writeConfigYaml(newValues: Record<string, unknown>) {
   }
   writeFileSync(getPath('config.yaml'), dump(data))
 }
+
+export function readWalletPasswordOrThrow(): string {
+  if (!configYamlExists()) {
+    throw Error('Attempted to read password, but config.yaml is not found')
+  }
+  const config = readConfigYaml()
+
+  if (!config.password) {
+    throw Error('Attempted to read password, but config.yaml does not contain it')
+  }
+
+  return config.password as string
+}

src/server.ts

@@ -13,7 +13,7 @@ import { URL } from 'url'
 import PACKAGE_JSON from '../package.json'
 import { getApiKey } from './api-key'
 import { sendBzzTransaction, sendNativeTransaction } from './blockchain'
-import { readConfigYaml, writeConfigYaml } from './config-yaml'
+import { readConfigYaml, readWalletPasswordOrThrow, writeConfigYaml } from './config-yaml'
 import { rebuildElectronTray } from './electron'
 import { createConfigFileAndAddress, createInitialTransaction, runLauncher } from './launcher'
 import { BeeManager } from './lifecycle'
@@ -197,7 +197,7 @@ export function runServer() {
 
 async function getPrivateKey(): Promise<string> {
   const v3 = await readFile(getPath(path.join('data-dir', 'keys', 'swarm.key')), 'utf-8')
-  const wallet = await Wallet.fromV3(v3, 'Test')
+  const wallet = await Wallet.fromV3(v3, readWalletPasswordOrThrow())
   const privateKeyString = wallet.getPrivateKeyString()
 
   return privateKeyString

test/config-yaml.spec.ts

@@ -0,0 +1,46 @@
+import { existsSync, unlinkSync, writeFileSync } from 'fs-extra'
+import { configYamlExists, readConfigYaml, readWalletPasswordOrThrow, writeConfigYaml } from '../src/config-yaml'
+
+jest.mock('env-paths', () =>
+  jest.fn().mockImplementation(() => ({
+    data: 'test/data',
+    config: 'test/data',
+    cache: 'test/data',
+    log: 'test/data',
+    temp: 'test/data',
+  })),
+)
+
+describe('config-yaml module', () => {
+  beforeEach(cleanUp)
+  afterAll(cleanUp)
+
+  test('check existence', () => {
+    expect(configYamlExists()).toBe(false)
+    writeFileSync('test/data/config.yaml', '')
+    expect(configYamlExists()).toBe(true)
+  })
+
+  test('read and write', () => {
+    writeFileSync('test/data/config.yaml', 'test: true')
+    writeConfigYaml({
+      abc: 123,
+    })
+    expect(readConfigYaml()).toHaveProperty('test', 'true')
+    expect(readConfigYaml()).toHaveProperty('abc', '123')
+  })
+
+  test('read password', () => {
+    expect(() => readWalletPasswordOrThrow()).toThrow('Attempted to read password, but config.yaml is not found')
+    writeFileSync('test/data/config.yaml', 'test: true')
+    expect(() => readWalletPasswordOrThrow()).toThrow('Attempted to read password, but config.yaml does not contain it')
+    writeFileSync('test/data/config.yaml', 'password: test')
+    expect(readWalletPasswordOrThrow()).toBe('test')
+  })
+})
+
+function cleanUp() {
+  if (existsSync('test/data/config.yaml')) {
+    unlinkSync('test/data/config.yaml')
+  }
+}