feat: support Range header for media downloads

Sleepstars · web-flow · commit 687ace49bf75 · 2026-01-06T17:17:10.000+01:00
All but ids that end with `files.export`.
diff --git a/src/generator/lib/util.py b/src/generator/lib/util.py
@@ -1342,5 +1342,13 @@ def unique(
     return unique(original, desired, attempts + 1)
 
 
+# Check if method supports Range header.
+# files.export doesn't support Range headers.
+def supports_range_download(m):
+    return m.get("supportsMediaDownload", False) and not m.get("id", "").endswith(
+        ".files.export"
+    )
+
+
 if __name__ == "__main__":
     raise AssertionError("For import only")
diff --git a/src/generator/templates/api/lib/mbuild.mako b/src/generator/templates/api/lib/mbuild.mako
@@ -12,7 +12,7 @@
                       re_find_replacements, ADD_PARAM_FN, ADD_PARAM_MEDIA_EXAMPLE, upload_action_fn, METHODS_RESOURCE,
                       method_name_to_variant, size_to_bytes, method_default_scope,
                       is_repeated_property, setter_fn_name, ADD_SCOPE_FN, ADD_SCOPES_FN, rust_doc_sanitize,
-                      CLEAR_SCOPES_FN, items, string_impl)
+                      CLEAR_SCOPES_FN, items, string_impl, supports_range_download)
 
     SIMPLE = "simple"
     RESUMABLE = "resumable"
@@ -130,6 +130,9 @@ pub struct ${ThisType}
  ${activity_rust_type(schemas, p)},
     % endif
 % endfor
+% if supports_range_download(m):
+    _range: Option<String>,
+% endif
 ## A generic map for additinal parameters. Sometimes you can set some that are documented online only
     ${api.properties.params}: HashMap<String, String>,
     % if method_default_scope(m):
@@ -211,6 +214,19 @@ ${self._setter_fn(resource, method, m, p, part_prop, ThisType, c)}\
         self
     }
     % endif
+    % if supports_range_download(m):
+
+    /// Sets the *Range* header for partial downloads.
+    ///
+    /// Use this to download only a portion of the file by specifying a byte range.
+    /// For example: "bytes=0-1023" downloads the first 1024 bytes.
+    ///
+    /// This is only effective when using `alt=media` parameter.
+    pub fn range(mut self, value: impl Into<String>) -> ${ThisType} {
+        self._range = Some(value.into());
+        self
+    }
+    % endif
 }
 </%def>
 
@@ -727,6 +743,12 @@ else {
                 }
                 % endif
 
+                % if supports_range_download(m):
+                if let Some(range_value) = self._range.as_ref() {
+                    req_builder = req_builder.header("Range", range_value.clone());
+                }
+                % endif
+
                 % if resumable_media_param:
                 upload_url_from_server = true;
                 if protocol == ${PROTOCOL_TYPE_MAP[resumable_media_param.protocol]} {
diff --git a/src/generator/templates/api/lib/rbuild.mako b/src/generator/templates/api/lib/rbuild.mako
@@ -6,7 +6,8 @@
                       rust_copy_value_s, organize_params, REQUEST_VALUE_PROPERTY_NAME,
                       build_all_params, rb_type_params_s, hub_type_params_s, mb_type_params_s, mb_additional_type_params,
                       struct_type_bounds_s, METHODS_RESOURCE, SPACES_PER_TAB, prefix_all_but_first_with,
-                      METHODS_BUILDER_MARKER_TRAIT, remove_empty_lines, method_default_scope, rust_doc_sanitize)
+                      METHODS_BUILDER_MARKER_TRAIT, remove_empty_lines, method_default_scope, rust_doc_sanitize,
+                      supports_range_download)
 %>\
 <%namespace name="util" file="../../../lib/util.mako"/>\
 <%namespace name="lib" file="lib.mako"/>\
@@ -119,6 +120,9 @@ impl${rb_params} ${ThisType} {
             % for p in optional_props:
             ${property(p.name)}: Default::default(),
             % endfor
+            % if supports_range_download(m):
+            _range: Default::default(),
+            % endif
 % for prop_key, custom_name in api.properties.items():
             % if prop_key == 'scopes' and not method_default_scope(m):
 <% continue %>\
