from flask import Blueprint, render_template, request, url_for, flash, redirect, send_from_directory,  make_response, current_app
from database import get_owp_db
from flask_autoindex import AutoIndex
from werkzeug.utils import secure_filename
from datetime import datetime
import os
import sqlite3
import time
import json
import shutil

# postgresql
import psycopg2
with open('/var/open-ww3-project-ww3-tw/config/blog/db_config.json', 'r') as file:
    data = json.load(file)
    # 设定数据库连接配置
    conn_params = {
    "dbname" : data['dbname'],
    "user" : data['user'],
    "password": data['password'],
    "host": data['host'],
    "port": data['port']
    }




def get_owp_db_conn():
    conn = sqlite3.connect('/var/open-ww3-project-ww3-tw/databases/sqlite/owp.db')
    conn.row_factory = sqlite3.Row
    return conn

def get_message_db_conn():
    conn = sqlite3.connect('/var/open-ww3-project-ww3-tw/databases/sqlite/message.db')
    conn.row_factory = sqlite3.Row
    return conn

blog_bp = Blueprint('blog', __name__)

index_path = '/var/open-ww3-project-ww3-tw/folders/'
index = AutoIndex(blog_bp, browse_root=index_path, add_url_rules=False)

# 重定向
@blog_bp.route('/folders/re/')
def re():
    return('404 not found<br><a style="color: white;">这是你不该知道的地方</a><br><iframe src="//player.bilibili.com/player.html?isOutside=true&aid=115578442356850&bvid=BV12CCdBkEL6&cid=34120008494&p=1&autoplay=0" scrolling="no" border="0" frameborder="no" framespacing="0" allowfullscreen="true" style="width: 100%; height: 100%;"></iframe>')

@blog_bp.route('/folders/paint/2024.2.13/临摹出来的乐色/不好说的人物/')
@blog_bp.route('/folders/paint/2024.2.13/临摹出来的乐色/不好说的人物/<path:subpath>')
def no_say_man(subpath=None):
    return redirect(url_for('blog.re'))

@blog_bp.route('/folders/game/')
@blog_bp.route('/folders/game/<path:subpath>')
def no_game(subpath=None):
    return redirect(url_for('blog.re'))
# 重定向

@blog_bp.route('/')
def home():
    conn = psycopg2.connect(**conn_params)
    cur = conn.cursor()
    sql_logs = "SELECT * FROM logs;"
    sql_posts = "SELECT* FROM posts WHERE status = 1;"

    cur.execute(sql_logs)
    cur_logs = cur.fetchall()
    cur.execute(sql_posts)
    cur_posts = cur.fetchall()
    cur.close()
    conn.close()

    # 格式转换
    dict_logs = [
        {
            "id": row[0],
            "date": row[1],
            "content": row[2]
        } for row in cur_logs
    ]
    dict_posts = [
        {
            "id": row[0],
            "date": row[1],
            "title": row[2],
            "content": row[3]
        } for row in cur_posts
    ]


    # 计数器
    count_file = '/var/open-ww3-project-ww3-tw/static/counter.txt'
    def load_couter():
        if os.path.exists(count_file):
            with open(count_file, 'r') as f:
                return int(f.read().strip())
        return 0
    def save_couter(couter):
        with open(count_file, 'w') as f:
            f.write(str(couter))
    counter = load_couter()
    counter += 1
    save_couter(counter)

    return render_template('blog/home.html', counter = counter, logs = dict_logs[::-1], posts = dict_posts[::-1])

@blog_bp.route('/about/')
def about():
    return render_template('blog/about.html')
    pass

@blog_bp.route('/posts/')
def posts_list():
    conn = psycopg2.connect(**conn_params)
    sql_posts = "SELECT * from posts where status= 1;"
    cur = conn.cursor()
    cur.execute(sql_posts)
    cur_posts = cur.fetchall()
    conn.close()

    dict_posts = [
        {
            "id": row[0],
            "date": row[1],
            "title": row[2],
            "content": row[3] 
        } for row in cur_posts
    ]
    return render_template('blog/list.html', posts = dict_posts[::-1])

# 迁移中
@blog_bp.route('/posts/<int:posts_id>/')
def show_posts_id(posts_id):
    # json
    message_path = f"/var/open-ww3-project-ww3-tw/static/message/{posts_id}"
    
    if not os.path.exists(message_path):
        os.makedirs(message_path)
    message_json = f"{message_path}/chat.json"
    if not os.path.exists(message_json):
        with open(message_json, 'w', encoding='utf-8') as file:
            json.dump([], file, ensure_ascii=False)
    message_ss = json.load(open(message_json, 'r', encoding='utf-8'))

    conn = psycopg2.connect(**conn_params)
    cur = conn.cursor()
    sql_posts = f"SELECT * FROM posts WHERE status = 1 AND id = {posts_id};"
    cur.execute(sql_posts)
    posts = cur.fetchone()
    cur.close()
    conn.close()
    if posts is None:
        shutil.rmtree(message_path)
        return f'未找到该文章{posts_id}<title>未找到该文章{posts_id}/</title><br><a href="../">返回文章列表</a>', 404
    dict_posts = [
        {
            "id": posts[0],
            "date": posts[1],
            "title": posts[2],
            "content": posts[3],
            "status": posts[4],
            "update_time": posts[5]
        }
    ]

    conn_message = get_message_db_conn()
    sql_message = "SELECT * FROM messages WHERE posts_id = ?"
    message = conn_message.execute(sql_message, (posts_id,)).fetchall()
    conn.close()
    conn_message.close()

    return render_template('blog/posts.html', posts = dict_posts, message = message[::-1], message_ss=message_ss[::-1])


# posts 留言
@blog_bp.route('/posts/<int:posts_id>/chat/', methods=['POST', 'GET'])
def posts_chat(posts_id):
    message_path = f"/var/open-ww3-project-ww3-tw/static/message/{posts_id}/chat.json"
    if request.method == "POST":
        name = request.form.get("name", None)
        data = request.form.get("data", None)
        
        date = datetime.now().strftime("%Y-%m-%d %H:%M:%S")

        json_push = {
            "name": name,
            "data": data,
            "date": date
        }
        
        with open(message_path, 'r', encoding='utf-8') as f:
            old_date_data = json.load(f)
        old_date_data.append(json_push)
        with open(message_path, 'w', encoding='utf-8') as f:
            json.dump(old_date_data, f, ensure_ascii=False, indent=4)
        return "留言成功<br><a href='../'>返回文章</a>"
    return "GET"

# 数据库测试
@blog_bp.route('/db_test/')
def test_db():
    db = get_owp_db()

    try:
        cur = db.execute('SELECT sqlite_version();')
        ver = cur.fetchone()[0]
        return f"数据库连接成功！SQLite 版本是: {ver}"
    except Exception as e:
        return f"连接失败: {str(e)}"

######
## 网站地图 及其 robots 配置

@blog_bp.route('/robots.txt/')
def robots():
    return send_from_directory(current_app.static_folders, 'robots.txt')

@blog_bp.route('/google02f6a3f6004a32c6.html')
def google02f6a3f6004a32c6():
    return send_from_directory(blog_bp.static_folders, 'google02f6a3f6004a32c6.html')


@blog_bp.route('/sitemap.xml/')
def sitemap():
    conn = get_owp_db_conn()
    sql_posts = "SELECT * from posts where status = 1;"
    posts = conn.execute(sql_posts).fetchall()
    conn.close()
    template = render_template('blog/sitemap.xml', posts=posts[::-1], base_url="https://open-ww3-project.ww3.tw/blog/")
    response = make_response(template)
    response.headers['Content-Type'] = 'application/xml'
    return response

@blog_bp.route('/kami/')
def kami():
    return redirect(url_for('kami.home'))

@blog_bp.route('/folders/')
@blog_bp.route('/folders/<path:path>')
def autoindex(path='.'):
    return index.render_autoindex(path, template='blog/folders.html')


@blog_bp.route('/upload/',  methods=['POST' , 'GET'])
def upload():
    if request.method == "POST":
        f = request.files.get('img')
        filename = f.filename
        with open(f'/var/open-ww3-project-ww3-tw/static/upload/img/{filename}', 'wb') as tf:
            tf.write(f.read())
        return '上传成功<meta http-equiv="refresh" content="1.2;url=https://ww3.tw/blog/upload/">'
    img_files = '/var/open-ww3-project-ww3-tw/static/upload/img/'
    images = [img for img in os.listdir(img_files) if img.endswith(('.png', '.jpg', '.jpeg', '.gif'))]
    return render_template('blog/upload.html', images=images)

@blog_bp.route('/message/')
def messages():
    return render_template('message/home.html')


@blog_bp.route('/class')
def class_page():
#    return ("this is Class_page")
    return render_template('blog/class.html')