From 07ab25f9b19fd238a68f932e8f0e2d143ba30a21 Mon Sep 17 00:00:00 2001
From: Serhiy Mytrovtsiy <mitrovtsiy@ukr.net>
Date: Mon, 8 Nov 2021 18:07:08 +0100
Subject: [PATCH] feat: added Pie Chart widget to the Disk module (#682)

---
 Kit/Widgets/PieChart.swift |  4 ++++
 Modules/Disk/config.plist  | 11 +++++++++--
 Modules/Disk/main.swift    |  4 ++++
 3 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/Kit/Widgets/PieChart.swift b/Kit/Widgets/PieChart.swift
index 73620988..3dcff0dd 100644
--- a/Kit/Widgets/PieChart.swift
+++ b/Kit/Widgets/PieChart.swift
@@ -56,6 +56,10 @@ public class PieChart: WidgetWrapper {
                     circle_segment(value: 0.12, color: NSColor.systemOrange),
                     circle_segment(value: 0.08, color: NSColor.systemPink)
                 ])
+            } else if self.title == "Disk" {
+                self.chart.setSegments([
+                    circle_segment(value: 0.86, color: NSColor.systemBlue)
+                ])
             }
         } else {
             self.labelState = Store.shared.bool(key: "\(self.title)_\(self.type.rawValue)_label", defaultValue: self.labelState)
diff --git a/Modules/Disk/config.plist b/Modules/Disk/config.plist
index e47563d9..ff5e84a6 100644
--- a/Modules/Disk/config.plist
+++ b/Modules/Disk/config.plist
@@ -65,6 +65,13 @@
 			<key>Order</key>
 			<integer>2</integer>
 		</dict>
+		<key>pie_chart</key>
+		<dict>
+			<key>Default</key>
+			<false/>
+			<key>Order</key>
+			<integer>3</integer>
+		</dict>
 		<key>memory</key>
 		<dict>
 			<key>Default</key>
@@ -75,7 +82,7 @@
 				<string>47.85 GB, 184.84 GB</string>
 			</dict>
 			<key>Order</key>
-			<integer>3</integer>
+			<integer>4</integer>
 		</dict>
 		<key>speed</key>
 		<dict>
@@ -89,7 +96,7 @@
 				<string>R</string>
 			</array>
 			<key>Order</key>
-			<integer>4</integer>
+			<integer>5</integer>
 		</dict>
 	</dict>
 </dict>
diff --git a/Modules/Disk/main.swift b/Modules/Disk/main.swift
index c9c2283b..678f9d26 100644
--- a/Modules/Disk/main.swift
+++ b/Modules/Disk/main.swift
@@ -206,6 +206,10 @@ public class Disk: Module {
             case let widget as Mini: widget.setValue(percentage)
             case let widget as BarChart: widget.setValue([[ColorValue(percentage)]])
             case let widget as MemoryWidget: widget.setValue((DiskSize(free).getReadableMemory(), DiskSize(usedSpace).getReadableMemory()))
+            case let widget as PieChart:
+                widget.setValue([
+                    circle_segment(value: percentage, color: NSColor.systemBlue)
+                ])
             default: break
             }
         }