tree: update part 2

Signed-off-by: AuxXxilium <info@auxxxilium.tech>

files/initrd/opt/arc/arc.sh

@@ -221,6 +221,7 @@ function arcModel() {
     writeConfigKey "synoinfo" "{}" "${USER_CONFIG_FILE}"
     writeConfigKey "modules" "{}" "${USER_CONFIG_FILE}"
     writeConfigKey "addons" "{}" "${USER_CONFIG_FILE}"
+    getLogo "${MODEL}"
     CONFDONE="$(readConfigKey "arc.confdone" "${USER_CONFIG_FILE}")"
     BUILDDONE="$(readConfigKey "arc.builddone" "${USER_CONFIG_FILE}")"
     if [[ -f "${ORI_ZIMAGE_FILE}" || -f "${ORI_RDGZ_FILE}" || -f "${MOD_ZIMAGE_FILE}" || -f "${MOD_RDGZ_FILE}" ]]; then

files/initrd/opt/arc/boot.sh

@@ -235,6 +235,17 @@ elif [ "${DIRECTBOOT}" = "false" ]; then
 
   echo -e "\033[1;37mLoading DSM kernel...\033[0m"
 
+  DSMLOGO="$(readConfigKey "arc.dsmlogo" "${USER_CONFIG_FILE}")"
+  if [ "${DSMLOGO}" = "true" -a -c "/dev/fb0" ]; then
+    [[ "${IPCON}" =~ ^169\.254\..* ]] && IPCON=""
+    [ -n "${IPCON}" ] && URL="http://${IPCON}:5000" || URL="http://find.synology.com/"
+    python ${ARC_PATH}/include/functions.py makeqr -d "${URL}" -l "6" -o "${TMP_PATH}/qrcode_boot.png"
+    [ -f "${TMP_PATH}/qrcode_boot.png" ] && echo | fbv -acufi "${TMP_PATH}/qrcode_boot.png" >/dev/null 2>/dev/null || true
+
+    python ${ARC_PATH}/include/functions.py makeqr -f "${ARC_PATH}/include/qhxg.png" -l "7" -o "${TMP_PATH}/qrcode_qhxg.png"
+    [ -f "${TMP_PATH}/qrcode_qhxg.png" ] && echo | fbv -acufi "${TMP_PATH}/qrcode_qhxg.png" >/dev/null 2>/dev/null || true
+  fi
+
   # Executes DSM kernel via KEXEC
   KEXECARGS=""
   kexec ${KEXECARGS} -l "${MOD_ZIMAGE_FILE}" --initrd "${MOD_RDGZ_FILE}" --command-line="${CMDLINE_LINE}" >"${LOG_FILE}" 2>&1 || dieLog

files/initrd/opt/arc/include/functions.py

@@ -0,0 +1,93 @@
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2022 Ing <https://github.com/wjz304>
+#
+# This is free software, licensed under the MIT License.
+# See /LICENSE for more information.
+#
+
+import os, click
+
+WORK_PATH = os.path.abspath(os.path.dirname(__file__))
+
+@click.group()
+def cli():
+    """
+    The CLI is a commands to Arc.
+    """
+    pass
+
+def mutually_exclusive_options(ctx, param, value):
+    other_option = 'file' if param.name == 'data' else 'data'
+    if value is not None and ctx.params.get(other_option) is not None:
+        raise click.UsageError(f'Illegal usage: `{param.name}` is mutually exclusive with `{other_option}`.')
+    return value
+
+def validate_required_param(ctx, param, value):
+    if not value and 'file' not in ctx.params and 'data' not in ctx.params:
+        raise click.MissingParameter(param_decls=[param.name])
+    return value
+
+@cli.command()
+@click.option('-d', "--data", type=str, callback=mutually_exclusive_options, is_eager=True, help="The data of QRCode.")
+@click.option('-f', "--file", type=str, callback=mutually_exclusive_options, is_eager=True, help="The file of QRCode.")
+@click.option('--validate', is_flag=True, callback=validate_required_param, expose_value=False, is_eager=True)
+@click.option('-l', "--location", type=click.IntRange(0, 7), required=True, help="The location of QRCode. (range 0<=x<=7)")
+@click.option('-o', "--output", type=str, required=True, help="The output file of QRCode.")
+def makeqr(data, file, location, output):
+    """
+    Generate a QRCode.
+    """
+    import fcntl, struct
+    import qrcode
+    from PIL import Image
+
+    FBIOGET_VSCREENINFO = 0x4600
+    FBIOPUT_VSCREENINFO = 0x4601
+    FBIOGET_FSCREENINFO = 0x4602
+    FBDEV = "/dev/fb0"
+    if data is not None:
+        qr = qrcode.QRCode(version=1, box_size=10, error_correction=qrcode.constants.ERROR_CORRECT_H, border=4)
+        qr.add_data(data)
+        qr.make(fit=True)
+        img = qr.make_image(fill_color="purple", back_color="white")
+        img = img.convert("RGBA")
+        pixels = img.load()
+        for i in range(img.size[0]):
+            for j in range(img.size[1]):
+                if pixels[i, j] == (255, 255, 255, 255):
+                    pixels[i, j] = (255, 255, 255, 0)
+        
+        if os.path.exists(os.path.join(WORK_PATH, "logo.png")):
+            icon = Image.open(os.path.join(WORK_PATH, "logo.png"))
+            icon = icon.convert("RGBA")
+            img.paste(icon.resize((int(img.size[0] / 5), int(img.size[1] / 5))), (int((img.size[0] - int(img.size[0] / 5)) / 2), int((img.size[1] - int(img.size[1] / 5)) / 2)))
+    
+    if file is not None:
+        img = Image.open(file)
+        # img = img.convert("RGBA")
+        # pixels = img.load()
+        # for i in range(img.size[0]):
+        #     for j in range(img.size[1]):
+        #         if pixels[i, j] == (255, 255, 255, 255):
+        #             pixels[i, j] = (255, 255, 255, 0)
+
+    (xres, yres) = (1920, 1080)
+    with open(FBDEV, 'rb')as fb:
+        vi = fcntl.ioctl(fb, FBIOGET_VSCREENINFO, bytes(160))
+        res = struct.unpack('I'*40, vi)
+        if res[0] != 0 and res[1] != 0:
+            (xres, yres) = (res[0], res[1]) 
+    xqr, yqr = (int(xres / 8), int(xres / 8))
+    img = img.resize((xqr, yqr))
+
+    alpha = Image.new("RGBA", (xres, yres), (0, 0, 0, 0))
+    if int(location) not in range(0, 8):
+      location = 0
+    loc = (img.size[0] * int(location), alpha.size[1] - img.size[1])
+    alpha.paste(img, loc)
+    alpha.save(output)
+
+
+if __name__ == "__main__":
+    cli()

files/initrd/opt/arc/include/functions.sh

@@ -285,6 +285,23 @@ function getIP() {
   return 0
 }
 
+###############################################################################
+# get logo of model
+# 1 - model
+function getLogo() {
+  MODEL="${1}"
+  rm -f "${PART3_PATH}/logo.png"
+  STATUS=$(curl -skL -m 10 -w "%{http_code}" "https://www.synology.com/api/products/getPhoto?product=${MODEL/+/%2B}&type=img_s&sort=0" -o "${PART3_PATH}/logo.png")
+  if [ $? -ne 0 -o ${STATUS:-0} -ne 200 -o ! -f "${PART3_PATH}/logo.png" ]; then
+    rm -f "${PART3_PATH}/logo.png"
+    return 1
+  fi
+  convert -rotate 180 "${PART3_PATH}/logo.png" "${PART3_PATH}/logo.png" 2>/dev/null
+  magick montage "${PART3_PATH}/logo.png" -background 'none' -tile '3x3' -geometry '350x210' "${PART3_PATH}/logo.png" 2>/dev/null
+  convert -rotate 180 "${PART3_PATH}/logo.png" "${PART3_PATH}/logo.png" 2>/dev/null
+  return 0
+}
+
 ###############################################################################
 # Find and mount the DSM root filesystem
 function findAndMountDSMRoot() {

files/initrd/opt/arc/init.sh

@@ -51,6 +51,7 @@ initConfigKey "arc.paturl" "" "${USER_CONFIG_FILE}"
 initConfigKey "arc.pathash" "" "${USER_CONFIG_FILE}"
 initConfigKey "arc.sn" "" "${USER_CONFIG_FILE}"
 initConfigKey "arc.ipv6" "false" "${USER_CONFIG_FILE}"
+initConfigKey "arc.dsmlogo" "true" "${USER_CONFIG_FILE}"
 initConfigKey "arc.emmcboot" "false" "${USER_CONFIG_FILE}"
 initConfigKey "arc.offline" "false" "${USER_CONFIG_FILE}"
 initConfigKey "arc.directboot" "false" "${USER_CONFIG_FILE}"