From cb00347822604358579de4c1dbc359da0069e8be Mon Sep 17 00:00:00 2001
From: AuxXxilium <info@auxxxilium.tech>
Date: Sat, 27 Apr 2024 18:50:56 +0200
Subject: [PATCH] tree: update part 2

Signed-off-by: AuxXxilium <info@auxxxilium.tech>
---
 files/initrd/opt/arc/arc.sh               |   1 +
 files/initrd/opt/arc/boot.sh              |  11 +++
 files/initrd/opt/arc/include/functions.py |  93 ++++++++++++++++++++++
 files/initrd/opt/arc/include/functions.sh |  17 ++++
 files/initrd/opt/arc/include/logo.png     | Bin 0 -> 1403 bytes
 files/initrd/opt/arc/init.sh              |   1 +
 6 files changed, 123 insertions(+)
 create mode 100755 files/initrd/opt/arc/include/functions.py
 create mode 100644 files/initrd/opt/arc/include/logo.png

diff --git a/files/initrd/opt/arc/arc.sh b/files/initrd/opt/arc/arc.sh
index 9169acc6..4fa9c097 100755
--- a/files/initrd/opt/arc/arc.sh
+++ b/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
diff --git a/files/initrd/opt/arc/boot.sh b/files/initrd/opt/arc/boot.sh
index 73a22b68..4651a16a 100755
--- a/files/initrd/opt/arc/boot.sh
+++ b/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
diff --git a/files/initrd/opt/arc/include/functions.py b/files/initrd/opt/arc/include/functions.py
new file mode 100755
index 00000000..90da8ee5
--- /dev/null
+++ b/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()
diff --git a/files/initrd/opt/arc/include/functions.sh b/files/initrd/opt/arc/include/functions.sh
index 655e7a10..de92bbeb 100755
--- a/files/initrd/opt/arc/include/functions.sh
+++ b/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() {
diff --git a/files/initrd/opt/arc/include/logo.png b/files/initrd/opt/arc/include/logo.png
new file mode 100644
index 0000000000000000000000000000000000000000..33f7d2c16db34a90b233a8c8a8f6fc0e2d2aa1a4
GIT binary patch
literal 1403
zcmeAS@N?(olHy`uVBq!ia0vp^DIm<j1|$m}O$`B3oCO|{#S9E$svykh8Km-ofq_-m
z)5S5QV$R#S-W@WbA`brPX^$*A)f7EiSPtf2aQLH8w`f6khC>X8abaSB$?X?z=8H7A
zl=4Ch6m|rPdkGwLaXtFKej$gOxM#o=3lEbXsq{%Ne5#DjAAQ@j?N`{>)%TzH?O>lZ
zZ(r#8d#|m&?h8G;#fihBETUu)_k%z3;z>O8LOlnCGOsb0Gfgl5=eOkBZ_E7Epa0nz
zNX@+7e4y6Da`m4Wi<8xC70eYfKUN+oG-JNqaC%o#>Zz+t%Nk5=qn$W5uGdv;xw}wq
zUiXRoEz9<p+)`}m6-@Prb5?BWm)3Cik2CbW#<4-Tf^)}?BUcv`1|P8tzcyw1F@+C&
z1)fj-$unNR9@D0&d&B?4&I9HLbWO#dc^t93Z2Ej=nm~m@(*Ej(Z0QsKOLS*kZ|7l-
zXVfnKX36)5vq$-DlJ?fUygKaHv|j!)Tz-U)VLzj@;LFQTBV6-m*P0gbr4?VYn<MyN
z@Q19$(u7C{t)?jFPc3`Aw)K2(2yZBFw4ZR!sC(f#yH}!t6Lb%#t6t_~xX<`{S5kI^
z=m8hES(h79SKeST()hV2#j@z6&?d$mk)84^#~Spxo?LX>f9v!F^?mZkCr-S&E2Kt6
z{cKfL_Ty|573NK?HcOAxv8{8MwL(c}>2^(R?|Ub^6C*mG@<wlsioISQc+u+X)ECNg
z0=rIcaND(v?Nh5va8l~Cyo(u5d7(+x4R?LTQszZW{AI%}V=Ksd(n)wzVCUP6T?(HM
za1;b;eKY5FIFY4Vw&7)yi&22$TfLMkSI)|<;uXnq&*nexS!>7jW_J6gBNOKA;XSbO
z%lz{?Oy`*2EiKn$;q9AtY7uATtY0?1$5hlNDPJ*vV|%6Y|E9KNUnc)61#_Y<wrrWW
zT0c?3LVD&M(b#qKna{C^P1lMwyZplWhEh-Bn*R+u6z>QvIXP?L4+$HEqSsPdYCq){
z&zfM}z`sg-d0mqKm84huPo19dMKX1IVjR1ivzU+Xr57s88K;%po9~(x=OD(Rf96I#
zTZL5Qls^KJp)=DKa(asW+E#j$t#Ny!^pwAo-*4ZWQWqs0d!4~m)y;!ppPQR^>aPh;
zPw1Yvy4J8I@tsbA@&}uO@T}Vm1@;>_cko%n7})2|{6D|xxQ0v5gu@3yML*A6GVkek
zn{7<{8TvJD>#=@Q>R!ywB~!d!=yprZbgke<pQM&Eis4?no#cAMKI_S`{$rXKxNWW5
zwa3RA&MWnAm*1(dJN-v?qkO~riBi$ap7AT3Hk*A}I;myLgxyK0tvr`MOe*bqdtz<V
z+%IO9=?wFgd~bJ8Jbs(?&b+R*PVYBs6>qYe{#C4qTXA(|6=S4d=fk{NJKqZYo#4D{
zZM@r7?LDsVb1!@`+R;{Kyz0c>FZmOfdwomKsTZo4d2i_=li(B1e2-Z_^(p>RY%hG$
z!ulk|FX*k}R)#-a9-e)YpP%w6zAc}`_TYX~r9Z=S=6t5>B~#g#$nNKT$-ZsU{{@Hb
z9xy#nd?L7P^MuZ#o1Z7yXk6c|_~Spn;n9<~+I1pgIt=N6{xN^~ui|pGYWaL%Y0BW~
L>gTe~DWM4fe56;u

literal 0
HcmV?d00001

diff --git a/files/initrd/opt/arc/init.sh b/files/initrd/opt/arc/init.sh
index 73ac919d..d1a413d9 100755
--- a/files/initrd/opt/arc/init.sh
+++ b/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}"