Improve #18, start #5

2026-02-25 22:33:49 +01:00 · 2018-10-17 17:03:16 +02:00
parent c116289e13
commit 8e5d88cd3f
2 changed files with 323 additions and 7 deletions
--- a/Source/src/DB.php
+++ b/Source/src/DB.php
@@ -51,6 +51,8 @@ class DB
        $this->createUsers($schema);
        $this->createCodes($schema);
        $this->createSources($schema);
+        $this->createLikes($schema);
+        $this->createSaves($schema);
    }

    private function createUsers(Doctrine\DBAL\Schema\AbstractSchemaManager $schema){
@@ -96,7 +98,7 @@ class DB
            $sources = new Doctrine\DBAL\Schema\Table('Sources');

            $sources->addColumn('IDC','integer',array('unsigned' => true, 'autoincrement' => true));
-            $sources->addColumn('IDU','integer',array('unsigned' => true,'autoincrement' => true));
+            $sources->addColumn('IDU','integer',array('unsigned' => true));
            $sources->addColumn('Lang','string',array('length' => 15));
            $sources->addColumn('Version','integer', array('unsigned' => true));
            $sources->addColumn('Modification','integer', array('unsigned' => true));
@@ -109,8 +111,43 @@ class DB
            $sources->addForeignKeyConstraint('Users',array('IDU'),array('IDU'));
            $sources->addForeignKeyConstraint('Codes',array('IDC'),array('IDC'),array('onUpdate'=>'CASCADE','onDelete'=>'CASCADE'));

-            $schema->createTable($sources)
+            $schema->createTable($sources);
+        }
+    }

+    private function createLikes(\Doctrine\DBAL\Schema\AbstractSchemaManager $schema){
+        if(!$schema->tablesExist(array('Likes'))){
+            $like = new Doctrine\DBAL\Schema\Table('Likes');
+
+            $like->addColumn('IDC','integer',array('unsigned' => true));
+            $like->addColumn('Lang','string',array('length' => 15));
+            $like->addColumn('IDU','integer'    ,array('unsigned'=>true));
+
+            $like->addIndex(array('IDU'));
+            $like->addIndex(array('IDC','Lang'));
+            $like->addForeignKeyConstraint('Users',array('IDU'),array('IDU'),array('onUpdate'=>'CASCADE','onDelete'=>'CASCADE') );
+            $like->addForeignKeyConstraint('Sources',array('IDC','Lang'),array('IDC','Lang'),array('onUpdate'=>'CASCADE','onDelete'=>'CASCADE'));
+            $like->setPrimaryKey(array('IDC','Lang','IDU'));
+
+            $schema->createTable($like);
+        }
+    }
+
+    private function createSaves(\Doctrine\DBAL\Schema\AbstractSchemaManager $schema){
+        if(!$schema->tablesExist(array('Saves'))){
+            $save = new Doctrine\DBAL\Schema\Table('Saves');
+
+            $save->addColumn('IDC','integer',array('unsigned' => true));
+            $save->addColumn('Lang','string',array('length' => 15));
+            $save->addColumn('IDU','integer'    ,array('unsigned'=>true));
+
+            $save->addIndex(array('IDU'));
+            $save->addIndex(array('IDC','Lang'));
+            $save->addForeignKeyConstraint('Users',array('IDU'),array('IDU'),array('onUpdate'=>'CASCADE','onDelete'=>'CASCADE'));
+            $save->addForeignKeyConstraint('Sources',array('IDC','Lang'),array('IDC','Lang'),array('onUpdate'=>'CASCADE','onDelete'=>'CASCADE'));
+            $save->setPrimaryKey(array('IDC','Lang','IDU'));
+
+            $schema->createTable($save);
        }
    }

@@ -737,12 +774,18 @@ class DB
        $queryBuilder
            ->select('token')
            ->from('users')
-            ->where($queryBuilder->expr()-eq('nick','?'))
+            ->where($queryBuilder->expr()->eq('nick','?'))
            ->setParameter(0,$nick);
        $tk = $this->getData($queryBuilder)[0]['token'];
        return $tk == hash('sha256',"$token-$timestamp");
    }

+    /**
+     * Generate a random token to restore password
+     *
+     * @param $email email from user to restore
+     * @return string token with timestamp
+     */
    public function createRestoreToken($email){
        $token = randomString(75);
        $now = time();
@@ -756,4 +799,233 @@ class DB
        $this->execute($queryBuilder);
        return "$token-$now";
    }
+
+    /* Save and like */
+
+    /**
+     * Vote or save a code by user
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang lang of source
+     * @param $table table, Votes or Save
+     * @return int 0 if correct, 1 if error
+     */
+    private function voteOrSave($idu,$idc,$lang,$table){
+        $queryBuilder = $this->newQueryBuilder();
+        $queryBuilder
+            ->insert($table)
+            ->values(
+                array(
+                    "IDU" => '?',
+                    "IDC" => '?',
+                    "Lang"=> '?'
+                )
+            )
+            ->setParameters(array($idu,$idc,$lang));
+
+        try{
+            $this->execute($queryBuilder);
+            return 0;
+        }catch(Exception $ex){
+            return 1;
+        }
+    }
+
+    /**
+     * Vote a code source
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @return int 0 if correct, 1 if error
+     */
+    public function vote($idu,$idc,$lang){
+        return $this->voteOrSave($idu,$idc,$lang,"Votes");
+    }
+
+    /**
+     * Save a code source
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @return int 0 if correct, 1 if error
+     */
+    public function save($idu,$idc,$lang){
+        return $this->voteOrSave($idu,$idc,$lang,"Saves");
+    }
+
+    /**
+     * Remove a vote or save
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang lang of source
+     * @param $table table, Votes or Saves
+     * @return int 0 if correct, 1 if error
+     */
+    private function unvoteOrUnsave($idu,$idc,$lang,$table){
+        $queryBuilder = $this->newQueryBuilder();
+        $queryBuilder
+            ->delete($table)
+            ->where($queryBuilder->expr()->andX(
+                $queryBuilder->expr()->eq('IDU','?'),
+                $queryBuilder->expr()->eq('IDC','?'),
+                $queryBuilder->expr()->eq('Lang','?')
+            ))
+            ->setParameters(array($idu,$idc,$lang));
+        try{
+            $this->execute($queryBuilder);
+            return 0;
+        }catch(Exception $ex){
+            return 1;
+        }
+    }
+
+    /**
+     * Unvote a code source
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @return int 0 if correct, 1 if error
+     */
+    public function unvote($idu,$idc,$lang){
+        return $this->unvoteOrUnsave($idu,$idc,$lang,"Votes");
+    }
+
+    /**
+     * Unsave a code source
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @return int 0 if correct, 1 if error
+     */
+    public function unsave($idu,$idc,$lang){
+        return $this->unvoteOrUnsave($idu,$idc,$lang,"Saves");
+    }
+
+    /**
+     * Get all sources saved by user
+     *
+     * @param $idu user identifier
+     * @return array IDC, Lang and Version
+     */
+    public function allSaves($idu){
+        $queryBuilder = $this->newQueryBuilder();
+        $subQuery = $this->newQueryBuilder();
+        $queryBuilder
+            ->select(array('s.IDC','s.Lang','s.Version'))
+            ->from('Saves','sa')
+            ->join('sa','Sources','s',
+                $queryBuilder->expr()->andX(
+                    $queryBuilder->expr()->eq("sa.IDC","s.IDC"),
+                    $queryBuilder->expr()->eq("sa.Lang","s.Lang")
+                ))
+            ->where(
+                $queryBuilder->expr()->andX(
+                    $queryBuilder->expr()->eq("IDU",'?'),
+                    $queryBuilder->expr()->eq(
+                        'so.Version','('.
+                        $subQuery
+                            ->select('MAX(Version)')
+                            ->from('Sources')
+                            ->where(
+                                $subQuery->expr()->andX(
+                                    $subQuery->expr()->eq('s.Lang','Lang'),
+                                    $subQuery->expr()->eq('s.IDC','IDC')
+                                ))
+                            ->getSql()
+                        .')')
+                )
+            )
+            ->setParameter(0,$idu);
+        return $this->getData($queryBuilder);
+    }
+
+    /**
+     * Count how many saved sources have an user
+     *
+     * @param $idu user identifier
+     * @return int how many saves
+     */
+    public function allSavesCount($idu){
+        $queryBuilder = $this->newQueryBuilder();
+        $queryBuilder
+            ->select('COUNT(*)')
+            ->from('Saves')
+            ->where($queryBuilder->expr()->eq("IDU",'?'))
+            ->setParameter(0,$idu);
+        return $this->getData($queryBuilder)[0]['MAX(*)'];
+    }
+
+    /**
+     * Know if user votes or saves a source
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @param $table Votes or Saves
+     * @return int 0 or 1
+     */
+    private function iVoteSave($idu,$idc,$lang,$table){
+        $queryBuilder = $this->newQueryBuilder();
+        $queryBuilder
+            ->select('COUNT(IDU)')
+            ->from($table)
+            ->where($queryBuilder->expr()->andX(
+                $queryBuilder->expr()->eq('IDU','?'),
+                $queryBuilder->expr()->eq('IDC','?'),
+                $queryBuilder->expr()->eq('Lang','?')
+            ))
+            ->setParameters(array($idu,$idc,$lang));
+        return $this->getData($queryBuilder)[0]['COUNT(IDU)'];
+    }
+
+    /**
+     * Know if user votes or saves a source
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @return int 0 or 1
+     */
+    public function iVote($idu,$idc,$lang){
+        return $this->iVoteSave($idu,$idc,$lang,"Votes");
+    }
+
+    /**
+     * Know if user votes saves a source
+     *
+     * @param $idu user identifier
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @return int 0 or 1
+     */
+    public function iSave($idu,$idc,$lang){
+        return $this->iVoteSave($idu,$idc,$lang,"Saves");
+    }
+
+    /**
+     * How many votes have a source
+     *
+     * @param $idc code identifier
+     * @param $lang source's lang
+     * @return int number of votes
+     */
+    public function sourceLikes($idc,$lang){
+        $queryBuilder = $this->newQueryBuilder();
+        $queryBuilder
+            ->select('COUNT(IDU)')
+            ->from('Votes')
+            ->where($queryBuilder->expr()->andX(
+                $queryBuilder->expr()->eq("IDC",'?'),
+                $queryBuilder->expr()->eq("Lang",'?')
+            ))
+            ->setParameters(array($idc,$lang));
+
+        return $this->getData($queryBuilder)[0]['COUNT(IDU)'];
+    }
 }