from itertools import count
from netrc import netrc
import os
import io
from typing import Union, Optional
from fastapi import FastAPI, File, UploadFile
from pydantic import BaseModel
from starlette.responses import FileResponse
from fastapi import FastAPI, Response, status
import pytjnetwork as tj
from tjnetwork import *

JUNCTION = 0
RESERVOIR = 1
TANK = 2
PIPE = 1
NODE_COUNT = 0
LINK_COUNT = 2

prjs = []
inpDir = "C:/inpfiles/"
tmpDir = "C:/tmpfiles/"

if not os.path.exists(inpDir):
    os.mkdir(inpDir)

if not os.path.exists(tmpDir):
    os.mkdir(tmpDir)

app = FastAPI()

# undo/redo
@app.post("/undo/")
async def fastapi_undo(network: str):
    undo(network)
    
    return True

@app.post("/redo/")
async def fastapi_redo(network: str):
    redo(network)
    
    return True

# project operations
@app.get("/haveproject/")
async def fastapi_have_project(network: str):
    return have_project(network)

@app.post("/createproject/")
async def fastapi_create_project(network: str):
    create_project(network)
    open_project(network)

    print(network)
    return network

@app.post("/closeproject/")
async def fastapi_close_project(network: str):
    close_project(network)
    return True

@app.post("/deleteproject/")
async def fastapi_delete_project(network: str):
    delete_project(network)
    return True

# element operations
@app.get("/getnodes/")
async def fastapi_get_nodes(network: str):
    nodeCount = get_count(network, NODE_COUNT)
    nodes = []
    for i in range(0, nodeCount.value + 1):
        node_id = get_node_id(network, i)
        if node_id.error_code == 0:
            nodes.append(node_id.value)

    return nodes

@app.post("/addnode/")
async def fastapi_add_node(network: str, node: str):
    idx = add_node(network, node, JUNCTION)
    print(idx)
    print("add node")
    count = get_count(network, NODE_COUNT)
    print(count.value)

    return idx

@app.post("/addnodewithcoord/")
async def fastapi_add_node_with_coord(network: str, node: str, x: int, y: int):
    idx = add_node(network, node, JUNCTION)
    print(idx)
    count = get_count(network, NODE_COUNT)
    print(count.value)
    set_node_coord(network, node, x, y)

    return idx

@app.post("/deletenode/")
async def fastapi_delete_node(network: str, node: str):
    delete_node(network, node)
    return True

@app.get("/countnode/")
async def fastapi_count_node(network: str):
    count = get_count(network, NODE_COUNT)
    print(count)

    return count.value

@app.get("/getnodecoord/")
async def fastapi_get_node_coord(network: str, node: str):
    pt = []
    value = get_node_coord(network, node).value
    pt.append(value.x)
    pt.append(value.y)

    return pt

@app.post("/setnodecoord/")
async def fastapi_set_node_coord(network: str, node: str, x: int, y: int):
    set_node_coord(network, node, x, y)
    
    return True  

@app.get("/getlinks/")
async def fastapi_get_links(network: str):
    linkCount = get_count(network, LINK_COUNT)
    links = []
    for i in range(0, linkCount.value + 1):
        link_id = get_link_id(network, i)
        if link_id.error_code == 0:
            links.append(link_id.value)

    return links

@app.post("/addlink/")
async def fastapi_add_link(network: str, link: str, fromNode: str, toNode: str):
    idx = add_link(network, link, PIPE, fromNode, toNode)
    print(idx)

    return idx

@app.post("/deletelink/")
async def fastapi_delete_link(network: str, link: str):
    delete_link(network, link)

    return True  

@app.get("/countlink/")
async def fastapi_count_link(network: str):
    count = get_count(network, LINK_COUNT)
    print(count)

    return count.value

@app.post("/uploadinp/", status_code=status.HTTP_200_OK)
async def upload_inp(file: bytes = File(), name: str = None):
    filePath = inpDir + str(name)
    f = open(filePath, 'wb')
    f.write(file)
    f.close()

    return True

@app.get("/downloadinp/", status_code=status.HTTP_200_OK)
async def download_inp(name: str, response: Response):
    filePath = inpDir + name 
    if os.path.exists(filePath):
        return FileResponse(filePath, media_type='application/octet-stream', filename="inp.inp")
    else:
        response.status_code = status.HTTP_400_BAD_REQUEST
        return True