/**
 * 各種設定
 */
const SPREADSHEET_ID       = "YOUR_SPREADSHEET_ID";  //スプレッドシートのID
const SHEET_NAME           = "YOUR_SHEET_NAME";      // 請求書情報を書き込むスプレッドシートのシート名

const SOURCE_FOLDER_ID     = "YOUR_FOLDER_ID";       // 請求書画像を置くフォルダのフォルダID
const DOC_FOLDER_ID        = "YOUR_FOLDER_ID2 or ''";//GoogleDocファイルを出力するフォルダのID、空欄時はファイルは保存しない
const PROCESSED_FOLDER_ID  = "YOUR_FOLDER_ID3 or ''";//処理済みの請求書画像を移動させるフォルダのID、空欄時は移動しない
const SUPPORTED_MIME_TYPES = [MimeType.PNG, MimeType.JPEG, MimeType.GIF, MimeType.BMP, MimeType.PDF];

/**
 * カスタムメニューを追加
 */
function onOpen() {
  SpreadsheetApp.getUi()
    .createMenu("請求書取込")
    .addItem("▼ 実行", "ocrInvoiceToSheet")
    .addToUi();
}

/**
 * 請求書画像をOCR処理し、情報を抽出してスプレッドシートに書き込む
 */
function ocrInvoiceToSheet() {

  const sourceFolder = DriveApp.getFolderById(SOURCE_FOLDER_ID);

  for (let mimeType of SUPPORTED_MIME_TYPES) {
    const files = sourceFolder.getFilesByType(mimeType);

    while (files.hasNext()) {
      const file = files.next();

      // OCR処理して、ドキュメントに変換、テキスト情報を取得
      const text = ocrInvoice(file);

      // 取得したテキスト情報を解析（正規表現等）して各項目の情報を抜き出す
      const data = extractInvoiceData(text);

      // シートに追記
      saveToSheet(data);

      // 後処理：処理が完了した画像ファイルを「processed」フォルダに移動します。
      if (PROCESSED_FOLDER_ID) {
        const processedFolder = DriveApp.getFolderById(PROCESSED_FOLDER_ID);
        file.moveTo(processedFolder)

      }
    }

  }

}

/**
 * Google Drive OCRで画像ファイルをテキスト化する
 */
function ocrInvoice(file) {
  // ファイルをOCR処理します（Googleドキュメントに変換）
  // ☆Drive Api v2 の場合
  // const resource = {
  //   title: file.getName()
  // };
  // const options = {
  //   "ocr": true,
  //   "ocrLanguage": "ja",
  // };
  // const doc = Drive.Files.copy(resource, file.getId(), options);

  // ☆Drive Api v3 の場合
  const resource = {
    title: file.getName(),
    mimeType: MimeType.GOOGLE_DOCS
  };
  const options = {
    ocr: true,
    ocrLanguage: 'ja'
  };
  const doc = Drive.Files.create(
    resource, // 新しいドキュメントのメタデータ
    file.getBlob(), // 元の画像ファイルの中身 (Blob)
    options // OCR設定
  );

  // 変換したGoogleドキュメントからテキストを取得します
  const text = DocumentApp.openById(doc.id).getBody().getText(); 

  // 後処理：作成したGoogleドキュメントを移動または削除します
  if ( DOC_FOLDER_ID ) {
    const docFolder = DriveApp.getFolderById(DOC_FOLDER_ID);
    // docをDOC_FOLDER_IDのフォルダに移動します。
    DriveApp.getFileById(doc.id).moveTo(docFolder);
  } else {
    // docを削除します
    DriveApp.getFileById(doc.id).setTrashed(true);
  }

  // テキスト情報を返却します
  return text;
}

/**
 * 正規表現で「金額」と「日付」を抽出
 */
function extractInvoiceData(text) {
  var data = {};

  // 1. 宛名の社名を抽出
  // 「ご担当:」の前にある社名を取得
  var recipientCompanyMatch = text.match(/^(.*?)\nご担当:/m);
  if (recipientCompanyMatch) {
    data.recipientCompany = recipientCompanyMatch[1].trim();
  }

  // 2. 宛名の担当者を抽出
  // 「ご担当:」と「様」の間にある名前を取得
  var recipientNameMatch = text.match(/ご担当:(.*?)様/);
  if (recipientNameMatch) {
    data.recipientName = recipientNameMatch[1].trim();
  }
  
  // 3. 合計金額を抽出
  var amountMatch = text.match(/¥\s?[\d,]+/);
  if (amountMatch) {
    data.amount = amountMatch[0].replace(/[¥,\s]/g, '');
  }

  // 4. 請求日付を抽出
  var dateMatch = text.match(/請求日:\s*(\d{4}[年\/\-]\d{1,2}[月\/\-]\d{1,2}日?)/);
  if (dateMatch) {
    data.date = dateMatch[1]
      .replace(/[\u200b\s]/g, '') // ゼロ幅スペースや空白を除去
      .replace(/[年月]/g, '/')
      .replace(/[日]/g, '');
  }

  // 6. 請求元の社名を抽出
  const claimantMatch = text.match(/^(.*)\n〒/m);
  if (claimantMatch) {
    data.claimant = claimantMatch[1];
  }

  // 7. 小計・消費税を抽出
  const subtotalMatch = text.match(/小計\s*\n.*\n(¥[\d,]+)/);
  if (subtotalMatch) {
    data.subtotal = subtotalMatch[1].replace(/[¥,\s]/g, '');
  }
  const taxMatch = text.match(/消費税額\s*\n.*\n(¥[\d,]+)/);
  if (taxMatch) {
    data.tax = taxMatch[1].replace(/[¥,\s]/g, '');
  }

  // 8. 明細情報を抽出
  // 明細表の開始と終了を特定し、その間のテキストを処理
  var lines = text.split('\n');
  var startLineIndex = -1;
  var endLineIndex = -1;
  var startLineIndex2 = -1;
  var endLineIndex2 = -1;

  // 明細表の開始行(No., 項目)を特定
  for (var i = 0; i < lines.length; i++) {
    if (lines[i] && lines[i].includes('No.') && lines[i].includes('項目')) {
      startLineIndex = i + 1;
      break;
    }
  }
  // 明細表の終了行(No., 項目)を特定
  for (var i = startLineIndex; i < lines.length; i++) {
    if (lines[i] && lines[i].includes('数量')) {
      endLineIndex = i-1;
      break;
    }
  }

  // 明細表の開始行(数量, 単価, 金額)を特定
  for (var i = startLineIndex; i < lines.length; i++) {
    if (lines[i] && lines[i+1] && lines[i+2] 
        && lines[i].includes('数量') && lines[i+1].includes('単価') && lines[i+2].includes('金額')) {
      startLineIndex2 = i+2 +1;
      break;
    }
  }
  // 明細表の終了行(数量, 単価, 金額)を特定
  for (var i = startLineIndex2; i < lines.length; i++) {
    if (lines[i] && lines[i].includes('小計')) {
      endLineIndex2 = i-1;
      break;
    }
  }
  data.details = [];
  if (startLineIndex !== -1 && endLineIndex !== -1) {
    for (var i = startLineIndex; i < endLineIndex; i++) {
      var no = lines[i].trim();
      i++;
      if (!lines[i]) {
        break;
      }
      var item = lines[i].trim();
      if (/^\d+$/.test(item)) {
        break;
      }
      data.details.push({
        no: no, 
        item: item, 
        quantity: "",
        unitPrice: "",
        price: "",
      });
    }
  }
  if (startLineIndex2 !== -1 && endLineIndex2 !== -1) {
    var detailsRow = 0;
    for (var i = startLineIndex2; i < endLineIndex2; i++) {
      var quantity = lines[i].trim();
      i++;
      if (!lines[i]) {
        break;
      }
      var unitPrice = lines[i].trim().replace(/[¥,\s]/g, '');
      i++;
      if (!lines[i]) {
        break;
      }
      var price = lines[i].trim().replace(/[¥,\s]/g, '');
      if ( !data.details[detailsRow] ) {
        data.details.push({
          no: "", 
          item: "", 
          quantity: "",
          unitPrice: "",
          price: "",
        });
      }
      data.details[detailsRow].quantity = quantity;
      data.details[detailsRow].unitPrice = unitPrice;
      data.details[detailsRow].price = price;
      detailsRow++;
    }
  }
  return data;
}

/**
 * スプレッドシートに自動転記
 */
function saveToSheet(data) {
  const sheet = SpreadsheetApp.openById(SPREADSHEET_ID).getSheetByName(SHEET_NAME);

  // 請求ヘッダ情報
  sheet.appendRow([
    new Date(), 
    data.recipientCompany ?? "",//宛名
    data.recipientName ?? "",   //宛名・担当者
    data.claimant ?? "",        //請求者
    data.date ?? "",            //請求日
    data.subtotal ?? "",        //小計
    data.tax ?? "",             //消費税
    data.amount ?? "",          //合計金額
  ]);
  // 請求明細情報
  if (data.details) {
    for (const detail of data.details) {
      sheet.appendRow([
        "",
        "",
        "",
        "",
        "",
        "",
        "",
        "",
        detail.item ?? "",        //明細情報・項目
        detail.quantity ?? "",    //明細情報・数量
        detail.unitPrice ?? "",   //明細情報・単価
        detail.price ?? "",       //明細情報・金額
      ]);
    }
  }
}