import json 
import datetime
import utils
import photos

class DailyPhoto:
    def __init__(self, date: datetime.date, photo: photos.SunriseOrSunsetPhoto):
        self.date = date
        self.photo = photo

    def as_dict(self) -> dict:
        return { 
            "date": self.date.strftime("%Y-%m-%d"), 
            "photo": self.photo.as_json()
        }
    
class DailyPhotoSet:
    def __init__(self, start_date: datetime.date = datetime.date(2023, 1, 21)):
        self._date = start_date
        self._photos = []

    def add_photo(self, photo: photos.SunriseOrSunsetPhoto):
        self._photos.append(DailyPhoto(self._date, photo))
        self.photos.sort(key=lambda photo: photo.photo.id)
        self._date += datetime.timedelta(days=1)

    def as_json(self) -> str:
        return json.dumps({ 
            "photos": [photo.as_dict() for photo in self.photos]
        })

    @property
    def photos(self) -> list[DailyPhoto]:
        return list(sorted(self._photos, key=lambda photo: photo.date))
        
    @property
    def current_date(self) -> datetime.date:
        return self._date;

def get_all_photos() -> list[photos.SunriseOrSunsetPhoto]:
    file_name = utils.make_relative_file_name('data/all_photos.json')
    print(f"Reading {file_name}...")

    with open(file_name, "r") as infile:
        photo_set = photos.SunriseOrSunsetPhotoSet.from_json(infile.read())
        print(f"Returning {len(photo_set.photos_sorted_by_id)} photos...")
        return photo_set.photos_sorted_by_id

def main(): 
    print("Creating daily photo mapping...")
    all_photos = get_all_photos()

    daily_photos = DailyPhotoSet()

    for photo in all_photos:
        daily_photos.add_photo(photo)

    with open(utils.make_relative_file_name('data/daily_photos.json'), 'w') as outfile:
        outfile.write(daily_photos.as_json())

main()