"""
This code is an implementation of Funsearch (https://www.nature.com/articles/s41586-023-06924-6) and is heavily inspired by the original code (https://github.com/google-deepmind/funsearch)

    **Citation**:
    
    @Article{FunSearch2023,
        author  = {Romera-Paredes, Bernardino and Barekatain, Mohammadamin and Novikov, Alexander and Balog, Matej and Kumar, M. Pawan and Dupont, Emilien and Ruiz, Francisco J. R. and Ellenberg, Jordan and Wang, Pengming and Fawzi, Omar and Kohli, Pushmeet and Fawzi, Alhussein},
        journal = {Nature},
        title   = {Mathematical discoveries from program search with large language models},
        year    = {2023},
        doi     = {10.1038/s41586-023-06924-6}
    }
"""
from typing import Any, Optional
import dataclasses


@dataclasses.dataclass
class AbstractArtifact:
    """ Abstract class for artifacts."""
    
    name: str
    args: str
    body: str
    return_type: Optional[str] = None
    docstring: Optional[str] = None

    def __str__(self)->str:
        raise NotImplementedError()
    
    def __setattr__(self, name: str, value: str) -> None:
        # Ensure there aren't leading & trailing new lines in `body`.
        if name == 'body':
            value = value.strip('\n')
        # Ensure there aren't leading & trailing quotes in `docstring``.
        if name == 'docstring' and value is not None:
            if '"""' in value:
                value = value.strip()
                value = value.replace('"""', '')
        super().__setattr__(name, value)
    
    def rename_artifact_calls(self, source_name, target_name) -> str:
        raise NotImplementedError
    
    def text_to_artifact(self):
        raise NotImplementedError
    
    def calls_ancestor(self,artifact_to_evolve: str) -> bool:
        raise NotImplementedError