From 9a08ab68bc4bb9ee64e46c22bb3b0eed4fb60b3e Mon Sep 17 00:00:00 2001
From: Audric Ackermann <audric@loki.network>
Date: Mon, 21 Jun 2021 12:08:51 +1000
Subject: [PATCH] fix onion path status light for orange color

---
 ts/components/OnionStatusDialog.tsx | 34 +++++++++++++----------------
 ts/session/onions/onionPath.ts      |  2 +-
 ts/state/ducks/onion.tsx            |  8 +++----
 ts/state/selectors/onions.ts        | 23 +++++++++++++++++++
 4 files changed, 43 insertions(+), 24 deletions(-)
 create mode 100644 ts/state/selectors/onions.ts

diff --git a/ts/components/OnionStatusDialog.tsx b/ts/components/OnionStatusDialog.tsx
index 90c2685c4..976546d98 100644
--- a/ts/components/OnionStatusDialog.tsx
+++ b/ts/components/OnionStatusDialog.tsx
@@ -19,6 +19,11 @@ import { useTheme } from 'styled-components';
 import { useNetwork } from '../hooks/useNetwork';
 import { Snode } from '../data/data';
 import { onionPathModal } from '../state/ducks/modalDialog';
+import {
+  getFirstOnionPath,
+  getFirstOnionPathLength,
+  getOnionPathsCount,
+} from '../state/selectors/onions';
 
 export type StatusLightType = {
   glowStartDelay: number;
@@ -27,8 +32,7 @@ export type StatusLightType = {
 };
 
 const OnionPathModalInner = () => {
-  const onionNodes = useSelector((state: StateType) => state.onionPaths.snodePath);
-  const onionPath = onionNodes;
+  const onionPath = useSelector(getFirstOnionPath);
   // including the device and destination in calculation
   const glowDuration = onionPath.length + 2;
 
@@ -36,7 +40,7 @@ const OnionPathModalInner = () => {
     {
       label: window.i18n('device'),
     },
-    ...onionNodes,
+    ...onionPath,
     {
       label: window.i18n('destination'),
     },
@@ -125,29 +129,21 @@ export const ActionPanelOnionStatusLight = (props: {
 }) => {
   const { isSelected, handleClick } = props;
 
-  let iconColor;
   const theme = useTheme();
-  const firstOnionPath = useSelector((state: StateType) => state.onionPaths.snodePath);
-  const hasOnionPath = firstOnionPath.length > 2;
+  const onionPathsCount = useSelector(getOnionPathsCount);
+  const firstPathLength = useSelector(getFirstOnionPathLength);
+  const isOnline = useNetwork();
 
   // Set icon color based on result
   const red = theme.colors.destructive;
   const green = theme.colors.accent;
   const orange = theme.colors.warning;
 
-  iconColor = hasOnionPath ? theme.colors.accent : theme.colors.destructive;
-  const onionState = useSelector((state: StateType) => state.onionPaths);
-
-  iconColor = red;
-  const isOnline = useNetwork();
-  if (!(onionState && onionState.snodePath) || !isOnline) {
-    iconColor = red;
-  } else {
-    const onionSnodePath = onionState.snodePath;
-    if (onionState && onionSnodePath && onionSnodePath.length > 0) {
-      const onionNodeCount = onionSnodePath.length;
-      iconColor = onionNodeCount > 2 ? green : onionNodeCount > 1 ? orange : red;
-    }
+  // start with red
+  let iconColor = red;
+  //if we are not online or the first path is not valid, we keep red as color
+  if (isOnline && firstPathLength > 1) {
+    iconColor = onionPathsCount > 2 ? green : onionPathsCount > 1 ? orange : red;
   }
 
   return (
diff --git a/ts/session/onions/onionPath.ts b/ts/session/onions/onionPath.ts
index 8160bb951..1154ca408 100644
--- a/ts/session/onions/onionPath.ts
+++ b/ts/session/onions/onionPath.ts
@@ -136,7 +136,7 @@ export async function getOnionPath(toExclude?: Snode): Promise<Array<Snode>> {
   if (onionPaths.length <= 0) {
     window.inboxStore?.dispatch(updateOnionPaths([]));
   } else {
-    window.inboxStore?.dispatch(updateOnionPaths(onionPaths[0]));
+    window.inboxStore?.dispatch(updateOnionPaths(onionPaths));
   }
 
   const onionPathsWithoutExcluded = toExclude
diff --git a/ts/state/ducks/onion.tsx b/ts/state/ducks/onion.tsx
index 4c6566762..2c607adf3 100644
--- a/ts/state/ducks/onion.tsx
+++ b/ts/state/ducks/onion.tsx
@@ -2,11 +2,11 @@ import { createSlice, PayloadAction } from '@reduxjs/toolkit';
 import { Snode } from '../../data/data';
 
 export type OnionState = {
-  snodePath: Array<Snode>;
+  snodePaths: Array<Array<Snode>>;
 };
 
 export const initialOnionPathState = {
-  snodePath: new Array<Snode>(),
+  snodePaths: new Array<Array<Snode>>(),
 };
 
 /**
@@ -16,8 +16,8 @@ const onionSlice = createSlice({
   name: 'onionPaths',
   initialState: initialOnionPathState,
   reducers: {
-    updateOnionPaths(state: OnionState, action: PayloadAction<Array<Snode>>) {
-      return { snodePath: action.payload };
+    updateOnionPaths(state: OnionState, action: PayloadAction<Array<Array<Snode>>>) {
+      return { snodePaths: action.payload };
     },
   },
 });
diff --git a/ts/state/selectors/onions.ts b/ts/state/selectors/onions.ts
new file mode 100644
index 000000000..1e3d95a0d
--- /dev/null
+++ b/ts/state/selectors/onions.ts
@@ -0,0 +1,23 @@
+import { createSelector } from 'reselect';
+
+import { StateType } from '../reducer';
+import { SectionType } from '../../components/session/ActionsPanel';
+import { OnionState } from '../ducks/onion';
+import { Snode } from '../../data/data';
+
+export const getOnionPaths = (state: StateType): OnionState => state.onionPaths;
+
+export const getOnionPathsCount = createSelector(
+  getOnionPaths,
+  (state: OnionState): SectionType => state.snodePaths.length
+);
+
+export const getFirstOnionPath = createSelector(
+  getOnionPaths,
+  (state: OnionState): Array<Snode> => state.snodePaths?.[0] || []
+);
+
+export const getFirstOnionPathLength = createSelector(
+  getFirstOnionPath,
+  (state: Array<Snode>): number => state.length || 0
+);