#!/usr/bin/env python3
# coding: utf-8
#
# automate UML (Unified Modeling Language) diagrams creation from CSV
# (Comma Separated Values) files using Python and PlantUM
#
# designed by SAMBUICHI, Nobuyuki (Sambuichi Professional Engineers Office)
# written by SAMBUICHI, Nobuyuki (Sambuichi Professional Engineers Office)
#
# MIT License
#
# (c) 2023 SAMBUICHI Nobuyuki (Sambuichi Professional Engineers Office)
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
import csv
import os
import subprocess
import re

SEP = os.sep  # Define the system-specific path separator

# Function to convert a given path to an absolute path
def file_path(pathname):
    if SEP == pathname[0:1]:
        return pathname
    else:
        pathname = pathname.replace('/', SEP)
        dir = os.path.dirname(__file__)
        new_path = os.path.join(dir, pathname)
        return new_path

# Function to load target class names from a file
def load_target_classes(file_path):
    with open(file_path, 'r', encoding='utf-8-sig') as file:
        return [line.strip() for line in file if line.strip()]

# Function to parse a CSV file and generate UML code
def parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules):
    classes       = {}
    relationships = []
    object_class  = file_path(f"{source}{object_class}")
    uml_file_path = file_path(f"{source}{uml_file_path}")

    with open(object_class, 'r', encoding='utf-8-sig') as file:
        reader = csv.reader(file)
        next(reader)  # Skip the header row
        target_class = []
        for row in reader:
            # Skip empty rows
            if not row or not any(row):
                continue
            # Extract and strip values from the row
            id, prop_type, level, multiplicity, class_name, prop_name, data_type, assoc_class, description = [x.strip() for x in row]
            pattern = r'([A-Z]+)[0-9]+'
            match = re.search(pattern, id)
            module = None
            if match:
                module = match.group(1)
            if module in target_modules:
                if class_name and class_name not in target_class:
                    target_class.append(class_name)
                if assoc_class and assoc_class not in target_class:
                    target_class.append(assoc_class)

    with open(object_class, 'r', encoding='utf-8-sig') as file:
        reader = csv.reader(file)
        next(reader)  # Skip the header row

        for row in reader:
            # Skip empty rows
            if not row or not any(row):
                continue

            # Extract and strip values from the row
            id, prop_type, level, multiplicity, class_name, prop_name, data_type, assoc_class, description = [x.strip() for x in row]
            if class_name in target_class:
                if class_name and class_name not in classes:
                    classes[class_name] = {"attributes": []}
                elif assoc_class and assoc_class not in classes:
                    classes[assoc_class] = {"attributes": []}

                # Handle attributes and relationships
                if 'Attribute' in prop_type:
                    # Additional handling for primary key attribute
                    if 'PK' in prop_type:
                        classes[class_name]["attributes"].append(f'+PK: {prop_name} : {data_type} ')
                    else:
                        classes[class_name]["attributes"].append(f'+{prop_name} : {data_type}')
                elif prop_type in ['Association', 'Aggregation', 'Composition', 'Specialized Class']:
                    # Determine the relationship symbol based on the property type
                    if prop_type == 'Association':
                        relation_symbol = '--'
                    elif prop_type == 'Aggregation':
                        relation_symbol = 'o--'
                    elif prop_type == 'Composition':
                        relation_symbol = '*--'
                    elif prop_type == 'Specialized Class':
                        relation_symbol = '--|>'
                    relation = f'"{class_name}" {relation_symbol} "{assoc_class}" : {prop_name}'
                    relationships.append(relation)

    uml_code = '@startuml\n'
    # Build UML code for each class
    for class_name, properties in classes.items():
        if not class_name:
            continue
        uml_code += f'class "{class_name}" {{\n'
        for attr in properties["attributes"]:
            uml_code += f'    {attr}\n'
        uml_code += '}\n\n'

    # Add relationships to UML code
    for relation in relationships:
        uml_code += relation + '\n'

    uml_code += '@enduml'

    # Write the resulting UML code to a text file
    with open(uml_file_path, 'w') as f:
        f.write(uml_code)

    # Function to generate UML image using PlantUML
# def generate_uml_image(plantuml_path, uml_file_path):
    subprocess.run(["java", "-DPLANTUML_LIMIT_SIZE=16384", "-jar", plantuml_path, uml_file_path])

# Main execution: Setting up paths and generating UML code and image
plantuml_path = file_path('UML/bin/plantuml-mit-1.2023.12.jar')
source = 'UML/'
object_class = 'source/mim.csv'

# Generate UML from CSV and then create the image
# Sales module
target_modules =  ['SAL','GEN','BS']
uml_file_path = 'diagram/SAL_UML.txt'
parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules)
print(uml_file_path)

# Accounts Receivable module
target_modules =  ['AR','GEN','BS']
uml_file_path = 'diagram/AR_UML.txt'
parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules)
print(uml_file_path)

# Purchase module
target_modules =  ['PUR','GEN','BS']
uml_file_path = 'diagram/PUR_UML.txt'
parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules)
print(uml_file_path)

# Accounts Payable module
target_modules =  ['AP','GEN','BS']
uml_file_path = 'diagram/AP_UML.txt'
parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules)
print(uml_file_path)

# General Ledger module
target_modules =  ['GL','GEN','BS']
uml_file_path = 'diagram/GL_UML.txt'
parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules)
print(uml_file_path)

# Inventory module
target_modules =  ['INV','GEN','BS']
uml_file_path = 'diagram/INV_UML.txt'
parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules)
print(uml_file_path)

# PPE module
target_modules =  ['PPE','GEN','BS']
uml_file_path = 'diagram/PPE_UML.txt'
parse_csv_to_uml(plantuml_path, source, object_class, uml_file_path, target_modules)
print(uml_file_path)