From d79286d7575b7dd98d58c14d759baf1fd8e68bc6 Mon Sep 17 00:00:00 2001
From: Roberto Alsina <ralsina@netmanagers.com.ar>
Date: Fri, 31 Jan 2020 18:18:24 -0300
Subject: [PATCH] Implement dialog action buttons

---
 main.py | 36 ++++++++++++++++++++++++++++++++++++
 main.ui | 30 ++++++++----------------------
 2 files changed, 44 insertions(+), 22 deletions(-)

diff --git a/main.py b/main.py
index ff5f0c8..574a68d 100644
--- a/main.py
+++ b/main.py
@@ -9,6 +9,22 @@ from PySide2.QtWidgets import QApplication, QGraphicsScene
 from monitor_item import MonitorItem
 
 
+def gen_xrandr_from_data(data):
+    """Takes monitor data and generates a xrandr command line."""
+    cli = ['xrandr']
+    for name, mon in data.items():
+        cli.append(f'--output {name}')
+        cli.append(f'--pos {int(mon["pos_x"])}x{int(mon["pos_y"])}')
+        cli.append(f'--mode {mon["current_mode"]}')
+        mod_x, mod_y = [int(n) for n in mon['current_mode'].split('x')]
+        cli.append(f'--scale {mon["res_x"]/mod_x}x{mon["res_y"]/mod_y}')
+        if mon['primary']:
+            cli.append('--primary')
+        if not mon['enabled']:
+            cli.append('--off')
+
+    return ' '.join(cli)
+
 def parse_monitor(line):
     parts = line.split()
     name = parts[0]
@@ -60,6 +76,23 @@ class Window(QObject):
         self.ui.horizontalScale.valueChanged.connect(self.scale_changed)
         self.ui.verticalScale.valueChanged.connect(self.scale_changed)
         self.ui.modes.currentTextChanged.connect(self.mode_changed)
+        self.ui.applyButton.clicked.connect(self.do_apply)
+        self.ui.okButton.clicked.connect(self.do_ok)
+        self.ui.resetButton.clicked.connect(self.do_reset)
+        self.ui.cancelButton.clicked.connect(self.ui.reject)
+
+    def do_reset(self):
+        for n in self.xrandr_info:
+            self.xrandr_info[n].update(self.orig_xrandr_info[n])
+        self.fill_ui()
+
+    def do_ok(self):
+        self.do_apply()
+        self.ui.accept()
+
+    def do_apply(self):
+        cli = gen_xrandr_from_data(self.xrandr_info)
+        print(cli)
 
     def fill_ui(self):
         """Load data from xrandr and setup the whole thing."""
@@ -74,6 +107,7 @@ class Window(QObject):
             self.scene.addItem(mon_item)
             monitor["item"] = mon_item
         self.adjust_view()
+        self.scale_changed()  # Trigger scale labels update
 
     def mode_changed(self):
         mon = self.ui.screenCombo.currentText()
@@ -153,6 +187,8 @@ class Window(QObject):
                     self.xrandr_info[a]["replica_of"].append(b)
 
     def monitor_selected(self, name):
+        if not name:
+            return
         # needed so we don't flip through all modes as they are added
         self.ui.modes.blockSignals(True)
         # Show modes
diff --git a/main.ui b/main.ui
index 4ee81b5..8f632af 100644
--- a/main.ui
+++ b/main.ui
@@ -168,23 +168,9 @@
    <item>
     <layout class="QHBoxLayout" name="horizontalLayout">
      <item>
-      <widget class="QPushButton" name="pushButton_6">
+      <widget class="QPushButton" name="resetButton">
        <property name="text">
-        <string>PushButton</string>
-       </property>
-      </widget>
-     </item>
-     <item>
-      <widget class="QPushButton" name="pushButton_5">
-       <property name="text">
-        <string>PushButton</string>
-       </property>
-      </widget>
-     </item>
-     <item>
-      <widget class="QPushButton" name="pushButton_4">
-       <property name="text">
-        <string>PushButton</string>
+        <string>Reset</string>
        </property>
       </widget>
      </item>
@@ -202,23 +188,23 @@
       </spacer>
      </item>
      <item>
-      <widget class="QPushButton" name="pushButton_3">
+      <widget class="QPushButton" name="okButton">
        <property name="text">
-        <string>PushButton</string>
+        <string>Ok</string>
        </property>
       </widget>
      </item>
      <item>
-      <widget class="QPushButton" name="pushButton_2">
+      <widget class="QPushButton" name="applyButton">
        <property name="text">
-        <string>PushButton</string>
+        <string>Apply</string>
        </property>
       </widget>
      </item>
      <item>
-      <widget class="QPushButton" name="pushButton">
+      <widget class="QPushButton" name="cancelButton">
        <property name="text">
-        <string>PushButton</string>
+        <string>Cancel</string>
        </property>
       </widget>
      </item>