import os
import time
import base64
import hmac
import hashlib
import requests
import json
from google.colab import drive

# Google Driveをマウント
drive.mount('/content/drive')

# 設定値
LOCATOR_URL = "https://www.pi-pe.co.jp/api/locator"
API_TOKEN = ""
API_SECRET = ""
DB_TITLE = "colabo"  # アップロードするDBタイトル
RESIST_DATE_FIELD_NAME = "resistDate"  # 登録日時フィールドの差し替えキーワード
RESIST_DATE = "now"  # 登録日時フィールドのvalue値
FILE_FIELD = "file"  # ファイル型フィールドの差し替えキーワード
DATA_PATH = "/content/drive/My Drive/Images"  # アップロードしたいファイルが格納されているGoogleDriveのパス
FILE_TYPES = ['png']  # 対象のファイルタイプ
REQUESTS_PER_MINUTE_LIMIT = 600  # リクエスト上限数

# 変更不要
MULTIPART_BOUNDARY = "SPIRAL_API_MULTIPART_BOUNDARY"

# APIサーバーのURLを取得する関数
def get_api_server_url(locator_url, token):
    headers = {
        "X-SPIRAL-API": "locator/apiserver/request",
        "Content-Type": "application/json; charset=UTF-8",
    }
    data = {
        "spiral_api_token": token
    }
    response = requests.post(locator_url, headers=headers, json=data)
    response_json = response.json()
    return response_json['location']

# 署名を作成する関数
def create_signature(token, passkey, secret):
    key = f"{token}&{passkey}"
    signature = hmac.new(secret.encode(), key.encode(), hashlib.sha1).hexdigest()
    return signature

# マルチパートデータを作成する関数
def create_multipart_data(file_path, parameters):
    with open(file_path, "rb") as file:
        file_data = file.read()

    postdata = []
    postdata.append(f"--{MULTIPART_BOUNDARY}")
    postdata.append('Content-Type: application/json; charset="UTF-8"')  # charsetを追加
    postdata.append('Content-Disposition: form-data; name="json"')
    postdata.append("")
    postdata.append(json.dumps(parameters))
    postdata.append(f"--{MULTIPART_BOUNDARY}")
    postdata.append(f'Content-Type: application/octet-stream')  # Content-Typeを修正
    postdata.append(f'Content-Disposition: form-data; name="{FILE_FIELD}"; filename="{os.path.basename(file_path)}"')
    postdata.append("")
    postdata.append(file_data)  # ファイルデータはバイナリ形式のまま送信
    postdata.append(f"--{MULTIPART_BOUNDARY}--")
    postdata.append("")  # 空行を追加

    postdata_bytes = []
    for item in postdata:
        if isinstance(item, str):  # 文字列の場合
            postdata_bytes.append(item.encode('utf-8'))  # UTF-8でエンコード
        else:
            postdata_bytes.append(item)  # すでにバイト列の場合はそのまま

    return b'\r\n'.join(postdata_bytes)

# APIリクエストを送信する関数
def api_request(api_url, data, headers):
    response = requests.post(api_url, headers=headers, data=data)
    return response.json()

# APIサーバーのURLを取得
API_URL = get_api_server_url(LOCATOR_URL, API_TOKEN)

processed_files = 0
start_time = time.time()

for root, dirs, files in os.walk(DATA_PATH):
    for file_name in files:
        if file_name.lower().endswith('png'):
            file_path = os.path.join(root, file_name)

            passkey = str(int(time.time()))
            signature = create_signature(API_TOKEN, passkey, API_SECRET)

            parameters = {
                "spiral_api_token": API_TOKEN,
                "db_title": DB_TITLE,
                "passkey": passkey,
                "signature": signature,
                "data": [
                    {"name": RESIST_DATE_FIELD_NAME, "value": RESIST_DATE}
                ],
            }

            multipart_data = create_multipart_data(file_path, parameters)

            headers = {
                "X-SPIRAL-API": "database/insert/request",
                "Content-Type": f'multipart/form-data; boundary={MULTIPART_BOUNDARY}',
            }

            api_response = api_request(API_URL, multipart_data, headers)

            # APIレスポンスの確認
            if api_response.get('code') == '0':
                print(f"File {file_name} uploaded successfully.")
            else:
                print(f"Error uploading file {file_name}: {api_response.get('message')}")
            processed_files += 1

            # リクエスト上限を超えないようにディレイ
            if processed_files % REQUESTS_PER_MINUTE_LIMIT == 0:
                time.sleep(60 - (time.time() - start_time))
                start_time = time.time()

print(f"Processed files: {processed_files}") 
            

from google.colab import drive
# Google Driveをマウント
drive.mount('/content/drive')