Initial Commit v0.0.0.0

.gitignore

@@ -0,0 +1,7 @@
+# Created by venv; see https://docs.python.org/3/library/venv.html
+bin/*
+lib/*
+lib64
+Dockerfile
+*.cfg
+*.sh

main.py

@@ -0,0 +1,156 @@
+from fastapi import FastAPI, Form, File, Request
+from fastapi.responses import HTMLResponse, RedirectResponse, JSONResponse
+from jinja2 import Template
+import os
+from typing import Annotated
+from pydantic import BaseModel
+import tempfile
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+import ipaddress
+import psycopg2
+
+from fastapi.exceptions import RequestValidationError
+from starlette.exceptions import HTTPException as StarletteHTTPException
+from fastapi.exceptions import RequestValidationError
+from fastapi.responses import PlainTextResponse
+from fastapi.exception_handlers import (
+    http_exception_handler,
+    request_validation_exception_handler,
+)
+
+temp_dir = r"/home/td/WebTemp"
+tempfile.tempdir = temp_dir
+con = psycopg2.connect("host=127.0.0.1 dbname=test user=alfred password='Alf red'")
+
+
+app = FastAPI(title="user manager", description="User session management.", version="0.0.0.1",
+               docs_url="/server/docs", redoc_url="/server/redoc", openapi_url="/server/openapi.json")
+
+app.mount("/server/static", StaticFiles(directory="static"), name="static")
+templates = Jinja2Templates(directory="templates")
+
+@app.get("/server", response_class=HTMLResponse)
+async def get_server(request: Request):
+    ip = request.headers['X-Real-IP']
+    if ipaddress.ip_address(ip) in ipaddress.ip_network('192.168.0.0/16'):
+        c = request.cookies.get("failed_reason", None)
+        if c is None:
+            return templates.TemplateResponse(request=request, name='index.html',context={"status":""})
+        else:
+            return templates.TemplateResponse(request=request, name='index.html',context={"status":f"Failed to login: {c}"})
+    else:
+        return templates.TemplateResponse(request=request, name='unknown_user.html',context={"IP":ip})
+
+
+@app.get("/server/user", response_class=HTMLResponse)
+async def get_user_server(request: Request):
+    ip = request.headers['X-Real-IP']
+    username = request.cookies.get('username', None)
+    session_key = request.cookies.get('session_key', None)
+    if username is None or len(username) == 0:
+        return RedirectResponse("/server", status_code=303)
+    if session_key is None or len(session_key) == 0:
+        return RedirectResponse("/server", status_code=303)
+
+    if ipaddress.ip_address(ip) in ipaddress.ip_network('192.168.0.0/16'):
+        active_users = get_active_users()
+        return templates.TemplateResponse(request=request, name='user.html',context={"username":username, "active_users": active_users})
+    else:
+        return templates.TemplateResponse(request=request, name='unknown_user.html',context={"IP":ip})
+
+
+def get_active_users() -> list[list]:
+    active_users = []
+    cur = con.cursor()
+    cur.execute("Select * from API_USERS where session_status = 'ACTIVE'")
+    results = cur.fetchall()
+    print(f"Results -> {len(results)}")
+    for i in results:
+        j = [i[0], i[3]]
+        active_users.append(j)
+    print("Active users: ", active_users)
+    return active_users
+
+
+# @app.post("/api/users/create", response_class=RedirectResponse)
+# async def root(username: Annotated[str, Form()], password: Annotated[str, Form()],
+#                new_user: Annotated[str, Form()], new_password: Annotated[str, Form()]):
+#      global internal_status
+#      val = helper.validate_credentials(username, password)
+#      print("Validation result: ", val)
+#      if val == "valid":
+#         print("Valid creds")
+#         helper.status_type = 'success'
+#         helper.internal_status = "Created New User"
+#         helper.create_new_user(new_user, new_password)
+#         return RedirectResponse(url=f"/api/users/", status_code=303)
+#      elif val == "invalid":
+#          print("Invalid creds")
+#          helper.status_type = 'danger'
+#          helper.internal_status = "Invalid Credentials"
+#          return RedirectResponse(url=f"/api/users/", status_code=303)
+#      else:
+#         print("Validation issue: ", val)
+#         helper.status_type = 'danger'
+#         helper.internal_status = val
+#         return RedirectResponse(url="/api/users/", status_code=303)
+     
+# @app.post("/api/users/login")
+# async def root(user: User, request: Request):
+#     ip = request.headers['X-Real-IP']
+#     if ipaddress.ip_address(ip) not in ipaddress.ip_network('192.168.0.0/16'):
+#         return {"message": "unknown user"}
+#     val = helper.validate_credentials(user.username, user.password)
+#     if val == 'valid':
+#         print("Valid creds generating session key")
+#         (session_key, expiry_date) = helper.create_new_session_key(user.username)
+#         return {"message": "credentials validated", "session_key": session_key, "expiry_date": expiry_date}
+#     elif val == 'invalid':
+#         print("Invalid creds.")
+#         return {"message": "invalid credentials"}
+#     elif val == 'session_active':
+#         print("session is active.")
+#         return {"message": "active session"}
+    
+
+# @app.post("/api/users/logout")
+# async def root(user: User, request: Request):
+#     ip = request.headers['X-Real-IP']
+#     if ipaddress.ip_address(ip) not in ipaddress.ip_network('192.168.0.0/16'):
+#         return {"message": "unknown user"}
+#     val = helper.validate_credentials(user.username, user.password)
+#     if val == 'valid':
+#         print(f"User {user.username} is not active.")
+#         return {"message", "user not active"}
+#     elif val == 'session_active':
+#         print(f"Logging out user: {user.username}, bye!")
+#         ret = helper.logout_user(user.username)
+#         return {"message": f"logout status: {str(ret)}"}
+#     else:
+#         print(f"invalid credentials.")
+#         return {"message": "invalid credentials"}
+
+
+
+
+
+# @app.exception_handler(RequestValidationError)
+# async def validation_exception_handler(request, exc):
+#     print(exc)
+#     return PlainTextResponse(str(exc), status_code=400)
+
+
+# @app.exception_handler(StarletteHTTPException)
+# async def custom_http_exception_handler(request, exc):
+#     print(f"OMG! An HTTP error!: {repr(exc)}")
+#     return await http_exception_handler(request, exc)
+
+
+# @app.exception_handler(RequestValidationError)
+# async def validation_exception_handler(request, exc):
+#     print(f"OMG! The client sent invalid data!: {exc}")
+#     return await request_validation_exception_handler(request, exc)
+
+     
+          

requirements.txt

@@ -0,0 +1,4 @@
+fastapi[standard]
+jinja2
+argon2-cffi
+psycopg2-binary

static/basic.css

@@ -0,0 +1,6 @@
+ .mytr {
+  background-color: #D6EEEE;
+} 
+ tr:nth-child(odd) {
+  background-color: #b2e6ca;
+} 

templates/index.html

@@ -0,0 +1,49 @@
+<!DOCTYPE html>
+<head>
+    <title>
+        Server Manager | Login
+    </title>
+    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-QWTKZyjpPEjISv5WaRU9OFeRpok6YctnYmDr5pNlyT2bRjXh0JMhjY6hW+ALEwIH" crossorigin="anonymous">
+</head>
+<body style="background-color:#d3e6e0">
+<script src="{{ url_for('static', path='/jquery.js') }}">
+</script>
+<script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js" integrity="sha384-YvpcrYf0tY3lHB60NNkmXc5s9fDVZLESaAA55NDzOxhy9GkcIdslK1eN7N6jIeHz" crossorigin="anonymous"></script>
+    <div align="center">
+        <h3>Server Manager Login</h3>
+    </div>
+    <br>
+    <br>
+    <br>
+    <span class="border border-primary">
+        <div align="center">
+            <form action="/api/users/login/" method="get">
+                <input type="hidden" name="redirect" value="/server/user">
+                <input type="hidden" name="session_key" value="">
+                <div class="form-row" width="50px">
+                    <div class="form-group col-md-6">
+                        <label for="username">Username</label>
+                        <input type="text" class="form-control" id="username" name="username" placeholder="Enter User Name">
+                    </div>
+                    <div class="form-group col-md-6">
+                        <label for="password">Password</label>
+                        <input type="password" class="form-control" id="password" name="password" placeholder="Enter Password">    
+                    </div>
+                </div>
+                <br>
+                <br>
+                <div class="form-row">
+                    <div class="form-group col-md-6">
+                        <button type="submit" class="btn btn-primary">Submit</button>
+                    </div>
+                </div>
+                <br>
+                <div class="form-row">
+                    <div class="form-group col-md-6" align="center">
+                        <p class="text-{{ status_type }}">{{status}}</p>
+                    </div>
+                </div>
+            </form>
+        </div>
+    </span>
+</body>

templates/unknown_user.html

@@ -0,0 +1,19 @@
+<!DOCTYPE html>
+<head>
+    <title>
+        User Creation
+    </title>
+    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-QWTKZyjpPEjISv5WaRU9OFeRpok6YctnYmDr5pNlyT2bRjXh0JMhjY6hW+ALEwIH" crossorigin="anonymous">
+</head>
+<body style="background-color:#d3e6e0">
+<script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js" integrity="sha384-YvpcrYf0tY3lHB60NNkmXc5s9fDVZLESaAA55NDzOxhy9GkcIdslK1eN7N6jIeHz" crossorigin="anonymous"></script>
+    <div align="center">
+        <h3>User Creation</h3>
+    </div>
+    <br>
+    <br>
+    <br>
+    <div align="center">
+	    <h4> You are not authorized to see this page -> {{ IP }} </h4>
+    </div>
+</body>

templates/user.html

@@ -0,0 +1,46 @@
+<!DOCTYPE html>
+<head>
+    <title>
+        Server Manager
+    </title>
+    <link href="https://cdn.jsdelivr.net/npm/[email protected]/dist/css/bootstrap.min.css" rel="stylesheet" integrity="sha384-QWTKZyjpPEjISv5WaRU9OFeRpok6YctnYmDr5pNlyT2bRjXh0JMhjY6hW+ALEwIH" crossorigin="anonymous">
+    <link href="{{ url_for('static', path='/basic.css')}}" rel="stylesheet">
+</head>
+<body style="background-color:#d3e6e0">
+<script src="{{ url_for('static', path='/jquery.js') }}">
+</script>
+<script src="https://cdn.jsdelivr.net/npm/[email protected]/dist/js/bootstrap.bundle.min.js" integrity="sha384-YvpcrYf0tY3lHB60NNkmXc5s9fDVZLESaAA55NDzOxhy9GkcIdslK1eN7N6jIeHz" crossorigin="anonymous"></script>
+    <div align="center">
+        <h3>Server Manager, Welcome {{ username }} </h3>
+    </div>
+    <br>
+    <br>
+    <br>
+    <span class="border border-primary">
+        <div align="center">
+            <table id="active_users" class="table table-sm">
+                <thead>
+                <tr>
+                    <th scope="col">
+                        Username
+                    </th>
+                    <th scope="col">
+                        Session Status
+                    </th>
+                </tr>
+                </thead>
+                <tbody>
+                {% for user in active_users %}
+                <tr class="mytr">
+                    {% for item in user %}
+                    <td>
+                        {{ item }}
+                    </td>
+                    {% endfor %}
+                </tr>
+                {% endfor %}
+                </tbody>
+            </table>
+        </div>
+    </span>
+</body>