Partition utility (pre-release)

contrib/README.md

@@ -3,5 +3,5 @@
 ## Utility Register
 
 * keysend/ - Experimental Keysend functionality
-
+* partitioner/ - Experimental tool for partitioning drives
 

contrib/partitioner/partitioner.py

@@ -0,0 +1,127 @@
+#!/usr/bin/env python3
+
+'''
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
+OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
+ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
+OTHER DEALINGS IN THE SOFTWARE.
+'''
+import os
+import sys
+import glob
+import re
+
+usb_dev_pattern = ['sd.*']
+usb_part_pattern = ['sd.[1-9]*']
+sd_dev_pattern = ['mmcblk*']
+sd_part_pattern = ['mmcblk.p[1-9]*']
+
+def dev_size(device):
+    path = '/sys/block/'
+    num_sectors = open(path + device + '/size').read().rstrip('\n')
+    sector_size = open(path + device + '/queue/hw_sector_size').read().rstrip('\n')
+    return (int(num_sectors)*int(sector_size))
+
+def usb_devs():
+    devices = []
+    for device in glob.glob('/sys/block/*'):
+        for pattern in usb_dev_pattern:
+            if re.compile(pattern).match(os.path.basename(device)):
+                devices.append(os.path.basename(device))
+    return devices
+
+def usb_partitions():
+    partitions = []
+    for device in usb_devs():
+        for partition in glob.glob('/sys/block/' + str(device) + '/*'):
+            for pattern in usb_part_pattern:
+                if re.compile(pattern).match(os.path.basename(partition)):
+                    partitions.append(os.path.basename(partition))
+    return partitions
+
+def usb_part_size(partition):
+    try:
+        path = '/sys/block/'
+        device = partition[:-1]
+        num_sectors = open(path + device + '/' + partition + '/size').read().rstrip('\n')
+        sector_size = open(path + device + '/queue/hw_sector_size').read().rstrip('\n')
+    except TypeError:
+        print("Not enough USB devices available")
+        sys.exit(1)
+    else:
+        return (int(num_sectors)*int(sector_size))
+
+def uuid_table():
+    device_table = os.popen('blkid').read().splitlines()
+    devices = {}
+    for device in device_table:
+        dev = device.split(":")[0].split("/")[2]
+        uuid = device.split('"')[1]
+        devices[dev] = uuid
+    return devices
+
+def get_uuid(device):
+    uuids = uuid_table()
+    return str(uuids[device])
+
+def usb_partition_table():
+    table = {}
+    for partition in usb_partitions():
+        table[partition] = int(usb_part_size(partition))
+    return table
+
+def main():
+    print('USB Configuration')
+    if len(usb_devs()) == 1:
+        first_part = dev_size('sda') / (1000*1000)
+        print('Size: ' + str(first_part))
+        prune_setting = int(first_part / 2)
+        partitions = usb_partitions()
+        if first_part < 512000:
+            print("Pruning the config")
+            os.system('/bin/sed -i "s/prune=550/prune=' + str(prune_setting) + '/g;" bitcoin/bitcoin.conf')
+        else:
+            print("Switching off pruning")
+            os.system('/bin/sed -i "s/prune=550/#prune=550/g;" bitcoin/bitcoin.conf')
+            os.system('/bin/sed -i "s/#txindex=1/txindex=1/g;" bitcoin/bitcoin.conf')
+
+        print('Creating data mount')
+        os.system('mkdir -p /mnt/data')
+        print('Initializing filesystem')
+        os.system('/sbin/mkfs.ext4 -F /dev/sda1')
+        first_partition_uuid = get_uuid('sda1')
+        os.system('/bin/mount -t ext4 /dev/sda1 /mnt/data')
+        print('Setup secrets, bitcoin, nginx, and lnd directory')
+        os.system('/bin/cp -fr /home/umbrel/secrets /mnt/data')
+        os.system('/bin/cp -fr /home/umbrel/bitcoin /mnt/data')
+        os.system('/bin/cp -fr /home/umbrel/lnd /mnt/data')
+        os.system('/bin/cp -fr /home/umbrel/nginx /mnt/data')
+        print('Setup filesystem permissions')
+        os.system('/bin/chown -R 1000.1000 /mnt/data')
+        print('Unmounting partition')
+        os.system('/bin/umount /mnt/data')
+        print('Update /etc/fstab')
+        os.system('echo "UUID=' + first_partition_uuid + ' /mnt/data ext4 defaults,noatime 0 0" >> /etc/fstab')
+        print('Remounting through /bin/mount')
+        os.system('/bin/mount -a');
+        print('Remove old folders')
+        os.system('/bin/rm -fr /home/umbrel/secrets')
+        os.system('/bin/rm -fr /home/umbrel/bitcoin')
+        os.system('/bin/rm -fr /home/umbrel/lnd')
+        os.system('/bin/rm -fr /home/umbrel/nginx')
+        print('Set up symlinks')
+        os.system('/bin/ln -s /mnt/data/secrets /home/umbrel/secrets')
+        os.system('/bin/ln -s /mnt/data/bitcoin /home/umbrel/bitcoin')
+        os.system('/bin/ln -s /mnt/data/lnd /home/umbrel/lnd')
+        os.system('/bin/ln -s /mnt/data/lnd /home/umbrel/nginx')
+    else:
+        print('No drives or unexpected number of drives detected!')
+if __name__ == '__main__':
+    if os.geteuid() == 0:
+        main()
+    else:
+        print("Must run as root")
+