from zzcore import StdAns

import feedparser, json

allowCMD = ['sub','unsub','list','disable']

# allSub = {
#     'https://farseerfc.me/feeds/atom.xml' : {
#         'url' : 'https://farseerfc.me/feeds/atom.xml',
#         'title' : 'Farseerfc的小窩',
#         'lastUpdated' : 'Sat, 12 Dec 2020 22:55:52 +0800',
#     }
# }

class Ans(StdAns):

    def push(self,title,posttitle,posturl):
        msg = f'{title}\n{posttitle} {posturl}'
        self.sendmsg(msg)
    
    def GETMSG(self):

        if len(self.parms) < 2:
            return '不加参数是坏文明！'
        
        gid = str(self.gid)
        cmd = self.parms[1]
        if cmd not in allowCMD:
            return f'咱不知道 {cmd} 是什么东西 ●﹏●'

        try:
            nowdata = json.loads(self.DATAGET()[gid])
        except:
            nowdata ={
                'allSub' : {},
                'status' : False,
            }
        
        if cmd == 'sub':
            if len(self.parms) < 3:
                return '汝想订阅什么呢？'
            
            suburl = self.parms[2]
            try:
                d = feedparser.parse(suburl)
                title = d.feed.title
                lastUpdated = d.feed.updated
            
            except:
                return "咱好像没能成功订阅 (╥_╥)"
            
            newsub = {
                'url': suburl,
                'title': title,
                'lastUpdated' : lastUpdated,
            }

            nowdata['allSub'][suburl] = newsub

            msg = f'《{title}》订阅成功'

        elif cmd == 'unsub':
            if len(self.parms) < 3:
                return '汝想退订什么呢？'

            suburl = self.parms[2]
            if suburl not in nowdata['allSub'].keys():
                return '汝没有订阅过这个！'
            
            unsub = nowdata['allSub'].pop(suburl)
            
            msg = f'《{unsub.title}》已退订'

        elif cmd == 'list':
            msg = '订阅列表'

            for sub in nowdata['allSub']:
                msg += f"\n{sub['title']}  {sub['url']}"

        elif cmd == 'enable':
            nowdata['status'] = True
            self.DATASET({gid:json.dumps(nowdata)})
            self.sendmsg("订阅已启用，咱会每隔五分钟抓去订阅，有新内容就会推送哦")
            while(nowdata['status']):
                nowdata = json.loads(self.DATAGET()[gid])
                for sub in nowdata['allSub']:
                    try:
                        d = feedparser.parse(sub['url'])
                    except:
                        self.sendmsg(f"咱抓取不到 《{sub['title']}》!")
                        continue
                    if d.feed.updated == sub['lastUpdated']:
                        continue
                    
                    newfeedtitle = d.entries[0].title
                    newfeedlink = d.entries[0].link
                    self.push(sub['title'], newfeedtitle, newfeedlink)
                    nowdata['allSub'][sub['url']]['lastUpdated'] = d.feed.updated
                self.DATASET({gid:json.dumps(nowdata)})

        elif cmd == 'disable':
            nowdata['status'] = False
            msg = '咱不会再推送了！'

        self.DATASET({gid:json.dumps(nowdata)})
        return msg