feat: add parse_arch func, testing passed

Author: initdc

lib/rootfs/distro/ubuntu/arch.rb

@@ -3,6 +3,8 @@
 module RootFS
   module Distro
     module Ubuntu
+      extend self
+
       ARCH = %w[
         amd64
         arm64
@@ -11,6 +13,20 @@ module Ubuntu
         riscv64
         s390x
       ].freeze
+
+      def parse_arch(any)
+        err_msg = "Valid Ubuntu arch: #{ARCH}"
+
+        puts err_msg unless any
+
+        str = any.to_s
+        puts err_msg if str.empty?
+
+        ARCH.each do |arch|
+          return { arch: arch } if str.include?(arch)
+        end
+        puts err_msg
+      end
     end
   end
 end

lib/rootfs/distro/ubuntu/edition.rb

@@ -50,7 +50,7 @@ def parse_edition(any)
         puts err_msg if str.empty?
 
         EDITION.each do |edition|
-          return { edition: edition } if any.include?(edition)
+          return { edition: edition } if str.include?(edition)
         end
         puts err_msg
       end

lib/rootfs/distro/ubuntu/file.rb

@@ -2,6 +2,7 @@
 
 require_relative "../../http"
 require_relative "../../data"
+require_relative "arch"
 
 module RootFS
   module Distro
@@ -46,6 +47,11 @@ def files_of(*argv, daily: false, **opts)
         arch = argv[2] || opts[:arch] || "amd64"
         devkit = argv[3] || opts[:devkit]
 
+        hash = RootFS::Distro::Ubuntu.parse_arch(arch)
+        return unless hash
+
+        arch = hash[:arch]
+
         keywords = [edition, arch]
         ignores = []
 

lib/rootfs/distro/ubuntu/file_test.rb

@@ -15,6 +15,8 @@ class MyClass
 p u.files_of("server", 20.04)
 p u.files_of("base", 20.04)
 p u.files_of("base", "focal")
+p u.files_of("base", 20.04, "riscv64")
+p u.files_of("base", "focal", "loongarch")
 p u.files_of("cloud", "focal")
 p u.files_of("cloud", 22.04)
 p u.files_of("minimal", 22.04)