import graphviz

class Graph:
    def __init__(self):
        self.dot = None
        self.edges = []
        self.node_counter = 0
    
    def create_graph(self, config, data):
        # TODO: process config, use for graph init
        self.dot = graphviz.Graph()

        blocks = data.get('blocks', [])
        for block in blocks:
            self.node_counter += 1

            # Create node id
            node_id = f'f{self.node_counter}'
            links = block.get('links', [])
            if len(links) > 0:
                node_id = f'{links[0]}'
            
            # Create text table for node
            texts = []
            for text in block.get('texts', []):
                person = text.get('text', '')
                link = text.get('links', [])
                if len(link) > 0:
                    person = f'<{link[0]}> {person}'
                    self.edges.append((link[0], f'{node_id}:{link[0]}'))
                texts.append(person)
            
            # Add node
            table = "|".join(texts)
            self.dot.node(node_id, fr'{table}')
        
        # Add edges
        self.dot.edges(self.edges)
    
    def __str__(self):
        return self.dot.source