<?php

use dokuwiki\Extension\SyntaxPlugin;

/**
 * DokuWiki Plugin dbquery (Syntax Component)
 *
 * @license GPL 2 http://www.gnu.org/licenses/gpl-2.0.html
 * @author  Andreas Gohr <dokuwiki@cosmocode.de>
 */
class syntax_plugin_dbquery_query extends SyntaxPlugin
{
    /** @inheritDoc */
    public function getType()
    {
        return 'substition';
    }

    /** @inheritDoc */
    public function getPType()
    {
        return 'block';
    }

    /** @inheritDoc */
    public function getSort()
    {
        return 135;
    }

    /** @inheritDoc */
    public function connectTo($mode)
    {
        $this->Lexer->addSpecialPattern('{{QUERY:\w+}}', $mode, 'plugin_dbquery_query');
    }

    /** @inheritDoc */
    public function handle($match, $state, $pos, Doku_Handler $handler)
    {
        return ['name' => substr($match, 8, -2)];
    }

    /** @inheritDoc */
    public function render($mode, Doku_Renderer $renderer, $data)
    {
        if ($mode !== 'xhtml') {
            return false;
        }

        /** @var helper_plugin_dbquery $hlp */
        $hlp = plugin_load('helper', 'dbquery');
        try {
            $qdata = $hlp->loadDataFromPage($data['name']);
            $result = $hlp->executeQuery($qdata['codeblocks']['_'], $qdata['macros']['dsn'] ?? null);
        } catch (\Exception $e) {
            msg(hsc($e->getMessage()), -1);
            return true;
        }

        if (count($result) === 1 && isset($result[0]['status']) && isset($qdata['codeblocks'][$result[0]['status']])) {
            $this->renderStatus($result, $qdata['codeblocks'][$result[0]['status']], $renderer);
        } elseif ($qdata['macros']['transpose']) {
            $this->renderTransposedResultTable($result, $renderer);
        } else {
            $this->renderResultTable($result, $renderer);
        }

        return true;
    }

    /**
     * Render given result via the given status HTML
     *
     * @param string[][] $result
     * @param string $html
     * @param Doku_Renderer $R
     */
    public function renderStatus($result, $html, Doku_Renderer $R)
    {
        $value = $result[0]['result'] ?? '';
        $html = str_replace(':result', hsc($value), $html);
        $R->doc .= $html;
    }

    /**
     * Render the given result as a table
     *
     * @param string[][] $result
     * @param Doku_Renderer $R
     */
    public function renderResultTable($result, Doku_Renderer $R)
    {
        global $lang;

        if (!count($result)) {
            $R->cdata($lang['nothingfound']);
            return;
        }

        $R->table_open();
        $R->tablethead_open();
        $R->tablerow_open();
        foreach (array_keys($result[0]) as $header) {
            $R->tableheader_open();
            $R->cdata($header);
            $R->tableheader_close();
        }
        $R->tablerow_close();
        $R->tablethead_close();

        $R->tabletbody_open();
        foreach ($result as $row) {
            $R->tablerow_open();
            foreach ($row as $cell) {
                $R->tablecell_open();
                $this->cellFormat($cell, $R);
                $R->tablecell_close();
            }
            $R->tablerow_close();
        }
        $R->tabletbody_close();
        $R->table_close();
    }

    /**
     * Render the given result as a table, but turned 90 degrees
     *
     * @param string[][] $result
     * @param Doku_Renderer $R
     */
    public function renderTransposedResultTable($result, Doku_Renderer $R)
    {
        global $lang;

        if (!count($result)) {
            $R->cdata($lang['nothingfound']);
            return;
        }

        $width = count($result[0]);
        $height = count($result);

        $R->table_open();
        for ($x = 0; $x < $width; $x++) {
            $R->tablerow_open();
            $R->tableheader_open();
            $R->cdata(array_keys($result[0])[$x]);
            $R->tableheader_close();

            for ($y = 0; $y < $height; $y++) {
                $R->tablecell_open();
                $this->cellFormat(array_values($result[$y])[$x], $R);
                $R->tablecell_close();
            }
            $R->tablerow_close();
        }
        $R->table_close();
    }

    /**
     * Pass the given cell content to the correct renderer call
     *
     * Detects a subset of the wiki link syntax
     *
     * @param string $content
     * @param Doku_Renderer $R
     * @return void
     */
    protected function cellFormat($content, Doku_Renderer $R)
    {
        // external urls
        if (preg_match('/^\[\[(https?:\/\/[^|\]]+)(|.*?)?]]$/', $content, $m)) {
            $url = $m[1];
            $title = $m[2] ?? '';
            $title = trim($title, '|');
            $R->externallink($url, $title);
            return;
        }

        // internal urls
        if (preg_match('/^\[\[([^|\]]+)(|.*?)?]]$/', $content, $m)) {
            $page = cleanID($m[1]);
            $title = $m[2] ?? '';
            $title = trim($title, '|');
            $R->internallink($page, $title);
            return;
        }

        $R->cdata($content);
    }
}