/**
 * 各種設定
 */
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) {
  
  // 請求書のレイアウトごとに処理をする
  const companyPatterns = {
   pattern1 : ['株式会社テストカンパニー', '株式会社A', 'Bサービス'], 
   pattern2 : ['株式会社 デモテスト', 'C商事'], 
  };

  var p = 'pattern1';
  var isFound = false;
  for (const [pattern, companys] of Object.entries(companyPatterns)) {
    for (const company of companys) {
      if (text.includes(company)) {
        p = pattern;
        isFound = true;
        break;
      }
    }
    if (isFound) {
      break;
    }
  }

  // 実行する関数を呼び出す
  const functionName = 'extractInvoiceData_' + p;
  var handler = this[functionName];
  if (typeof handler !== 'function') {
    handler = extractInvoiceData_pattern1;
  }
  return handler(text);
}
 

/**
 * 正規表現などを使い必要情報を抽出
 *   株式会社テストカンパニー 用
 */
function extractInvoiceData_pattern1(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 extractInvoiceData_pattern2(text) {

  var data = {};

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

  // 2. 宛名の担当者を抽出
  var recipientNameMatch = text.match(/^(.*?)\n平素は格別のご高配を賜り厚く御礼申し上げます。/m);
  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(/令和\d+年\d+月\d+日/);
  if (dateMatch) {
    var date = warekiToDateAny(dateMatch[0]);
    if (date) {
      data.date = Utilities.formatDate(date, "Asia/Tokyo", "yyyy/MM/dd")
    }
  }

  // 5. 請求元の社名を抽出
  const regex = new RegExp(
    '※お振込手数料は御社ご負担にてお願い致します。'.replace(/[.*+?^${}()|[\]\\]/g, '\\$&') + 
    '\\n' +      
    '([^\\n]*)' +
    '\\n'        
  );
  var claimantMatch = text.match(regex);
  if (claimantMatch) {
    data.claimant = claimantMatch[1].trim();
  }

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

  // 7. 明細情報を抽出
  // 抽出対象の範囲を特定する
  // '商品名 / 品目' から '備考欄:' の直前までを対象とする
  var isDetailOk = true;
  const startKeyword = "商品名 / 品目";
  const endKeyword = "備考欄:";

  var startIndex = text.indexOf(startKeyword);
  var endIndex = text.indexOf(endKeyword);
  if (startIndex === -1 || endIndex === -1) {
    isDetailOk = false;
  }

  if (isDetailOk) {
    // > 明細部分のテキストを抽出
    // '商品名 / 品目' から始まるヘッダー行全体を含めて一旦抽出
    var detailBlock = text.substring(startIndex, endIndex).trim();

    // > ヘッダー行を除去する
    // ヘッダー部分（例: '商品名 / 品目\n数量\n単価\n金額'）を除去
    // OCRの読み取り誤差でヘッダー行の区切りが不正確な可能性があるため、
    // 明細の最初の品目名までで区切る
    const headerEndMark = "金額"; // ヘッダーの最後の項目名
    detailBlock = detailBlock.substring(detailBlock.indexOf(headerEndMark) + headerEndMark.length).trim();

    // > 行ごとに分割し、各行のデータを整形する
    // 各行のデータが連続した文字列になっているため、改行文字で分割
    let lines = detailBlock.split('\n')
                          .map(line => line.trim())
                          .filter(line => line.length > 0 && line !== "."); // 空行を除去
    
    // >> 数値だけの行を格納する配列
    const numberOnlyLines = [];
    // 数値以外の要素（品目名、数量、単位など）を含む行を格納する配列
    const otherLines = [];
    
    // >> 各行をチェックして配列を分ける
    lines.forEach(line => {
      // 数値判定用の正規表現:
      // ^      : 行頭
      // [\d,]+ : 数字またはカンマが1回以上続く
      // $      : 行末
      // ※ 請求書データなので、カンマ区切りも許容します
      if (/^[\d,]+$/.test(line)) {
        numberOnlyLines.push(line);
      } else {
        otherLines.push(line);
      }
    });

    // > データを行（レコード）に再構成する
    let details = [];
    
    // 数量単位の連結文字は除去

    // 連続した4つの要素で1つの明細行を構成
    const otherColumnCount = 2;// 品目と数量
    for (let i = 0; i < otherLines.length; i += otherColumnCount) {
      if (i + otherColumnCount <= otherLines.length) {
        let item = {
          no        : details.length + 1 ,
          item      : lines[i].replace(/\s*(\.+)\s*/g, '').trim(), // 例外的な文字（.）や余分なスペースを除去
          quantity  : lines[i+1],
          unitPrice : numberOnlyLines[i].replace(/[¥,\s]/g, '')   ?? "",
          price     : numberOnlyLines[i+1].replace(/[¥,\s]/g, '') ?? ""
        };      
        details.push(item);
      }
    }
    data.details = details;
  }
  return data;
}

/**
 * 和暦からDateオブジェクトに変換する
 */
function warekiToDateAny(str) {
  const regex = /(令和|平成|昭和)(\d+)年(\d+)月(\d+)日/;
  const match = str.match(regex);
  if (!match) return null;

  const era = match[1];
  const year = parseInt(match[2]);
  const month = parseInt(match[3]) - 1;
  const day = parseInt(match[4]);
  
  let seirekiYear;
  switch (era) {
    case "令和": seirekiYear = 2018 + year; break;
    case "平成": seirekiYear = 1988 + year; break;
    case "昭和": seirekiYear = 1925 + year; break;
    default: return null;
  }

  return new Date(seirekiYear, month, day);
}

/**
 * スプレッドシートに自動転記
 */
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 ?? "",       //明細情報・金額
      ]);
    }
  }
}