recce-nightly 1.16.0.20250813__py3-none-any.whl → 1.16.0.20250817__py3-none-any.whl

recce/state/state_loader.py

@@ -0,0 +1,179 @@
+import logging
+import threading
+import time
+from abc import ABC, abstractmethod
+from typing import Dict, Literal, Optional, Tuple, Union, final
+
+from recce.exceptions import RecceException
+from recce.pull_request import fetch_pr_metadata
+
+from ..util.io import SupportedFileTypes, file_io_factory
+from .const import (
+    RECCE_CLOUD_TOKEN_MISSING,
+)
+from .state import RecceState
+
+logger = logging.getLogger("uvicorn")
+
+
+class RecceStateLoader(ABC):
+    def __init__(
+        self,
+        review_mode: bool = False,
+        cloud_mode: bool = False,
+        state_file: Optional[str] = None,
+        cloud_options: Optional[Dict[str, str]] = None,
+        initial_state: Optional[RecceState] = None,
+    ):
+        self.review_mode = review_mode
+        self.cloud_mode = cloud_mode
+        self.state_file = state_file
+        self.cloud_options = cloud_options or {}
+        self.error_message = None
+        self.hint_message = None
+        self.state: RecceState | None = initial_state
+        self.state_lock = threading.Lock()
+        self.state_etag = None
+        self.pr_info = None
+        self.catalog: Literal["github", "preview"] = "github"
+        self.share_id = None
+
+        if self.cloud_mode:
+            if self.cloud_options.get("github_token"):
+                self.catalog = "github"
+                self.pr_info = fetch_pr_metadata(
+                    cloud=self.cloud_mode, github_token=self.cloud_options.get("github_token")
+                )
+                if self.pr_info.id is None:
+                    raise RecceException("Cannot get the pull request information from GitHub.")
+            elif self.cloud_options.get("api_token"):
+                self.catalog = "preview"
+                self.share_id = self.cloud_options.get("share_id")
+            else:
+                raise RecceException(RECCE_CLOUD_TOKEN_MISSING.error_message)
+
+    @abstractmethod
+    def verify(self) -> bool:
+        """
+        Verify the state loader configuration.
+        Returns:
+            bool: True if the configuration is valid, False otherwise.
+        """
+        raise NotImplementedError("Subclasses must implement this method.")
+
+    @property
+    def token(self):
+        return self.cloud_options.get("github_token") or self.cloud_options.get("api_token")
+
+    @property
+    def error_and_hint(self) -> (Union[str, None], Union[str, None]):
+        return self.error_message, self.hint_message
+
+    def update(self, state: RecceState):
+        self.state = state
+
+    @final
+    def load(self, refresh=False) -> RecceState:
+        if self.state is not None and refresh is False:
+            return self.state
+        self.state_lock.acquire()
+        try:
+            self.state, self.state_etag = self._load_state()
+        finally:
+            self.state_lock.release()
+        return self.state
+
+    @abstractmethod
+    def _load_state(self) -> Tuple[RecceState, str]:
+        """
+        Load the state from the specified source (file or cloud).
+        Returns:
+            RecceState: The loaded state object.
+            str: The etag of the state file (if applicable).
+        """
+        raise NotImplementedError("Subclasses must implement this method.")
+
+    def save_as(self, state_file: str, state: RecceState = None):
+        if self.cloud_mode:
+            raise Exception("Cannot save the state to Recce Cloud.")
+
+        self.state_file = state_file
+        self.export(state)
+
+    @final
+    def export(self, state: RecceState = None) -> Union[str, None]:
+        if state is not None:
+            self.update(state)
+
+        start_time = time.time()
+        self.state_lock.acquire()
+        try:
+            message, state_etag = self._export_state()
+            self.state_etag = state_etag
+            end_time = time.time()
+            elapsed_time = end_time - start_time
+        finally:
+            self.state_lock.release()
+        logger.info(f"Store state completed in {elapsed_time:.2f} seconds")
+        return message
+
+    @abstractmethod
+    def _export_state(self, state: RecceState = None) -> Tuple[Union[str, None], str]:
+        """
+        Export the current Recce state to a file or cloud storage.
+        Returns:
+            str: A message indicating the result of the export operation.
+            str: The etag of the exported state file (if applicable).
+        """
+        raise NotImplementedError("Subclasses must implement this method.")
+
+    def _export_state_to_file(self, file_path: str, file_type: SupportedFileTypes = SupportedFileTypes.FILE) -> str:
+        """
+        Store the state to a file. Store happens when terminating the server or run instance.
+        """
+
+        json_data = self.state.to_json()
+        io = file_io_factory(file_type)
+
+        io.write(file_path, json_data)
+        return f"The state file is stored at '{file_path}'"
+
+    def refresh(self):
+        new_state = self.load(refresh=True)
+        return new_state
+
+    def check_conflict(self) -> bool:
+        if not self.cloud_mode:
+            return False
+
+        metadata = self._get_metadata_from_recce_cloud()
+        if not metadata:
+            return False
+
+        state_etag = metadata.get("etag")
+        return state_etag != self.state_etag
+
+    def info(self):
+        if self.state is None:
+            self.error_message = "No state is loaded."
+            return None
+
+        state_info = {
+            "mode": "cloud" if self.cloud_mode else "local",
+            "source": None,
+        }
+        if self.cloud_mode:
+            state_info["source"] = "Recce Cloud"
+            state_info["pull_request"] = self.pr_info
+        else:
+            state_info["source"] = self.state_file
+        return state_info
+
+    @abstractmethod
+    def purge(self) -> bool:
+        """
+        Purge the state file or cloud storage.
+        Returns:
+            bool: True if the purge was successful, False otherwise.
+        """
+        raise NotImplementedError("Subclasses must implement this method.")

{recce_nightly-1.16.0.20250813.dist-info → recce_nightly-1.16.0.20250817.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: recce-nightly
-Version: 1.16.0.20250813
+Version: 1.16.0.20250817
 Summary: Environment diff tool for dbt
 Home-page: https://github.com/InfuseAI/recce
 Author: InfuseAI Dev Team

{recce_nightly-1.16.0.20250813.dist-info → recce_nightly-1.16.0.20250817.dist-info}/RECORD

@@ -1,7 +1,7 @@
-recce/VERSION,sha256=-dX_0PWVh-oDWK0ExN15I2skJU33S3e627n4IM8y4Qo,16
+recce/VERSION,sha256=LgdhFDUzTwEfPBZOqGNV1DvNaGwGqdifO__NmVKrjwk,16
 recce/__init__.py,sha256=yNb0QT-yoStex0VZALNJvUwtPLommoVCStcow31guqo,2392
 recce/artifact.py,sha256=ds0erp0-gGnH4Nx2vvapWR02pX8FmKQcqcd6QGqiVPw,6536
-recce/cli.py,sha256=KB9LEzeBasl2xJ4t5oBaNMIipOMOX0NprdyizszIU-Q,44564
+recce/cli.py,sha256=ECfV2IQg79_kOElCYBxd2ubBeG6KfoQqf6xcn4Tm9D4,44732
 recce/config.py,sha256=A4CbKcIdwQ4A9Q2ba4riHUshVQw1D-qDelMOdlt2khU,4661
 recce/connect_to_cloud.py,sha256=_SkX2pdyXa9FNyaHvItyYVPF2nZxy2JnCyd_o_psh2Y,4750
 recce/core.py,sha256=MtBWxemvCQDdUITwkU2JyuQYqcjlA0a76M8Kg53ECxQ,10754
@@ -12,7 +12,6 @@ recce/github.py,sha256=PEpM6ZRiinsPbXSWj4aJCKbZrN1jUXzpzAfJq_CGah4,7420
 recce/pull_request.py,sha256=aW0B1NE2LUKTam1S4TQ7smXB9KLE1DV8GnyBqNXA6j8,3832
 recce/run.py,sha256=PNafwUUJdIG8b1o0y1QuvjACpA3E-5a3keH45CrWHN0,14044
 recce/server.py,sha256=qW81QRZVYEfeDT0n6FwsN9TGhzk6-Zx2mPUjjy01DRM,22473
-recce/state.py,sha256=p4CBneaat-OFwhABfj_zG6n0mPGzrYE-F5xi28Cnzuc,34069
 recce/summary.py,sha256=Mbxvxr9KazR5o9icqhhjiGHsoAiWxQU4PdN7HytBJ1c,19154
 recce/adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 recce/adapter/base.py,sha256=T_JNeLHgiHSaegw-DbrvHOaYjMyZcjj2Qtg5cWh_fco,3548
@@ -24,51 +23,51 @@ recce/apis/check_api.py,sha256=KMCXSMl1qqzx2jQgRqCrD4j_cY3EHBbM3H2-t-6saAU,6227
 recce/apis/check_func.py,sha256=gktbCcyk3WGvWRJJ-wDnwv7NrIny2nTHWLl1-kdiVRo,4183
 recce/apis/run_api.py,sha256=eOaxOxXDkH59uqGCd4blld7edavUx7JU_DCd2WAYrL8,3416
 recce/apis/run_func.py,sha256=6wC8TDU-h7TLr2VZH7HNsWaUVlQ9HBN5N_dwqfi4lMY,7440
-recce/data/404.html,sha256=iYLvgsmZMX8xL2Y10i-icLyZQKhb6ExhA-OHKgVVIvM,59079
+recce/data/404.html,sha256=ZTSM5IfTlY2NM10T3Mt2C8i_zYm_UXDDGROsq4QQ8dM,59081
 recce/data/auth_callback.html,sha256=H-XfdlAFiw5VU2RpKBVQbwh1AIqJrPHrFA0S09nNJZA,94779
 recce/data/favicon.ico,sha256=B2mBumUOnzvUrXrqNkrc5QfdDXjzEXRcWkWur0fJ6sM,2565
-recce/data/index.html,sha256=FZI-LIX9jM3An6yKo1Xz-6PArjEs3M5BA6K13HyBmQ0,77443
-recce/data/index.txt,sha256=loUpOEFS9tebUWupB7f9PEv42g4QGPI8rB8D-xm3peI,6403
-recce/data/_next/static/RX_nnCGyE7ZOeoTStwYCZ/_buildManifest.js,sha256=xih3ZGwUimzWECoZJsiH_MfTi3cNvK2DbJ_VXPRZHWY,544
-recce/data/_next/static/RX_nnCGyE7ZOeoTStwYCZ/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
-recce/data/_next/static/chunks/0376eeba-3db2196398d62270.js,sha256=ef3PXUZocFm_JH0fYU_EfcaLKov9vMUhrat1YvV-IIU,80953
-recce/data/_next/static/chunks/068b80ea-833a129468ee1622.js,sha256=LrX07Wm6qBJ1uVFYOIkium2ZZ7TkhoCzaz5F_D5P-sM,4627
-recce/data/_next/static/chunks/0ddaf06c-c7961285f66460f6.js,sha256=rY2q_JCikDHZGB8z6nHMV7TON5MlsD4B2zxkj_9DFkE,697
-recce/data/_next/static/chunks/1268aea1-6dc1251c01bd724b.js,sha256=o2kP28fiIxMMpjU4S_PbKzQyLCofztY5NuKb3dE95Js,81786
-recce/data/_next/static/chunks/12f8fac4-16838e42d28d45c3.js,sha256=v1Bc31aD0IOpfixlIyQgOkteLw4ZCwYttEg37O8CdQk,152794
-recce/data/_next/static/chunks/235b8375-8c84c51d7bd4f6aa.js,sha256=CuHpoYgoktXz63qQW5XLGJdY1Cq0-ZJqk3LTmVOe2ug,1682
-recce/data/_next/static/chunks/2541941f-2cd3a7c2d629bd33.js,sha256=03PEb_eExziXtGvmd8-QLru-pPbrkJZh2dj3JR1VZx0,689
-recce/data/_next/static/chunks/273-f3fa401bd2b6fc91.js,sha256=HopLAKoEGP0dFN57b_41PkPqWAdxDAGzXPrVB0tHSt4,1277071
-recce/data/_next/static/chunks/2fc37c1e-910deebeb3d77c90.js,sha256=npExGSkJthgMbnqcvsiEb9b8aCCf7EQLWu1Io-6By0U,3836
-recce/data/_next/static/chunks/338-2e7eed5135c64550.js,sha256=0MRikUq4wS_iboITYISGCEWffqgtIr95bvSA_gVx5KY,264552
-recce/data/_next/static/chunks/367-ae35f5a152ee1ce5.js,sha256=PgRntGGaiUZc3lYFiGE1kddE12rkmi7Ao112KXIDzVo,84706
-recce/data/_next/static/chunks/3a92ee20-0400ffe460c7c803.js,sha256=wI3cuxjDqcE8hRKMg29Mt2dl16jcQ6lBr4JYcLk3_IU,177908
-recce/data/_next/static/chunks/62446465-423c03bb8c1f59b6.js,sha256=xiGo3FRqRpbbBqPwj2UV2zXSfCAPA35TRbyA77kiNMY,2622
-recce/data/_next/static/chunks/6af7f9e9-60aa8706f49dae45.js,sha256=bLrPO62eNT0ez4mooE8DDJITbbm_v6kFB_t8e5hlhx8,767
-recce/data/_next/static/chunks/6cf54382-49d52ae6e564e2ac.js,sha256=5v9VAkSAZMKNXwaGWjMAkkuP8YKP79F-y6c7RA4U_wE,8973
-recce/data/_next/static/chunks/6dc81886-78e2efe4538794ae.js,sha256=IM7QQ9wo-m0E1H1X7mrvEzSFUwm90eJzaom1QIDa_6M,69715
-recce/data/_next/static/chunks/715e4acc-9e2e6df4eb3809d1.js,sha256=Oc4RwCTpratqwzi7e5BvF5UIB_kC1foA4F4DKFuigMc,867
-recce/data/_next/static/chunks/72-181b430654230f0e.js,sha256=1U7NpDH1sUKrXLx8XebUVjJY9MBy2VDOEf4Z0iY2Ck4,166422
-recce/data/_next/static/chunks/786-774e3e3ed70a41b3.js,sha256=ylQ9e5VQSdHyrOZdJsXSycnJL5ZxcAQ3uv5peMz-EQo,11654
-recce/data/_next/static/chunks/8d700b6a.7fe2c8c3f4e333a6.js,sha256=drJF3LrBc6gwgD-Yj_rRz-Ap5Cf0_XH7w6HTEwZhcrU,120029
-recce/data/_next/static/chunks/a69d64b4-d6890125a87b0aba.js,sha256=M75FwsnNr_AA9jDad2sHuXK9WZxgvN10xCMcn9gRIHU,172734
-recce/data/_next/static/chunks/ae307f12-01100009689ace61.js,sha256=-5FIEA3ucphP6IGvMeJkSXJ4jxnxJ-jPuuNl1isHn48,211281
-recce/data/_next/static/chunks/c0015c5c-82c219792582c104.js,sha256=39vsBp9YjOMCTy1IehEBh7UCxYbzeh07NQcifOqMV2g,481
-recce/data/_next/static/chunks/d90cfbaa-e7d779b3912afeec.js,sha256=fEHB0fWh_Fl1TlK7NaBe11fokveLejfj4R3yOfhrAVQ,1909
-recce/data/_next/static/chunks/e07c302e-cd170429646873e1.js,sha256=0uUFHO2Mm5q2SWTHTC5loXxKoWuwOwIvt2sAyH_Rj0c,555
-recce/data/_next/static/chunks/fa5fb511-15fb438349ad5b97.js,sha256=UaaRM07bG8X9-E08mxaKLCA98d3V3OP8MqFOvZlBFjc,8899
-recce/data/_next/static/chunks/framework-7950757d31580329.js,sha256=WJSQaB_D-Qck95sQYmkMg3clAB7GfsgPYVAmEmOMkrk,182740
-recce/data/_next/static/chunks/main-app-4df79eb11c34d43c.js,sha256=3QIEMM6pO9b9MvW7xK4Toukx48-I-X1-0eqc4Bv7YaY,567
-recce/data/_next/static/chunks/main-cd6c104af638214a.js,sha256=pYGvgDmnAThczzBHShkI4dMuXUSZnzq6TIkW4ZI-61U,117829
+recce/data/index.html,sha256=tLnxhyDTmIZsWkniEJCUjUlrIa_ni7BC6T8HJiRvd0c,77442
+recce/data/index.txt,sha256=lTm8hehQUC-KN98SXNlAnYltKgudJPck8TNFbYIqGbw,6402
+recce/data/_next/static/-zBI1h60_HeMe-9ZrCw7s/_buildManifest.js,sha256=l8kfhgQIEvC1Ede_vT6iLavFimajvA32VzTAPK28pSw,544
+recce/data/_next/static/-zBI1h60_HeMe-9ZrCw7s/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
+recce/data/_next/static/chunks/068b80ea-7fab85837b21fed5.js,sha256=irzMcuBMTPe09LlF8z1CI-7_-Tr1rkA7oyuL6K_77DU,4628
+recce/data/_next/static/chunks/0ddaf06c-05b8d2a8a3e5c6f3.js,sha256=vcGJX-WEXhfmuI5WKhFYPOZgNQoddWfFD565DniftLw,697
+recce/data/_next/static/chunks/12f8fac4-170216f89d5576c6.js,sha256=1RvykSI64NYLTPwaCyxexU91wmra1Hg2XpvNfDu2mpw,152794
+recce/data/_next/static/chunks/235b8375-c2d62e0b44e3ae74.js,sha256=yxUPrQ5pIhTYt-MijQhQxl8uXailxmebL5-_p20RmI8,1682
+recce/data/_next/static/chunks/2541941f-d65696bd5d37dfa2.js,sha256=uxtHe-PUeC7kFG88JwOHTZnhuUcLEwcMpwy8M15E7Oo,689
+recce/data/_next/static/chunks/25c097a4-51d7c9060de49a6c.js,sha256=CAPNgJd_YPVRR4bEIlqSJvzEelxKbBHDhaT-Jgj3S5M,80951
+recce/data/_next/static/chunks/27e92eb0-4d8ec3c6cdf0d5f1.js,sha256=L5tol9qPfJurNZyLt9stkOVIeSgWUqLzEh6kMUL8Hjo,172733
+recce/data/_next/static/chunks/2fc37c1e-fd3c94ef85344810.js,sha256=aCCYev-qd6QV11nP0Ki-TxOFReBULkAU15ieCNys8wQ,3836
+recce/data/_next/static/chunks/370-bd3e0bf7237858bc.js,sha256=epoqgDgP1Y2vEnkJDQnArFVj8ObuW4qqxSWHToWVcAY,166418
+recce/data/_next/static/chunks/3a92ee20-a33bb5964d0b2f37.js,sha256=9O1MzNoBzy_CJsJ_9kNXsOfUCEPu6IPF1LaZl4SdXTo,177908
+recce/data/_next/static/chunks/490-ce854af0bc670d3e.js,sha256=qcOfVDlsS9UEioRtaUSUjZetYGrehutf3E64V8BUHl4,84719
+recce/data/_next/static/chunks/575-2443813ff441d02c.js,sha256=Uz72cwal6dDOkGZzBsC7a4ER7odRIupMYe204Rw8n3A,11652
+recce/data/_next/static/chunks/62446465-5277e7ad93431c07.js,sha256=38AOs2vIyqmxHVXRG4VRFu1QjG1y6i1Ws3CpEzRrC8E,2622
+recce/data/_next/static/chunks/6af7f9e9-1b5f068628caacd4.js,sha256=jAFwFVuMQAGoqIPgFBVjYe0v0rfb_JHxVocgh3oRQiY,767
+recce/data/_next/static/chunks/6cf54382-9bc9b4c7b7e4b8d8.js,sha256=W7aT-oWMJkkONrHvS7m1a9uGleQwxkwto1uUtLDMP5c,8973
+recce/data/_next/static/chunks/6dc81886-2b329a8d619994c9.js,sha256=iwW6qRDhe1OiueluOLtTjSCe5JM2EeDc150erqN5MJ4,69716
+recce/data/_next/static/chunks/715e4acc-be8c974e746c0e77.js,sha256=eAMUMEVesI_Y2ILgBXKis3j1aMAanMemVF9f8wudSGw,867
+recce/data/_next/static/chunks/76-54eeed3fa75d6146.js,sha256=P5XYY99i4SmK7CuxV_jipeIofpUtX0VeeTQH048ZegE,1278576
+recce/data/_next/static/chunks/8d700b6a.6c883c62894c0d11.js,sha256=UWq6kJeCbsXU53rRLrWwJ2LovmFdSgtxNtbBHd8i6Fk,120029
+recce/data/_next/static/chunks/975-341ea105d3d5e6ca.js,sha256=8GgemELCpkQG96SeizeRnvemGVTxG8NoN7SbntqmDGg,264544
+recce/data/_next/static/chunks/ae307f12-0f5382a8bd52593f.js,sha256=gUtta4lpvB204FEngGBQSiZGtkOIDcnA16HZjmXUCFY,211281
+recce/data/_next/static/chunks/c0015c5c-6adfd10682179dba.js,sha256=c-3i-rXXLgMYYXTI6XUpSbzVqGUjc-r43MwXVv6OCNA,480
+recce/data/_next/static/chunks/c41d1589-27a36c0adae3f11b.js,sha256=Q9hCSQqvLuOlaNdLhgoyi_puHreAi7AvqXQND_FwXrQ,81783
+recce/data/_next/static/chunks/d90cfbaa-10a3b2cefa2eb9ba.js,sha256=MWVOsgm6cUWBJ46ro6W7nKHAdlytfLiOu2xXVH-s9Xo,1909
+recce/data/_next/static/chunks/e07c302e-21163cafc81583fb.js,sha256=SwwtRu6Qf8wbzNPzCsSviwvV3rlCPcKVMbp6lyUNFvI,555
+recce/data/_next/static/chunks/fa5fb511-722a622636dc535b.js,sha256=2zOvlL6XYPTD3ZqP8FHHbCQLAjY3WmStWdhWyLgLDBE,8899
+recce/data/_next/static/chunks/framework-d3cdc9ad10f589c3.js,sha256=mcscUrllIKLVC_A3tCZvZj_2rXmEmTPmIXdhlFEhJ5Q,182740
+recce/data/_next/static/chunks/main-0aa311ca5eab4c3d.js,sha256=GLyhf-IMRzYUoaMIyGz5WHNtq7D69Lm7p-VWICm8Q-k,117836
+recce/data/_next/static/chunks/main-app-a6bd599e25466331.js,sha256=wxhVwz8VAfzs7joHV0YqpehyfOgrMdPFkN82-ecRqaU,570
 recce/data/_next/static/chunks/polyfills-42372ed130431b0a.js,sha256=CXPB1kyIrcjjyVBBDLWLKI9yEY1ZZbeASUON648vloM,112594
-recce/data/_next/static/chunks/webpack-84df6dd5ae3cf908.js,sha256=_FsecrHvMVA-0EfxZv-v8svx7W50DkEH-jb1NbVSSlU,3580
-recce/data/_next/static/chunks/app/layout-744f0a78e9e50e60.js,sha256=C1sJ1DG8RYccuG-VVTUBO-mmSjiOvJPdxqS9z0maJLk,1410
-recce/data/_next/static/chunks/app/page-e8f798c2ae3f59c2.js,sha256=5zckfX5OoWAY3HAqSLKdawS1XgWiWkbFLvbwP02LDs4,166586
-recce/data/_next/static/chunks/app/_not-found/page-c7ef8ed6dc07aaeb.js,sha256=1FE2NEJaSY3ZE8PzMkICM5FcbAbGa8K5ee0ucm2gNfU,2673
-recce/data/_next/static/chunks/pages/_app-73008661edbd5e05.js,sha256=wV9XbJ2FbdpnEztEcX5YE7ecw4ec7LPp2fYhOnLbcoo,240
-recce/data/_next/static/chunks/pages/_error-cf8bbdc3cf76c83f.js,sha256=k0UKnF-ueCkgYTKsUejnYmf31Stt9q5mOjX66XWVqrY,217
-recce/data/_next/static/css/188a3a1687e2a064.css,sha256=fKJMV0A2hM3cjuRmNhx2V_IL8SPaEsfucP-1sc0QVy4,12258
+recce/data/_next/static/chunks/webpack-e707d93a7b1691e8.js,sha256=Ibkw2UCDBVUi1MRPLO8wEAN86p380LWw6zF1R_Qamb8,3580
+recce/data/_next/static/chunks/app/layout-6b127f6a9d30f81d.js,sha256=O9k3qTFd19Z-tiHYGAiMGUNY6OQ6scwZQXSwEqhun5E,1412
+recce/data/_next/static/chunks/app/page-6e7f45c0d0eb69b3.js,sha256=-d989hXb8kBf-4J_YmS4iHjQUWh3Baag8xKBq0vWWLU,166630
+recce/data/_next/static/chunks/app/_not-found/page-f6afb3793e4b6ce1.js,sha256=yi2tvMWSs05jxM-LZ_P9JKOPztQSJF_van1IJIVeV5A,2680
+recce/data/_next/static/chunks/pages/_app-f73a7d040f1ea45c.js,sha256=0llEmHlHOPS9UZbHENhdUObZ7tDbQV2VCZ0LId_f22w,240
+recce/data/_next/static/chunks/pages/_error-c550ce619d2ef6f8.js,sha256=44UdnARhGJhgwgzpgZubFW_oD2tPnoqrV1xRUyquY_E,221
 recce/data/_next/static/css/8edca58d4abcf908.css,sha256=Ab-wk7RhsXgiX5fkJsrCJ-EVvrS1hZtOFpI0uZQSgL0,8093
+recce/data/_next/static/css/9ba7df14573e8fe7.css,sha256=HbimhkWMsVpAgDq9vpbDaQE-7ProdWfvkS6RzO7meMk,12308
 recce/data/_next/static/css/abdb9814a3dd18bb.css,sha256=Do5x9FJaOIe-pSBlbk_PP7PQAYFbDXTTl_0PWY4cnBM,1327
 recce/data/_next/static/css/c21263c1520b615b.css,sha256=2TrfHvYUzCBJKLdPzYN6hOAg1n8CoOsJFo36LJPQbOI,13037
 recce/data/_next/static/media/montserrat-cyrillic-800-normal.22628180.woff2,sha256=P5Sx5PNkdTlDYzAdulW_OPRfMokDLi6XTpmS8KJSRoI,11148
@@ -95,6 +94,12 @@ recce/models/__init__.py,sha256=F7cgALtdWnwv37R0eEgKZ_yBsMwxWnUfo3jAZ3u6qyU,209
 recce/models/check.py,sha256=jjR1SGyczjrqyK-0Zas6ikLIGSgVp54lvfcQA19AniI,1588
 recce/models/run.py,sha256=QK2gvOWvko9YYhd2NLs3BPt5l4MSCZGwpzTAiqx9zJw,1161
 recce/models/types.py,sha256=9ReOyIv3rUD3_cIQfB9Rplb0L2YQuZr4kS9Zai30nB8,5234
+recce/state/__init__.py,sha256=V1FRPTQJUz-uwI3Cn8wDa5Z9bueVs86MR_1Ti4RGfPc,650
+recce/state/cloud.py,sha256=FJS9ktORHT17nBeh1ZDq0PVC21W1Dj8Xz5u4u8X3FLM,15443
+recce/state/const.py,sha256=Me3uVQHi_uZysl3tfpmghPInuKF236X6LfMKVvJlCgA,827
+recce/state/local.py,sha256=bZIkl7cAfyYaGgYEr3uD_JLrtwlHBnu8_o1Qz2djQzw,1920
+recce/state/state.py,sha256=L1cEAl6S9rxTvSIc8ICNwOYoCYK5tR3M0kBo4NTcXK4,3617
+recce/state/state_loader.py,sha256=vGM8HCIv_3PGfcKPgq_GgG_3roloaDL_Am0oTreHmTk,5979
 recce/tasks/__init__.py,sha256=b553AtDHjYROgmMePv_Hv_X3fjh4iEn11gzzpUJz6_o,610
 recce/tasks/core.py,sha256=JFYa1CfgOiRPQ7KVTwMuxJjhMB-pvCwB-xezVt-h3RU,4080
 recce/tasks/dataframe.py,sha256=03UBWwt0DFTXlaEOtnV5i_mxdRKD7UbRayewEL2Ub48,3650
@@ -120,16 +125,15 @@ recce/util/pydantic_model.py,sha256=KumKuyCjbTzEMsKLE4-b-eZfp0gLhYDdmVtw1-hxiJw,
 recce/util/recce_cloud.py,sha256=mqZn63Fh1kZaT0L6AR9K4yqTnGpZBGEWGBHFapLmS_w,9882
 recce/util/singleton.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
 recce/yaml/__init__.py,sha256=EgXYlFeJZchatUClRDXbIC5Oqb2_nBvB2NqItYVihio,1292
-recce_nightly-1.16.0.20250813.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
+recce_nightly-1.16.0.20250817.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/test_cli.py,sha256=_ikuZW4bYEGyj5A_V7nDSyBa1tVZqc0TVm-TJ0rEYVo,5467
+tests/test_cli.py,sha256=4N_F_lg00fGBuK3I9EyaF2Doi7PpKCrjqbuJ5YI1KqE,5467
 tests/test_config.py,sha256=ODDFe_XF6gphmSmmc422dGLBaCCmG-IjDzTkD5SJsJE,1557
 tests/test_connect_to_cloud.py,sha256=b2fgV8L1iQBdEwh6RumMsIIyYg7GtMOMRz1dvE3WRPg,3059
-tests/test_core.py,sha256=7NKQc_8Krjsp3imMMmXHV6hQ7Kvw8QGE2nVtigebZYE,735
+tests/test_core.py,sha256=WgCFm8Au3YQI-V5UbZ6LA8irMNHRc7NWutIq2Mny0LE,6242
 tests/test_dbt.py,sha256=VzXvdoJNwwEmKNhJJDNB1N_qZYwp1aoJQ1sLcoyRBmk,1316
 tests/test_pull_request.py,sha256=HmZo5MoDaoKSgPwbLxJ3Ur3ajZ7IxhkzJxaOmhg6bwE,3562
-tests/test_server.py,sha256=l2EDIonpDU3CzuCpZSXTCyH_X32xdscSz-bdNSuTkjA,3397
-tests/test_state.py,sha256=phSYlyxTx2Sq0K8DXUrfaO5Jc7lgZeeWNNtc9JD3fYc,4988
+tests/test_server.py,sha256=sq03awzG-kvLNJwHaZGDvVjd2uAs502XUKZ2nLNwvRQ,3391
 tests/test_summary.py,sha256=D0WvAkdO-vzGcvholH2rfS1wTxUXjVHwWm59fWy45eA,2876
 tests/adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/adapter/dbt_adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -149,8 +153,8 @@ tests/tasks/test_row_count.py,sha256=21PaP2aq-x8-pqwzWHRT1sixhQ8g3CQNRWOZTTmbK0s
 tests/tasks/test_schema.py,sha256=7ds4Vx8ixaiIWDR49Lvjem4xlPkRP1cXazDRY3roUak,3121
 tests/tasks/test_top_k.py,sha256=YR_GS__DJsbDlQVaEEdJvNQ3fh1VmV5Nb3G7lb0r6YM,1779
 tests/tasks/test_valuediff.py,sha256=_xQJGgxsXoy2NYk_Z6Hsw2FlVh6zk2nN_iUueyRN1e8,2046
-recce_nightly-1.16.0.20250813.dist-info/METADATA,sha256=4LHS2M2wdSQcUMBikpNhYio1TFQxcwYDLKRoVnqnuR4,9366
-recce_nightly-1.16.0.20250813.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-recce_nightly-1.16.0.20250813.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
-recce_nightly-1.16.0.20250813.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
-recce_nightly-1.16.0.20250813.dist-info/RECORD,,
+recce_nightly-1.16.0.20250817.dist-info/METADATA,sha256=defY6OjhIYa5CEEp1JalxmRs-7d2XrJ1XUO4fheyqBY,9366
+recce_nightly-1.16.0.20250817.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+recce_nightly-1.16.0.20250817.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
+recce_nightly-1.16.0.20250817.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
+recce_nightly-1.16.0.20250817.dist-info/RECORD,,

tests/test_cli.py

@@ -6,7 +6,7 @@ from click.testing import CliRunner
 from recce.cli import run as cli_command_run
 from recce.cli import server as cli_command_server
 from recce.core import RecceContext
-from recce.state import RecceStateLoader
+from recce.state import CloudStateLoader
 
 
 def test_cmd_version():
@@ -48,11 +48,11 @@ class TestCommandServer(TestCase):
     @patch.object(RecceContext, "verify_required_artifacts")
     @patch("recce.util.recce_cloud.get_recce_cloud_onboarding_state")
     @patch("recce.cli.uvicorn.run")
-    @patch("recce.cli.RecceStateLoader")
+    @patch("recce.cli.CloudStateLoader")
     def test_cmd_server_with_cloud(
         self, mock_state_loader_class, mock_run, mock_get_recce_cloud_onboarding_state, mock_verify_required_artifacts
     ):
-        mock_state_loader = MagicMock(spec=RecceStateLoader)
+        mock_state_loader = MagicMock(spec=CloudStateLoader)
         mock_state_loader.verify.return_value = True
         mock_state_loader.review_mode = True
         mock_get_recce_cloud_onboarding_state.return_value = "completed"

tests/test_core.py

@@ -1,6 +1,153 @@
 # noinspection PyUnresolvedReferences
+import os
+import unittest
+from datetime import datetime
+
+from recce.core import RecceContext
+from recce.models import Check, Run, RunType
+from recce.state import ArtifactsRoot, FileStateLoader, RecceState
 from tests.adapter.dbt_adapter.conftest import dbt_test_helper  # noqa: F401
 
+current_dir = os.path.dirname(os.path.abspath(__file__))
+
+
+class TestRecceState(unittest.TestCase):
+    def test_load(self):
+        run = Run(type=RunType.QUERY, params=dict(sql_template="select * from users"))
+        check = Check(name="check 1", description="desc 1", type=run.type, params=run.params)
+
+        state = RecceState(runs=[run], checks=[check])
+        json_content = state.to_json()
+        new_state = RecceState.from_json(json_content)
+
+        run_loaded = new_state.runs[0]
+        check_loaded = new_state.checks[0]
+
+        assert run.run_id == run_loaded.run_id
+        assert check.check_id == check_loaded.check_id
+
+    def test_merge_checks(self):
+        check1 = Check(name="test1", description="", type="query")
+        check2 = Check(name="test2", description="", type="query", updated_at=datetime(2000, 1, 1))
+        check2_2 = Check(
+            name="test2_2", description="", type="query", updated_at=datetime(2020, 1, 1), check_id=check2.check_id
+        )
+        check3 = Check(name="test3", description="", type="query")
+
+        context = RecceContext()
+        state = RecceState(checks=[check1], runs=[])
+        context.import_state(state)
+        self.assertEqual(1, len(context.checks))
+        self.assertEqual(check1.name, context.checks[0].name)
+
+        context = RecceContext(checks=[check1, check2])
+        state = RecceState(checks=[check1, check2_2, check3], runs=[])
+        context.import_state(state)
+        self.assertEqual(3, len(context.checks))
+        self.assertEqual(check2_2.name, context.checks[1].name)
+
+    def test_merge_preset_checks(self):
+        check1 = Check(
+            name="test1",
+            description="test1",
+            type="query",
+            params=dict(foo="bar"),
+            updated_at=datetime(2000, 1, 1),
+            is_preset=True,
+        )
+        check2 = Check(
+            name="test2",
+            description="test2",
+            type="query",
+            params=dict(foo="bar"),
+            updated_at=datetime(2001, 1, 1),
+            is_preset=True,
+        )
+
+        context = RecceContext(checks=[check1])
+        state = RecceState(checks=[check2], runs=[])
+        context.import_state(state)
+        self.assertEqual(1, len(context.checks))
+        self.assertEqual(check2.name, context.checks[0].name)
+
+        context = RecceContext(checks=[check2])
+        state = RecceState(checks=[check1], runs=[])
+        context.import_state(state)
+        self.assertEqual(1, len(context.checks))
+        self.assertEqual(check2.name, context.checks[0].name)
+
+    def test_revert_checks(self):
+        check1 = Check(name="test1", description="", type="query")
+        check2 = Check(name="test2", description="", type="query")
+        check2_2 = Check(name="test2_2", description="", type="query", check_id=check2.check_id)
+        check3 = Check(name="test3", description="", type="query")
+
+        context = RecceContext(checks=[check1, check2])
+        state = RecceState(checks=[check2_2, check3], runs=[])
+        context.import_state(state, merge=False)
+        self.assertEqual(2, len(context.checks))
+        self.assertEqual(check2_2.name, context.checks[0].name)
+
+    def test_merge_runs(self):
+        run1 = Run(type="query")
+        run2 = Run(type="query")
+        run3 = Run(type="query")
+
+        context = RecceContext(runs=[])
+        state = RecceState(runs=[run1])
+        context.import_state(state)
+        self.assertEqual(1, len(context.runs))
+
+        context = RecceContext(runs=[run1, run2])
+        state = RecceState(runs=[run2, run3])
+        context.import_state(state)
+        self.assertEqual(3, len(context.runs))
+
+    def test_merge_dbt_artifacts(self):
+        import json
+        import os
+
+        with open(os.path.join(current_dir, "manifest.json"), "r") as f:
+            manifest = json.load(f)
+        manifest["metadata"]["generated_at"] = "2000-01-01T00:00:00Z"
+        artifacts = ArtifactsRoot(
+            base=dict(
+                manifest=manifest,
+            ),
+            current=dict(
+                manifest=manifest,
+            ),
+        )
+
+        from tests.adapter.dbt_adapter.dbt_test_helper import DbtTestHelper
+
+        adapter = DbtTestHelper().adapter
+        adapter.import_artifacts(artifacts)
+        self.assertNotEqual(adapter.base_manifest.metadata.invocation_id, manifest.get("metadata").get("invocation_id"))
+
+        manifest["metadata"]["generated_at"] = "2099-01-01T00:00:00Z"
+        adapter.import_artifacts(artifacts)
+        self.assertEqual(adapter.base_manifest.metadata.invocation_id, manifest.get("metadata").get("invocation_id"))
+
+    def test_state_loader(self):
+        # copy ./recce_state.json to temp and open
+
+        # use library to create a temp file in the context
+        import os
+        import shutil
+        import tempfile
+
+        with tempfile.NamedTemporaryFile() as f:
+            # copy ./recce_state.json to temp file
+            current_dir = os.path.dirname(os.path.abspath(__file__))
+            state_file = os.path.join(current_dir, "recce_state.json")
+            shutil.copy(state_file, f.name)
+
+            # load the state file
+            state_loader = FileStateLoader(state_file=f.name)
+            state = state_loader.load()
+            assert len(state.runs) == 17
+
 
 def test_lineage_diff(dbt_test_helper):
     sql_model1 = """

tests/test_server.py

@@ -31,9 +31,9 @@ def test_health():
 
 def test_stateless(dbt_test_helper):
     context = default_context()
-    from recce.state import RecceStateLoader
+    from recce.state import FileStateLoader
 
-    context.state_loader = RecceStateLoader()
+    context.state_loader = FileStateLoader()
     client = TestClient(app)
     response = client.get("/api/info")
     assert response.status_code == 200
@@ -44,9 +44,9 @@ def test_stateless(dbt_test_helper):
 
 def test_file_mode(dbt_test_helper):
     context = default_context()
-    from recce.state import RecceStateLoader
+    from recce.state import FileStateLoader
 
-    context.state_loader = RecceStateLoader(state_file="/tmp/recce_state.json")
+    context.state_loader = FileStateLoader(state_file="/tmp/recce_state.json")
     client = TestClient(app)
     response = client.get("/api/info")
     assert response.status_code == 200
@@ -63,9 +63,9 @@ def test_saveas_and_rename(dbt_test_helper, temp_folder):
     state_file3 = os.path.join(temp_folder, "recce_state3.json")
     os.makedirs(os.path.join(temp_folder, "dir.json"))
 
-    from recce.state import RecceStateLoader
+    from recce.state import FileStateLoader
 
-    context.state_loader = RecceStateLoader(state_file=state_file)
+    context.state_loader = FileStateLoader(state_file=state_file)
     client = TestClient(app)
 
     response = client.post("/api/save", json={"filename": "recce_state2.json"})