updated version

argonfanhat.sh

@@ -8,7 +8,6 @@ argon_create_file() {
 	sudo touch $1
 	sudo chmod 666 $1
 }
-
 argon_check_pkg() {
 	RESULT=$(dpkg-query -W -f='${Status}\n' "$1" 2> /dev/null | grep "installed")
 
@@ -125,7 +124,6 @@ echo '	bus = smbus.SMBus(0)' >> $shutdownscript
 
 echo 'if len(sys.argv)>1:' >> $shutdownscript
 echo '	address=0x1a' >> $shutdownscript
-
 # Fan off
 echo "	bus.write_byte(address,0)"  >> $shutdownscript
 
@@ -148,7 +146,23 @@ echo '	bus = smbus.SMBus(1)' >> $powerbuttonscript
 echo 'else:' >> $powerbuttonscript
 echo '	bus = smbus.SMBus(0)' >> $powerbuttonscript
 
+echo 'GPIO.setwarnings(False)' >> $powerbuttonscript
+echo 'GPIO.setmode(GPIO.BCM)' >> $powerbuttonscript
+echo 'shutdown_pin=4' >> $powerbuttonscript
+echo 'GPIO.setup(shutdown_pin, GPIO.IN,  pull_up_down=GPIO.PUD_DOWN)' >> $powerbuttonscript
 
+echo 'def shutdown_check():' >> $powerbuttonscript
+echo '	while True:' >> $powerbuttonscript
+echo '		pulsetime = 1' >> $powerbuttonscript
+echo '		GPIO.wait_for_edge(shutdown_pin, GPIO.RISING)' >> $powerbuttonscript
+echo '		time.sleep(0.01)' >> $powerbuttonscript
+echo '		while GPIO.input(shutdown_pin) == GPIO.HIGH:' >> $powerbuttonscript
+echo '			time.sleep(0.01)' >> $powerbuttonscript
+echo '			pulsetime += 1' >> $powerbuttonscript
+echo '		if pulsetime >=2 and pulsetime <=3:' >> $powerbuttonscript
+echo '			os.system("reboot")' >> $powerbuttonscript
+echo '		elif pulsetime >=4 and pulsetime <=5:' >> $powerbuttonscript
+echo '			os.system("shutdown now -h")' >> $powerbuttonscript
 
 echo 'def get_fanspeed(tempval, configlist):' >> $powerbuttonscript
 echo '	for curconfig in configlist:' >> $powerbuttonscript
@@ -231,10 +245,14 @@ echo '			temp=""' >> $powerbuttonscript
 echo '		time.sleep(30)' >> $powerbuttonscript
 
 echo 'try:' >> $powerbuttonscript
+echo '	t1 = Thread(target = shutdown_check)' >> $powerbuttonscript
 echo '	t2 = Thread(target = temp_check)' >> $powerbuttonscript
+echo '	t1.start()' >> $powerbuttonscript
 echo '	t2.start()' >> $powerbuttonscript
 echo 'except:' >> $powerbuttonscript
+echo '	t1.stop()' >> $powerbuttonscript
 echo '	t2.stop()' >> $powerbuttonscript
+echo '	GPIO.cleanup()' >> $powerbuttonscript
 
 sudo chmod 755 $powerbuttonscript