From 7c0a3774e63766c9028780bb7ee8b3bdd6499984 Mon Sep 17 00:00:00 2001
From: Luke Childs <lukechilds123@gmail.com>
Date: Tue, 19 Mar 2024 03:34:26 +0700
Subject: [PATCH] Set ondemand cpu governer for Raspberry Pi

---
 packages/umbreld/source/index.ts          | 18 ++++++++++++++++++
 packages/umbreld/source/modules/system.ts |  4 ++++
 2 files changed, 22 insertions(+)

diff --git a/packages/umbreld/source/index.ts b/packages/umbreld/source/index.ts
index b3b6edc18..962ffd921 100644
--- a/packages/umbreld/source/index.ts
+++ b/packages/umbreld/source/index.ts
@@ -12,6 +12,7 @@ import Server from './modules/server/index.js'
 import User from './modules/user.js'
 import AppStore from './modules/apps/app-store.js'
 import Apps from './modules/apps/apps.js'
+import {detectDevice, setCpuGovernor} from './modules/system.js'
 
 import {commitOsPartition} from './modules/system.js'
 
@@ -69,6 +70,20 @@ export default class Umbreld {
 		this.apps = new Apps(this)
 	}
 
+	async setupPiCpuGoverner() {
+		// TODO: Move this to a system module
+		// Set ondemand cpu governer for Raspberry Pi
+		try {
+			const {productName} = await detectDevice()
+			if (productName === 'Raspberry Pi') {
+				await setCpuGovernor('ondemand')
+				this.logger.log(`Set ondemand cpu governor`)
+			}
+		} catch (error) {
+			this.logger.error(`Failed to set ondemand cpu governor: ${(error as Error).message}`)
+		}
+	}
+
 	async start() {
 		this.logger.log(`☂️  Starting Umbrel v${this.version}`)
 		this.logger.log()
@@ -80,6 +95,9 @@ export default class Umbreld {
 		// If we've successfully booted then commit to the current OS partition
 		commitOsPartition(this)
 
+		// Set ondemand cpu governer for Raspberry Pi
+		this.setupPiCpuGoverner()
+
 		// Run migration module before anything else
 		// TODO: think through if we want to allow the server module to run before migration.
 		// It might be useful if we add more complicated migrations so we can signal progress.
diff --git a/packages/umbreld/source/modules/system.ts b/packages/umbreld/source/modules/system.ts
index 0e71da92d..a71690011 100644
--- a/packages/umbreld/source/modules/system.ts
+++ b/packages/umbreld/source/modules/system.ts
@@ -214,3 +214,7 @@ export async function detectDevice() {
 export async function isUmbrelOS() {
 	return fse.exists('/umbrelOS')
 }
+
+export async function setCpuGovernor(governor: string) {
+	await fse.writeFile('/sys/devices/system/cpu/cpu0/cpufreq/scaling_governor', governor)
+}