aboutsummaryrefslogtreecommitdiff
path: root/bridges/DanbooruBridge.php
blob: 3ca4476e0ce47897444c19d11ded027b0dab395e (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
<?php

class DanbooruBridge extends BridgeAbstract
{
    const MAINTAINER = 'mitsukarenai, logmanoriginal';
    const NAME = 'Danbooru';
    const URI = 'http://donmai.us/';
    const CACHE_TIMEOUT = 1800; // 30min
    const DESCRIPTION = 'Returns images from given page';

    const PARAMETERS = [
        'global' => [
            'p' => [
                'name' => 'page',
                'defaultValue' => 1,
                'type' => 'number'
            ],
            't' => [
                'type' => 'text',
                'name' => 'tags',
                'exampleValue' => 'cosplay',
            ]
        ],
        0 => []
    ];

    const PATHTODATA = 'article';
    const IDATTRIBUTE = 'data-id';
    const TAGATTRIBUTE = 'alt';

    protected function getFullURI()
    {
        return $this->getURI()
        . 'posts?&page=' . $this->getInput('p')
        . '&tags=' . urlencode($this->getInput('t'));
    }

    protected function getTags($element)
    {
        return $element->find('img', 0)->getAttribute(static::TAGATTRIBUTE);
    }

    protected function getItemFromElement($element)
    {
        // Fix links
        defaultLinkTo($element, $this->getURI());

        $item = [];
        $item['uri'] = html_entity_decode($element->find('a', 0)->href);
        $item['postid'] = (int)preg_replace('/[^0-9]/', '', $element->getAttribute(static::IDATTRIBUTE));
        $item['timestamp'] = time();
        $thumbnailUri = $element->find('img', 0)->src;
        $item['categories'] = array_filter(explode(' ', $this->getTags($element)));
        $item['title'] = $this->getName() . ' | ' . $item['postid'];
        $item['content'] = '<a href="'
        . $item['uri']
        . '"><img src="'
        . $thumbnailUri
        . '" /></a><br>Tags: '
        . $this->getTags($element);

        return $item;
    }

    public function collectData()
    {
        $html = getSimpleHTMLDOMCached($this->getFullURI());

        foreach ($html->find(static::PATHTODATA) as $element) {
            $this->items[] = $this->getItemFromElement($element);
        }
    }
}