From c5d5445362dd7a1f33840da23b0e2b339d91fcbb Mon Sep 17 00:00:00 2001
From: futpib <futpib@gmail.com>
Date: Wed, 14 Nov 2018 16:53:13 +0300
Subject: [PATCH] Show monitor source edges

---
 components/graph/index.js       | 69 ++++++++++++++++++++++-----------
 components/preferences/index.js |  7 ++++
 index.css                       |  8 ++++
 reducers/preferences.js         |  1 +
 selectors/index.js              | 17 +++++++-
 5 files changed, 79 insertions(+), 23 deletions(-)

diff --git a/components/graph/index.js b/components/graph/index.js
index dc2caa8..09521a5 100644
--- a/components/graph/index.js
+++ b/components/graph/index.js
@@ -25,7 +25,10 @@ const {
 	icons: iconsActions,
 } = require('../../actions');
 
-const { getPaiByTypeAndIndex } = require('../../selectors');
+const {
+	getPaiByTypeAndIndex,
+	getDerivedMonitorSources,
+} = require('../../selectors');
 
 const {
 	PA_VOLUME_NORM,
@@ -50,6 +53,9 @@ const dgoToPai = new WeakMap();
 const key = pao => `${pao.type}-${pao.index}`;
 
 const sourceKey = pai => {
+	if (pai.type === 'monitorSource') {
+		return `sink-${pai.sinkIndex}`;
+	}
 	if (pai.clientIndex === -1) {
 		return `module-${pai.moduleIndex}`;
 	}
@@ -57,6 +63,9 @@ const sourceKey = pai => {
 };
 
 const targetKey = pai => {
+	if (pai.type === 'monitorSource') {
+		return `source-${pai.sourceIndex}`;
+	}
 	if (pai.type === 'sinkInput') {
 		return `sink-${pai.sinkIndex}`;
 	}
@@ -377,28 +386,36 @@ const ModuleText = ({ dgo, pai, state }) => r.div([
 	r(DebugText, { dgo, pai, state }),
 ]);
 
-const renderNodeText = state => (dgo, i, selected) => r('foreignObject', {
-	x: -s2,
-	y: -s2,
-}, r.div({
-	className: 'node-text',
-	style: {
-		width: size,
-		height: size,
+const renderNodeText = state => (dgo, i, selected) => {
+	const pai = dgoToPai.get(dgo);
 
-		backgroundImage: (icon => icon && `url(${icon})`)(state.icons[getPaiIcon(dgoToPai.get(dgo))]),
-	},
-}, r({
-	sink: SinkText,
-	source: SourceText,
-	client: ClientText,
-	module: ModuleText,
-}[dgo.type] || ModuleText, {
-	dgo,
-	pai: dgoToPai.get(dgo),
-	state,
-	selected,
-})));
+	if (!pai) {
+		return r(React.Fragment);
+	}
+
+	return r('foreignObject', {
+		x: -s2,
+		y: -s2,
+	}, r.div({
+		className: 'node-text',
+		style: {
+			width: size,
+			height: size,
+
+			backgroundImage: (icon => icon && `url(${icon})`)(state.icons[getPaiIcon(pai)]),
+		},
+	}, r({
+		sink: SinkText,
+		source: SourceText,
+		client: ClientText,
+		module: ModuleText,
+	}[dgo.type] || ModuleText, {
+		dgo,
+		pai,
+		state,
+		selected,
+	})));
+};
 
 const renderEdge = props => r(Edge, {
 	classSet: {
@@ -529,6 +546,7 @@ class Graph extends React.Component {
 		let edges = map(paoToEdge, flatten(map(values, [
 			this.props.objects.sinkInputs,
 			this.props.objects.sourceOutputs,
+			this.props.derivations.monitorSources,
 		])));
 
 		const connectedNodeKeys = new Set();
@@ -581,6 +599,9 @@ class Graph extends React.Component {
 		]))));
 
 		edges = filter(edge => {
+			if (this.props.preferences.hideMonitorSourceEdges && edge.type === 'monitorSource') {
+				return false;
+			}
 			return filteredNodeKeys.has(edge.source) && filteredNodeKeys.has(edge.target);
 		}, edges);
 
@@ -641,6 +662,10 @@ module.exports = connect(
 		objects: state.pulse.objects,
 		infos: state.pulse.infos,
 
+		derivations: {
+			monitorSources: getDerivedMonitorSources(state),
+		},
+
 		icons: state.icons,
 
 		preferences: state.preferences,
diff --git a/components/preferences/index.js b/components/preferences/index.js
index ff73a4a..d04f91d 100644
--- a/components/preferences/index.js
+++ b/components/preferences/index.js
@@ -71,6 +71,13 @@ const Preferences = withStateHandlers(
 
 	r.hr(),
 
+	r.div([
+		r(Checkbox, {
+			checked: props.preferences.hideMonitorSourceEdges,
+			onChange: () => props.actions.toggle('hideMonitorSourceEdges'),
+		}, 'Hide monitor source edges'),
+	]),
+
 	r.div([
 		r(Checkbox, {
 			checked: props.preferences.hideMonitors,
diff --git a/index.css b/index.css
index 7330337..df83085 100644
--- a/index.css
+++ b/index.css
@@ -130,6 +130,14 @@ div {
 	marker-end: url(#my-sink-arrow-selected);
 }
 
+.view-wrapper .monitorSource {
+	stroke-dasharray: 40;
+	pointer-events: none;
+}
+.view-wrapper .monitorSource .edge-mouse-handler {
+	pointer-events: none;
+}
+
 #edge-custom-container .edge-path {
 	marker-end: none;
 	stroke: var(--themeSelectedBgColor);
diff --git a/reducers/preferences.js b/reducers/preferences.js
index 553889d..aed51ae 100644
--- a/reducers/preferences.js
+++ b/reducers/preferences.js
@@ -13,6 +13,7 @@ const initialState = {
 	hideDisconnectedSources: false,
 	hideDisconnectedSinks: false,
 
+	hideMonitorSourceEdges: false,
 	hideMonitors: false,
 	hidePulseaudioApps: true,
 
diff --git a/selectors/index.js b/selectors/index.js
index e2488ad..d75e761 100644
--- a/selectors/index.js
+++ b/selectors/index.js
@@ -2,15 +2,30 @@
 const {
 	map,
 	prop,
+	path,
+	filter,
 	indexBy,
 } = require('ramda');
 
+const { createSelector } = require('reselect');
+
 const { things } = require('../constants/pulse');
 
 const storeKeyByType = map(prop('key'), indexBy(prop('type'), things));
 
-const getPaiByTypeAndIndex = (type, index) => state => state.pulse.infos[storeKeyByType[type]][index];
+const getPaiByTypeAndIndex = (type, index) => state => path([ storeKeyByType[type], index ], state.pulse.infos);
+
+const getDerivedMonitorSources = createSelector(
+	state => state.pulse.infos.sources,
+	sources => map(source => ({
+		index: source.index,
+		type: 'monitorSource',
+		sinkIndex: source.monitorSourceIndex,
+		sourceIndex: source.index,
+	}), filter(source => source.monitorSourceIndex >= 0, sources)),
+);
 
 module.exports = {
 	getPaiByTypeAndIndex,
+	getDerivedMonitorSources,
 };