워크플로우 업데이트

scripts/docker-image/download_works_drive.sh

@@ -59,7 +59,11 @@ list_children() {
   local response_body
   local http_status
 
-  endpoint="$(resolve_files_endpoint "$parent_file_id")/children"
+  if [[ -n "$parent_file_id" ]]; then
+    endpoint="$(resolve_files_endpoint "$parent_file_id")/children"
+  else
+    endpoint="$(resolve_files_endpoint)"
+  fi
   response="$("$curl_bin" -sS -w $'\n%{http_code}' \
     -H "Authorization: Bearer $access_token" \
     "$endpoint")"

scripts/docker-image/upload_works_drive.sh

@@ -68,6 +68,7 @@ image_ref="${DOCKER_IMAGE_REF:-${IMAGE_REF:-}}"
 
 commit_container="${WORKS_DOCKER_COMMIT_CONTAINER:-${DOCKER_COMMIT_CONTAINER:-}}"
 archive_root="${WORKS_DOCKER_IMAGE_ARCHIVE_DIR:-/tmp/baron-sso-docker-image-upload}"
+folder_cache_file="${WORKS_DOCKER_IMAGE_FOLDER_CACHE_FILE:-${archive_root}/.works-folder-cache.json}"
 image_root_dir="${WORKS_DRIVE_DOCKER_IMAGE_DIR:-${WORKS_SHAREDRIVE_DOCKER_IMAGE_DIR:-baron-sso}}"
 dry_run="${WORKS_DRIVE_DRY_RUN:-false}"
 target="${WORKS_DRIVE_TARGET:-sharedrive}"
@@ -78,6 +79,34 @@ upload_scope="${WORKS_DRIVE_OAUTH_SCOPE:-file}"
 
 WORKS_DRIVE_SHARED_DRIVE_ID="${WORKS_DRIVE_SHARED_DRIVE_ID:-${WORKS_DRIVE_SHAREDRIVE_ID:-${WORKS_SHAREDRIVE_ID:-}}}"
 
+folder_cache_scope() {
+  printf '%s:%s:%s' "$target" "${WORKS_DRIVE_SHARED_DRIVE_ID:-${WORKS_DRIVE_USER_ID:-${WORKS_DRIVE_GROUP_ID:-${WORKS_DRIVE_SHARED_FOLDER_ID:-}}}}" "${WORKS_DRIVE_PARENT_FILE_ID:-root}"
+}
+
+read_cached_folder_id() {
+  local path="$1"
+  local key
+
+  [[ -f "$folder_cache_file" ]] || return 0
+  key="$(folder_cache_scope):${path}"
+  jq -er --arg key "$key" '.[$key] // empty' "$folder_cache_file" 2>/dev/null || true
+}
+
+write_cached_folder_id() {
+  local path="$1"
+  local folder_id="$2"
+  local key
+  local tmp_file
+
+  [[ -n "$folder_id" ]] || return 0
+  mkdir -p "$(dirname "$folder_cache_file")"
+  [[ -f "$folder_cache_file" ]] || printf '{}\n' >"$folder_cache_file"
+  key="$(folder_cache_scope):${path}"
+  tmp_file="${folder_cache_file}.tmp"
+  jq --arg key "$key" --arg folderId "$folder_id" '. + {($key): $folderId}' "$folder_cache_file" >"$tmp_file"
+  mv "$tmp_file" "$folder_cache_file"
+}
+
 urlencode_path() {
   jq -nr --arg value "$1" '$value|@uri'
 }
@@ -163,7 +192,11 @@ resolve_target_upload_endpoint() {
 
 resolve_target_children_endpoint() {
   local parent_file_id="${1:-${WORKS_DRIVE_PARENT_FILE_ID:-}}"
-  printf '%s/children\n' "$(resolve_target_upload_endpoint "$parent_file_id")"
+  if [[ -n "$parent_file_id" ]]; then
+    printf '%s/children\n' "$(resolve_target_upload_endpoint "$parent_file_id")"
+  else
+    resolve_target_upload_endpoint
+  fi
 }
 
 resolve_target_create_folder_endpoint() {
@@ -365,24 +398,33 @@ ensure_child_folder() {
   local children_json
   local folder_id
 
-  children_endpoint="$(resolve_target_children_endpoint "$parent_file_id")"
-  create_folder_endpoint="$(resolve_target_create_folder_endpoint "$parent_file_id")"
-  children_json="$(list_child_folders "$access_token" "$children_endpoint")"
-  folder_id="$(jq -er --arg name "$folder_name" '
-    [
-      (.files // .children // .items // [])[]
-      | select((.fileName // .name) == $name)
-      | select(((.fileType // .type // "") | ascii_downcase) == "folder")
-      | .fileId // .id
-    ][0] // empty
-  ' <<<"$children_json" 2>/dev/null || true)"
+  if [[ -n "$parent_file_id" ]]; then
+    children_endpoint="$(resolve_target_children_endpoint "$parent_file_id")"
+    create_folder_endpoint="$(resolve_target_create_folder_endpoint "$parent_file_id")"
+    if ! children_json="$(list_child_folders "$access_token" "$children_endpoint")"; then
+      return 1
+    fi
+    folder_id="$(jq -er --arg name "$folder_name" '
+      [
+        (.files // .children // .items // [])[]
+        | select((.fileName // .name) == $name)
+        | select(((.fileType // .type // "") | ascii_downcase) == "folder")
+        | .fileId // .id
+      ][0] // empty
+    ' <<<"$children_json" 2>/dev/null || true)"
 
-  if [[ -n "$folder_id" ]]; then
-    printf '%s\n' "$folder_id"
-    return
+    if [[ -n "$folder_id" ]]; then
+      printf '%s\n' "$folder_id"
+      return
+    fi
+  else
+    create_folder_endpoint="$(resolve_target_create_folder_endpoint "$parent_file_id")"
   fi
 
-  create_child_folder "$access_token" "$create_folder_endpoint" "$folder_name"
+  if ! folder_id="$(create_child_folder "$access_token" "$create_folder_endpoint" "$folder_name")"; then
+    return 1
+  fi
+  printf '%s\n' "$folder_id"
 }
 
 ensure_folder_path() {
@@ -390,11 +432,22 @@ ensure_folder_path() {
   local path="$2"
   local parent_file_id="${WORKS_DRIVE_PARENT_FILE_ID:-}"
   local component
+  local accumulated_path=""
+  local cached_folder_id
 
   IFS='/' read -r -a components <<<"$path"
   for component in "${components[@]}"; do
     [[ -n "$component" ]] || continue
-    parent_file_id="$(ensure_child_folder "$access_token" "$parent_file_id" "$component")"
+    accumulated_path="${accumulated_path:+$accumulated_path/}$component"
+    cached_folder_id="$(read_cached_folder_id "$accumulated_path")"
+    if [[ -n "$cached_folder_id" ]]; then
+      parent_file_id="$cached_folder_id"
+      continue
+    fi
+    if ! parent_file_id="$(ensure_child_folder "$access_token" "$parent_file_id" "$component")"; then
+      return 1
+    fi
+    write_cached_folder_id "$accumulated_path" "$parent_file_id"
   done
 
   printf '%s\n' "$parent_file_id"