|
| 1 | +name: 'Download and configure Kernel Source code' |
| 2 | + |
| 3 | +inputs: |
| 4 | + source_location: |
| 5 | + description: 'Folder path to save Kernel source' |
| 6 | + required: true |
| 7 | + type: string |
| 8 | + github_token: |
| 9 | + description: 'GitHub Token' |
| 10 | + required: true |
| 11 | + debug: |
| 12 | + description: 'Enable Logs' |
| 13 | + required: false |
| 14 | + type: boolean |
| 15 | + default: false |
| 16 | + |
| 17 | +runs: |
| 18 | + using: 'composite' |
| 19 | + steps: |
| 20 | + - name: Download and Prepare Manifest |
| 21 | + shell: bash |
| 22 | + working-directory: ${{ inputs.source_location }} |
| 23 | + run: | |
| 24 | + # Download and Prepare Manifest |
| 25 | + if [[ "$OP_MANIFEST" == https://* ]]; then |
| 26 | + curl --fail --show-error --location --proto '=https' "$OP_MANIFEST" -o manifest.xml |
| 27 | + elif [[ "$OP_BRANCH" == wild/* ]]; then |
| 28 | + cp "../manifests/$(echo "$OP_OS_VERSION" | tr '[:upper:]' '[:lower:]')/$OP_MANIFEST" manifest.xml |
| 29 | + else |
| 30 | + curl --fail --show-error --location --proto '=https' "https://raw.githubusercontent.com/OnePlusOSS/kernel_manifest/refs/heads/$OP_BRANCH/$OP_MANIFEST" -o manifest.xml |
| 31 | + fi |
| 32 | +
|
| 33 | + - name: Download Manifest Archives |
| 34 | + shell: python |
| 35 | + env: |
| 36 | + PYTHONUNBUFFERED: "1" |
| 37 | + GITHUB_TOKEN: ${{ inputs.github_token }} |
| 38 | + DEBUG: ${{ inputs.debug }} |
| 39 | + working-directory: ${{ inputs.source_location }} |
| 40 | + run: | |
| 41 | + # Download Manifest Archives |
| 42 | + import xml.etree.ElementTree as ET |
| 43 | + import subprocess |
| 44 | + import os, shutil |
| 45 | + import time |
| 46 | + import glob |
| 47 | + from concurrent.futures import ThreadPoolExecutor |
| 48 | + import requests |
| 49 | + |
| 50 | + MAX_WORKERS = (os.cpu_count() or 2) * 4 |
| 51 | + TARGET_REPO = "${{ github.repository }}" |
| 52 | + TOOLCHAIN_KEYWORDS = ["clang", "rust"] |
| 53 | + DEBUG = os.environ.get("DEBUG", "false").lower() == "true" |
| 54 | + |
| 55 | + print("::group::Download Manifest Archives") |
| 56 | + |
| 57 | + def get_release_parts(label, rev): |
| 58 | + url = f"https://api.github.com/repos/{TARGET_REPO}/releases/tags/toolchain-cache" |
| 59 | + headers = {"Authorization": f"token {os.environ['GITHUB_TOKEN']}"} |
| 60 | + response = requests.get(url, headers=headers).json() |
| 61 | + |
| 62 | + prefix = f"{label}-{rev}.tar.gz" |
| 63 | + return [a['name'] for a in response.get('assets', []) if a['name'].startswith(prefix)] |
| 64 | + |
| 65 | + def sync_project(task): |
| 66 | + name, path, url, strip, rev = task |
| 67 | + if path not in ["./", "."]: |
| 68 | + os.makedirs(path, exist_ok=True) |
| 69 | + |
| 70 | + headers = ( |
| 71 | + "-H 'User-Agent: Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36' " |
| 72 | + "-H 'Accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8' " |
| 73 | + "-H 'Accept-Encoding: gzip, deflate, br' " |
| 74 | + "-H 'Connection: keep-alive' " |
| 75 | + "--tcp-fastopen" |
| 76 | + ) |
| 77 | + |
| 78 | + print(f"Syncing: {name} -> {path}") |
| 79 | + start_time = time.time() |
| 80 | + print(f" [PENDING] {name}") |
| 81 | + try: |
| 82 | + match = next((x for x in TOOLCHAIN_KEYWORDS if x in name.lower()), None) |
| 83 | + if match: |
| 84 | + label = match |
| 85 | + base_filename = f"{label}-{rev}.tar.gz" |
| 86 | + |
| 87 | + asset_names = get_release_parts(label, rev) |
| 88 | + |
| 89 | + if not asset_names: |
| 90 | + print(f" [ERROR] No assets found for {base_filename} in release!") |
| 91 | + return False |
| 92 | + |
| 93 | + if DEBUG: |
| 94 | + print(f" [CACHE] Fetching {label} toolchain: {base_filename}...") |
| 95 | + |
| 96 | + for asset in asset_names: |
| 97 | + download_url = f"https://github.com/{TARGET_REPO}/releases/download/toolchain-cache/{asset}" |
| 98 | + aria_cmd = ( |
| 99 | + f"aria2c -x8 -s8 -k1M --file-allocation=none " |
| 100 | + f"--header='Authorization: token {os.environ['GITHUB_TOKEN']}' " |
| 101 | + f"-o {asset} {download_url}" |
| 102 | + ) |
| 103 | + subprocess.run(aria_cmd, shell=True, check=True, capture_output=True) |
| 104 | + |
| 105 | + parts = sorted(glob.glob(f"{base_filename}.part*")) |
| 106 | + if parts: |
| 107 | + if DEBUG: |
| 108 | + print(f" [MERGE] Combining {len(parts)} parts for {rev}...") |
| 109 | + with open(base_filename, 'wb') as outfile: |
| 110 | + for part in parts: |
| 111 | + with open(part, 'rb') as infile: |
| 112 | + shutil.copyfileobj(infile, outfile) |
| 113 | + os.remove(part) |
| 114 | + |
| 115 | + if not os.path.exists(base_filename): |
| 116 | + print(f" [ERROR] {base_filename} was not found after download/merge!") |
| 117 | + return False |
| 118 | + |
| 119 | + subprocess.run(f"tar -I pigz -x -f {base_filename} -C {path} {strip}", shell=True, check=True) |
| 120 | + if os.path.exists(base_filename): |
| 121 | + os.remove(base_filename) |
| 122 | + else: |
| 123 | + cmd = f"curl -LfsS {headers} --retry 5 --connect-timeout 30 '{url}' | tar -I pigz -x -C {path} {strip}" |
| 124 | + subprocess.run(cmd, shell=True, check=True) |
| 125 | + |
| 126 | + duration = time.time() - start_time |
| 127 | + print(f"Synced {name} successfully! ({duration:.2f}s)") |
| 128 | + return True |
| 129 | + except subprocess.CalledProcessError as e: |
| 130 | + print(f" [ERROR] Command failed for {name}: {e.cmd}") |
| 131 | + print(f" Stderr: {e.stderr.decode() if e.stderr else 'No stderr'}") |
| 132 | + return False |
| 133 | + except Exception as e: |
| 134 | + print(f" [ERROR] Failed to sync {name}") |
| 135 | + return False |
| 136 | + |
| 137 | + global_start = time.time() |
| 138 | + |
| 139 | + with open('manifest.xml', 'r') as f: |
| 140 | + manifest_content = f.read() |
| 141 | + |
| 142 | + root = ET.fromstring(manifest_content) |
| 143 | + top_dir = os.getcwd() |
| 144 | + |
| 145 | + remotes = {r.get('name'): r.get('fetch').rstrip('/') for r in root.findall('remote')} |
| 146 | + default = root.find('default') |
| 147 | + def_remote = default.get('remote') if default is not None else None |
| 148 | + def_rev = default.get('revision') if default is not None else None |
| 149 | + |
| 150 | + sync_tasks = [] |
| 151 | + post_process_data = [] |
| 152 | + |
| 153 | + for project in root.findall('project'): |
| 154 | + name = project.get('name') |
| 155 | + path = project.get('path', name) |
| 156 | + remote_name = project.get('remote', def_remote) |
| 157 | + rev = project.get('revision', def_rev) |
| 158 | + base_url = remotes.get(remote_name) |
| 159 | + |
| 160 | + if not base_url: continue |
| 161 | + |
| 162 | + if "github.com" in base_url: |
| 163 | + url = f"{base_url}/{name}/archive/{rev}.tar.gz" |
| 164 | + strip = "--strip-components=1" |
| 165 | + elif "googlesource.com" in base_url: |
| 166 | + url = f"{base_url}/{name}/+archive/{rev}.tar.gz" |
| 167 | + strip = "" |
| 168 | + elif "git.codelinaro.org" in base_url: |
| 169 | + url = f"{base_url}/{name}/-/archive/{rev}.tar.gz" |
| 170 | + strip = "--strip-components=1" |
| 171 | + else: |
| 172 | + continue |
| 173 | + |
| 174 | + sync_tasks.append((name, path, url, strip, rev)) |
| 175 | + |
| 176 | + for child in project: |
| 177 | + if child.tag in ['linkfile', 'copyfile']: |
| 178 | + post_process_data.append((path, child)) |
| 179 | + |
| 180 | + if DEBUG: |
| 181 | + print(f"Starting parallel sync of {len(sync_tasks)} projects...") |
| 182 | + with ThreadPoolExecutor(max_workers=MAX_WORKERS) as executor: |
| 183 | + success_list = list(executor.map(sync_project, sync_tasks)) |
| 184 | + |
| 185 | + if not all(success_list): |
| 186 | + print("::error::One or more projects failed to sync!") |
| 187 | + print("::endgroup::") |
| 188 | + exit(1) |
| 189 | + |
| 190 | + print("Processing linkfiles and copyfiles...") |
| 191 | + for path, child in post_process_data: |
| 192 | + src_rel = child.get('src') |
| 193 | + dest_rel = child.get('dest') |
| 194 | + if not src_rel or not dest_rel: continue |
| 195 | + |
| 196 | + src_path = os.path.join(top_dir, path, src_rel) |
| 197 | + dest_path = os.path.join(top_dir, dest_rel) |
| 198 | + os.makedirs(os.path.dirname(dest_path), exist_ok=True) |
| 199 | + |
| 200 | + if child.tag == 'linkfile': |
| 201 | + if os.path.lexists(dest_path): os.remove(dest_path) |
| 202 | + rel_target = os.path.relpath(src_path, os.path.dirname(dest_path)) |
| 203 | + os.symlink(rel_target, dest_path) |
| 204 | + print(f" [Link] {dest_rel} -> {src_rel}") |
| 205 | + elif child.tag == 'copyfile': |
| 206 | + shutil.copy2(src_path, dest_path) |
| 207 | + print(f" [Copy] {dest_rel} from {src_rel}") |
| 208 | + |
| 209 | + |
| 210 | + total_duration = time.time() - global_start |
| 211 | + minutes = int(total_duration // 60) |
| 212 | + seconds = total_duration % 60 |
| 213 | + print(f"Kernel Sync completed in {minutes}m {seconds:.2f}s") |
| 214 | + print("::endgroup::") |
0 commit comments