import argparse
from instagrapi import Client
import json, os
from datetime import datetime, timedelta
import library.lib as lib

# Start timing the script
start_time = datetime.now()

# Argument parser for command-line options
parser = argparse.ArgumentParser(description="Upload meal information to Instagram and Discord.")
parser.add_argument('--today', action='store_true', help="Use today's date instead of tomorrow's date for the MLSV_YMD timestamp.")
args = parser.parse_args()

# Set MLSV_YMD date based on the --today option
if args.today:
    MLSV_YMD = datetime.now().strftime('%Y%m%d')
else:
    MLSV_YMD = (datetime.now() + timedelta(days=1)).strftime('%Y%m%d')

print("🍪 | Retrieving saved account information.")
cl = Client(json.load(open('./temp/cookies.json')))
cl.set_proxy("socks5h://localhost:9999")

print("🍪 | Account information was successfully retrieved.")

print("📅 | Getting MLSV_YMD Timestamp")
print("📅 | Date:", MLSV_YMD)

print("🍲 | Getting Meal Info Image")
lib.얻기(MLSV_YMD)

lib.스토리_얻기(MLSV_YMD)
print("📸 | Uploading Story")
cl.photo_upload_to_story(
    path=f"temp/{MLSV_YMD}-story.png",
    caption=f"#인천상정중학교 #상정중학교 #급식 \n{MLSV_YMD}일자 급식",
    extra_data={'is_paid_partnership': False}
)
print("📸 | Story Uploaded")

lib.얻기(MLSV_YMD)
print("🖼️ | Uploading Post")
cl.photo_upload(
    f"temp/{MLSV_YMD}.png",
    caption=f"#인천상정중학교 #상정중학교 #급식 \n{MLSV_YMD}일자 급식",
    extra_data={'is_paid_partnership': False}
)
print("🖼️ | Post Uploaded")

print("🗨️ | Uploading at Discord")
lib.디스코드(MLSV_YMD)
print("🗨️ | Uploaded at Discord")

print("🎉 | All tasks completed.")
# Calculate running time
print("🕒 | Running Time:", datetime.now() - start_time)