Apps Script の6分制限を跨いで Gemini Files API に大容量ファイルを再開アップロードする

/**
 * Drive ファイルの [start, end] バイトだけを取得する。
 * end は inclusive（HTTP Range の仕様）。
 * 返り値はバイト配列で、Blob 化してもファイル全体は載らない。
 */
function fetchDriveRange_(fileId, start, end) {
  const url = 'https://www.googleapis.com/drive/v3/files/'
    + encodeURIComponent(fileId) + '?alt=media&supportsAllDrives=true';
 
  const res = UrlFetchApp.fetch(url, {
    method: 'get',
    headers: {
      Authorization: 'Bearer ' + ScriptApp.getOAuthToken(),
      Range: 'bytes=' + start + '-' + end,
    },
    muteHttpExceptions: true,
  });
 
  const code = res.getResponseCode();
  // 206 Partial Content が正常。200 が返ったら Range が無視されている
  if (code !== 206) {
    throw new Error('Range fetch failed: HTTP ' + code + ' (expected 206)');
  }
  return res.getContent(); // byte[]
}

const CHUNK = 8 * 1024 * 1024; // 256 KiB の倍数であること
 
/**
 * 1チャンクを送る。isLast のときだけ finalize してファイルを確定させる。
 * finalize 時のレスポンス JSON に file.uri / file.name が入る。
 */
function putChunk_(uploadUrl, bytes, offset, isLast) {
  const command = isLast ? 'upload, finalize' : 'upload';
 
  const res = UrlFetchApp.fetch(uploadUrl, {
    method: 'post',
    contentType: 'application/octet-stream',
    payload: Utilities.newBlob(bytes),
    headers: {
      'X-Goog-Upload-Command': command,
      'X-Goog-Upload-Offset': String(offset),
    },
    muteHttpExceptions: true,
  });
 
  const code = res.getResponseCode();
  if (code < 200 || code >= 300) {
    throw new Error('chunk upload failed at offset ' + offset + ': HTTP ' + code);
  }
  return isLast ? JSON.parse(res.getContentText()) : null;
}

const SAFE_MS = 60 * 1000; // 6分のうち最後の1分は撤退に使う
 
function resumeUpload() {
  const props = PropertiesService.getScriptProperties();
  const state = JSON.parse(props.getProperty('UPLOAD_STATE'));
  const started = Date.now();
 
  while (state.offset < state.total) {
    // 実行時間を使い切る前に、状態を残して撤退する
    if (Date.now() - started > 6 * 60 * 1000 - SAFE_MS) {
      props.setProperty('UPLOAD_STATE', JSON.stringify(state));
      return; // 次のトリガーが続きを引き受ける
    }
 
    const end = Math.min(state.offset + CHUNK, state.total) - 1;
    const bytes = fetchDriveRange_(state.fileId, state.offset, end);
    const isLast = end === state.total - 1;
    const done = putChunk_(state.uploadUrl, bytes, state.offset, isLast);
 
    state.offset = end + 1;
    props.setProperty('UPLOAD_STATE', JSON.stringify(state));
 
    if (done) {
      props.setProperty('FILE_URI', done.file.uri);
      props.setProperty('FILE_NAME', done.file.name);
      props.deleteProperty('UPLOAD_STATE');
      return;
    }
  }
}

/**
 * サーバ側が受理済みのバイト数を取得する。
 * ローカルに保存した offset より、常にこちらを信用する。
 */
function queryServerOffset_(uploadUrl) {
  const res = UrlFetchApp.fetch(uploadUrl, {
    method: 'post',
    headers: { 'X-Goog-Upload-Command': 'query' },
    muteHttpExceptions: true,
  });
 
  const status = res.getHeaders()['x-goog-upload-status']; // active / final / cancelled
  if (status === 'final') return -1; // 既に確定済み。再送は不要
 
  const received = res.getHeaders()['x-goog-upload-size-received'];
  return Number(received || 0);
}

/**
 * ファイルが ACTIVE になるまで待つ。
 * Apps Script の残り時間を食い潰さないよう、上限回数で必ず抜ける。
 */
function waitUntilActive_(fileName, maxAttempts) {
  const url = 'https://generativelanguage.googleapis.com/v1beta/' + fileName;
 
  for (let i = 0; i < maxAttempts; i++) {
    const res = UrlFetchApp.fetch(url, {
      headers: { 'x-goog-api-key': getApiKey_() },
      muteHttpExceptions: true,
    });
    const file = JSON.parse(res.getContentText());
 
    if (file.state === 'ACTIVE') return file;
    if (file.state === 'FAILED') throw new Error('file processing failed: ' + fileName);
 
    Utilities.sleep(Math.min(2000 * Math.pow(2, i), 30000)); // 指数バックオフ・上限30秒
  }
  return null; // 未完了。次のトリガーで再確認する
}

/**
 * display_name を決定的に作る。同じ内容なら同じ名前になる。
 * Drive の md5Checksum を混ぜることで、差し替えを検知できる。
 */
function uploadKey_(fileId, md5) {
  return 'drive-' + fileId.slice(0, 12) + '-' + md5.slice(0, 8);
}
 
function findReusable_(key) {
  const res = UrlFetchApp.fetch(
    'https://generativelanguage.googleapis.com/v1beta/files?pageSize=100',
    { headers: { 'x-goog-api-key': getApiKey_() }, muteHttpExceptions: true }
  );
  const files = JSON.parse(res.getContentText()).files || [];
  return files.find(f => f.displayName === key && f.state === 'ACTIVE') || null;
}