import requests
import boto3

from datetime import datetime
from os import environ
from time import sleep

class AlarmpyBridge():
    URL = "https://www.oref.org.il/WarningMessages/alert/alerts.json"

    HEADERS = {
        "Referer": "https://www.oref.org.il/11226-he/pakar.aspx",
        "X-Requested-With": "XMLHttpRequest",
        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.100 Safari/537.36",
    }

    def __init__(self):
        self.sess = requests.Session()
        self.sess.headers.update(self.HEADERS)
        self.last = None
        self.s3 = boto3.client("s3")
        self.bucket_name = environ.get("S3_BUCKET_NAME")

    def loop(self):
        ts = datetime.now().isoformat()
        print(f"{ts} Starting Alarmpy Bridge loop...")
        while True:
            try:
                self.fetch()
            except Exception as e:
                print(e)
                pass
            finally:
                sleep(0.5)
    
    def fetch(self):
        res = self.sess.get(self.URL).content
        if res != self.last:
            self.last = res
            self.update()
        
    
    def update(self):
        res = self.last
        ts = datetime.now().isoformat()
        print(f"{ts} Updating {res}")
        self.s3.put_object(
            Bucket=self.bucket_name,
            Body=res,
            Key="alerts.json",
            CacheControl="max-age=1, public",
            StorageClass='STANDARD',
        )

if __name__== "__main__":
    ab = AlarmpyBridge()
    ab.loop()