<?php
//查询term字典
require_once "../path.php";
require_once "../public/_pdo.php";

//is login
if (isset($_COOKIE["username"]) && !empty($_COOKIE["username"])) {
    $username = $_COOKIE["username"];
} else {
    $username = "";
}

if (isset($_GET["op"])) {
    $_op = $_GET["op"];
} else if (isset($_POST["op"])) {
    $_op = $_POST["op"];
}
if (isset($_GET["word"])) {
    $_word = mb_strtolower($_GET["word"], 'UTF-8');

}

if (isset($_GET["id"])) {
    $_id = $_GET["id"];
}

global $PDO;
PDO_Connect("" . _FILE_DB_PALI_SENTENCE_);

if (isset($_GET["sent"])) {
    $_sent = mb_strtolower($_GET["sent"], 'UTF-8');
}

// 输入一个句子，输出整个句子的单词 array
function words_of_sentence(string $sent)
{
    $words = preg_split("/[ \.\[\]\{\}\-,';‘’–0123456789]+/", $sent); // 去除标点、数字
    $words = array_filter($words); // 去除空词
    $words = array_filter($words, function ($item) {
        if ($item != 'ca' && $item != 'vā' && $item != 'na') {
            return $item;
        }
    }); // 去除 ca, vā 和 na
    return $words;
}

// 采用 jaccard 相似度，考虑到圣典中的相似句单词、句式都是非常接近的
function jaccard_similarity($words_of_sent1, $words_of_sent2)
{
    $intersect = count(array_intersect($words_of_sent1, $words_of_sent2));
    $union = count($words_of_sent1) + count($words_of_sent2) - $intersect;
    if ($union) {
        return $intersect / $union;
    } else {
        return 0;
    }
}

// 带顺序的 jaccard 算法，当前效果一般，TODO: 切片相同时加入得分
function ordered_jaccard_similarity($words_of_sent1, $words_of_sent2)
{
    $score = 0;
    $k = min(count($words_of_sent1), count($words_of_sent2));
    for ($i = 1; $i < $k; $i++) {
        $score += jaccard_similarity(
            array_slice($words_of_sent1, 0, $i),
            array_slice($words_of_sent2, 0, $i));
    }
    return $score / $k;
}

// 定义一个链表节点，方便 sim_sent_id 按照相似度插入
class sim_sent_node
{
    public $id;
    public $jaccard_score;
    public $next;

    public function __construct($id = null, $jaccard_score = null, $next = null)
    {
        $this->id = $id;
        $this->jaccard_score = $jaccard_score;
        $this->next = $next;
    }
}

// 定义链表
class sim_sent_list
{
    public $head; // 头节点，默认一个虚头节点
    public $size;

    public function __construct()
    {
        $this->head = new sim_sent_node();
        $this->size = 0;
    }

    // 按照 jaccard_score 相似度插入
    public function jaccard_add($id, $jaccard_score)
    {
        $prev = $this->head;
        while ($prev->next != null && $prev->next->jaccard_score > $jaccard_score) {
            $prev = $prev->next;
        }
        $prev->next = new sim_sent_node($id, $jaccard_score, $prev->next);
        $this->size++;
    }

    public function get_text_list()
    {
        $prev = $this->head;
        if ($this->size == 0) {
            return;
        }

        $ids = "";
        while ($prev->next != null) {
            $ids = $ids . "," . $prev->next->id;
            $prev = $prev->next;
        }
        $ids = substr($ids, 1); // 去掉第一个逗号
        return $ids;
    }

    public function print_list()
    {
        $prev = $this->head;
        if ($this->size == 0) {
            return;
        }
        while ($prev->next != null) {
            print($prev->next->id . "\t" . $prev->next->jaccard_score . "\n");
            $prev = $prev->next;
        }
    }
}

// 将相似句列表存入数据库
function insert_similar_sent_list_into_sqlite($current_id, $text_list)
{
    /* 使用这部分代码先为数据库添加一个 sim_sents 字段
    $add_column = "ALTER TABLE pali_sent ADD COLUMN sim_sents TEXT";
    $Action_add = PDO_Execute($add_column);
    $query = "PRAGMA TABLE_INFO (pali_sent)";
    $Fetch = PDO_FetchALL($query);
    print_r($Fetch);
     */
    global $PDO;
    $Update = "UPDATE pali_sent SET sim_sents = " . $PDO->quote($text_list) . " WHERE id = " . $current_id;
    $Result = PDO_Execute($Update);
    return;
}

// 预计算，存入数据库
function similar_sent_matrix()
{
    // 按照 count = 18, 8, ..., 255 依次获得查询结果 (i-3,i+3)
    //          count = 17,16,...,7                                       (i-2,i+2)
    for ($current_count = 17; $current_count > 7; $current_count--) {
        print("单词数：" . $current_count . "\n");
        $current_query = "select id,text from pali_sent where count=" . $current_count;
        $Current = PDO_FetchAll($current_query);
        if (count($Current)) {
            foreach ($Current as $current_row) {
                $current_id = $current_row['id'];
                $current_sent = $current_row['text'];
                $current_words = words_of_sentence($current_sent);

                // 按照 count > $current_count-3 and count <$current_count+3 查询希望比较的语句
                $compare_query = "select id,text from pali_sent where count>" . ($current_count - 2) . " and count<" . ($current_count + 2);
                $Compare = PDO_FetchALL($compare_query);

                $current_sim_sent_list = new sim_sent_list(); // 新建相似句链表
                foreach ($Compare as $compare_row) {
                    if ($current_row != $compare_row) {
                        $compare_id = $compare_row['id'];
                        $compare_sent = $compare_row['text'];
                        $compare_words = words_of_sentence($compare_sent);

                        $jaccard_score = jaccard_similarity($current_words, $compare_words);
                        if ($jaccard_score > 0.3) {
                            $current_sim_sent_list->jaccard_add($compare_id, $jaccard_score);
                        }
                    }
                } // end of foreach $compare_row

                if ($current_sim_sent_list->size != 0) {
                    print("update " . $current_id . "\n");
                    $text_list = $current_sim_sent_list->get_text_list();
                    insert_similar_sent_list_into_sqlite($current_id, $text_list);
                }
            } // end of foreach $current_row
        }
    }
    return;
}

// 实时计算相似句
function sents_similar_to_id($id)
{
    $query = "SELECT count,text FROM pali_sent WHERE id=" . $id;
    $Current = PDO_FetchALL($query);
    if (count($Current)) {
        foreach ($Current as $current_row) {
            $current_count = $current_row['count'];
            $current_sent = $current_row['text'];
            $current_words = words_of_sentence($current_sent);
            print("current text: \n" . $current_sent . "\n");

            if ($current_count <= 5) {
                print("[-] too short.\n");
                return;
            }

            // 只和单词数大于 5 的比较
            $compare_query = "SELECT id,text FROM pali_sent WHERE count>5";
            $Compare = PDO_FetchALL($compare_query);

            $current_sim_sent_list = new sim_sent_list(); // 新建相似句链表
            foreach ($Compare as $compare_row) {
                if ($current_row != $compare_row) {
                    $compare_id = $compare_row['id'];
                    $compare_sent = $compare_row['text'];
                    $compare_words = words_of_sentence($compare_sent);

                    $jaccard_score = jaccard_similarity($current_words, $compare_words);
                    if ($jaccard_score > 0.3) {
                        print("Jaccard similarity: " . $jaccard_score . "\tSentence id:" . $compare_id . "\n");
                        print("Text: \n" . $compare_sent . "\n");
                        $current_sim_sent_list->jaccard_add($compare_id, $jaccard_score);
                    }
                }
            } // end of foreach $compare_row

            if ($current_sim_sent_list->size != 0) {
                // $current_sim_sent_list->print_list();
            } else {
                print("[-]not found.\n");
            }

        } // end of foreach($Current)
    } // end of if (count($Current))
}

$id = $argv[1];
//sents_similar_to_id($id);

if (!isset($_op)) {
    exit(0);
}

switch ($_op) {
    case "get":
        {
            $Fetch = array();
            if (isset($_word)) {
                $queryWord = str_replace(" ", "", $_word);
                $query = "select book,paragraph,text from pali_sent where \"real\" like " . $PDO->quote("%" . $queryWord . '%') . " limit 0,5";
                $Fetch = PDO_FetchAll($query);
                $newList = array();
                //去掉重复的
                foreach ($Fetch as $onerow) {
                    $found = false;
                    foreach ($newList as $new) {
                        if ($onerow["text"] == $new["text"]) {
                            $found = true;
                            break;
                        }
                    }
                    if ($found == false) {
                        array_push($newList, $onerow);
                    }
                }
                $Fetch = $newList;

                if (count($Fetch) < 5) {
                    $query = "select text from pali_sent where \"real_en\" like " . $PDO->quote('%' . $queryWord . '%') . " limit 0,5";
                    $Fetch2 = PDO_FetchAll($query);
                    //去掉重复的
                    foreach ($Fetch2 as $onerow) {
                        $found = false;
                        foreach ($Fetch as $oldArray) {
                            if ($onerow["word"] == $oldArray["word"]) {
                                $found = true;
                                break;
                            }
                        }
                        if ($found == false) {
                            array_push($Fetch, $onerow);
                        }
                    }
                }

            } else if (isset($_id)) {
            }
            echo json_encode($Fetch, JSON_UNESCAPED_UNICODE);
            break;
        }
}