Replace setCountry with setCustomRelay

2 files changed, 9 insertions, 5 deletions

diff --git a/app/lib/backend.js b/app/lib/backend.js
index 9567c91fd7..9d1fb21d35 100644
--- a/app/lib/backend.js
+++ b/app/lib/backend.js
@@ -231,8 +231,8 @@ export class Backend {
 
     this._store.dispatch(connectionActions.connectingTo(addr));
 
-
-    return this._ipc.setCountry(addr)
+    // TODO: Don't hardcode these values
+    return this._ipc.setCustomRelay(addr, 1300, 'udp')
       .then( () => {
         return this._ipc.connect();
       })
diff --git a/app/lib/ipc-facade.js b/app/lib/ipc-facade.js
index 063dfd11ae..0a9ceebf65 100644
--- a/app/lib/ipc-facade.js
+++ b/app/lib/ipc-facade.js
@@ -31,7 +31,7 @@ export interface IpcFacade {
   getAccountData(AccountToken): Promise<AccountData>,
   getAccount(): Promise<?AccountToken>,
   setAccount(accountToken: AccountToken): Promise<void>,
-  setCountry(address: string): Promise<void>,
+  setCustomRelay(host: string, port: number, protocol: 'udp' | 'tcp'): Promise<void>,
   connect(): Promise<void>,
   disconnect(): Promise<void>,
   getIp(): Promise<Ip>,
@@ -86,8 +86,12 @@ export class RealIpc implements IpcFacade {
     return;
   }
 
-  setCountry(address: string): Promise<void> {
-    return this._ipc.send('set_country', address)
+  setCustomRelay(host: string, port: number, protocol: 'udp' | 'tcp'): Promise<void> {
+    return this._ipc.send('set_custom_relay', {
+      host,
+      port,
+      protocol,
+    })
       .then(this._ignoreResponse);
   }