<gh_stars>100-1000
" seen.py: written by sklnd in about two beers July 2009"

from builtins import object
import time
import unittest

from util import hook, timesince


def db_init(db):
    "check to see that our db has the the seen table and return a connection."
    db.execute(
        "create table if not exists seen(name, time, quote, chan, "
        "primary key(name, chan))"
    )
    db.commit()


@hook.singlethread
@hook.event("PRIVMSG", ignorebots=False)
def seeninput(paraml, input=None, db=None, bot=None):
    db_init(db)
    db.execute(
        "insert or replace into seen(name, time, quote, chan)" "values(?,?,?,?)",
        (input.nick.lower(), time.time(), input.msg, input.chan),
    )
    db.commit()


@hook.command
def seen(inp, nick="", chan="", db=None, input=None):
    ".seen <nick> -- Tell when a nickname was last in active in irc"

    inp = inp.lower()

    if input.conn.nick.lower() == inp:
        # user is looking for us, being a smartass
        return "You need to get your eyes checked."

    if inp == nick.lower():
        return "Have you looked in a mirror lately?"

    db_init(db)

    last_seen = db.execute(
        "select name, time, quote from seen where" " name = ? and chan = ?", (inp, chan)
    ).fetchone()

    if last_seen:
        reltime = timesince.timesince(last_seen[1])
        if last_seen[2][0:1] == "\x01":
            return "%s was last seen %s ago: *%s %s*" % (
                inp,
                reltime,
                inp,
                last_seen[2][8:-1],
            )
        else:
            return "%s was last seen %s ago saying: %s" % (inp, reltime, last_seen[2])
    else:
        return "I've never seen %s" % inp


class SeenTest(unittest.TestCase):
    class Mock(object):
        def __init__(self, **kwargs):
            self.__dict__.update(kwargs)

    def setUp(self):
        import sqlite3

        self.db = sqlite3.connect(":memory:")

    def seeninput(self, nick, msg, chan="#test"):
        seeninput(None, db=self.db, input=self.Mock(nick=nick, msg=msg, chan=chan))

    def seen(self, inp, nick="bob", chan="#test", bot_nick="skybot"):
        return seen(
            inp,
            nick=nick,
            chan=chan,
            db=self.db,
            input=self.Mock(conn=self.Mock(nick=bot_nick)),
        )

    def test_missing(self):
        assert "I've never seen nemo" in self.seen("NEMO")

    def test_seen(self):
        self.seeninput("nemo", "witty banter")
        assert "nemo was last seen" in self.seen("nemo")
        assert "witty banter" in self.seen("nemo")

    def test_seen_missing_channel(self):
        self.seeninput("nemo", "msg", chan="#secret")
        assert "never seen" in self.seen("nemo")

    def test_seen_ctcp(self):
        self.seeninput("nemo", "\x01ACTION test lol\x01")
        assert self.seen("nemo").endswith("ago: *nemo test lol*")

    def test_snark_eyes(self):
        assert "eyes checked" in self.seen("skybot", bot_nick="skybot")

    def test_snark_mirror(self):
        assert "mirror" in self.seen("bob", nick="bob")

" seen.py: written by sklnd in about two beers July 2009"

from builtins import object
import time
import unittest

from util import hook, timesince


def db_init(db):
    "check to see that our db has the the seen table and return a connection."
    db.execute(
        "create table if not exists seen(name, time, quote, chan, "
        "primary key(name, chan))"
    )
    db.commit()


@hook.singlethread
@hook.event("PRIVMSG", ignorebots=False)
def seeninput(paraml, input=None, db=None, bot=None):
    db_init(db)
    db.execute(
        "insert or replace into seen(name, time, quote, chan)" "values(?,?,?,?)",
        (input.nick.lower(), time.time(), input.msg, input.chan),
    )
    db.commit()


@hook.command
def seen(inp, nick="", chan="", db=None, input=None):
    ".seen <nick> -- Tell when a nickname was last in active in irc"

    inp = inp.lower()

    if input.conn.nick.lower() == inp:
        # user is looking for us, being a smartass
        return "You need to get your eyes checked."

    if inp == nick.lower():
        return "Have you looked in a mirror lately?"

    db_init(db)

    last_seen = db.execute(
        "select name, time, quote from seen where" " name = ? and chan = ?", (inp, chan)
    ).fetchone()

    if last_seen:
        reltime = timesince.timesince(last_seen[1])
        if last_seen[2][0:1] == "\x01":
            return "%s was last seen %s ago: *%s %s*" % (
                inp,
                reltime,
                inp,
                last_seen[2][8:-1],
            )
        else:
            return "%s was last seen %s ago saying: %s" % (inp, reltime, last_seen[2])
    else:
        return "I've never seen %s" % inp


class SeenTest(unittest.TestCase):
    class Mock(object):
        def __init__(self, **kwargs):
            self.__dict__.update(kwargs)

    def setUp(self):
        import sqlite3

        self.db = sqlite3.connect(":memory:")

    def seeninput(self, nick, msg, chan="#test"):
        seeninput(None, db=self.db, input=self.Mock(nick=nick, msg=msg, chan=chan))

    def seen(self, inp, nick="bob", chan="#test", bot_nick="skybot"):
        return seen(
            inp,
            nick=nick,
            chan=chan,
            db=self.db,
            input=self.Mock(conn=self.Mock(nick=bot_nick)),
        )

    def test_missing(self):
        assert "I've never seen nemo" in self.seen("NEMO")

    def test_seen(self):
        self.seeninput("nemo", "witty banter")
        assert "nemo was last seen" in self.seen("nemo")
        assert "witty banter" in self.seen("nemo")

    def test_seen_missing_channel(self):
        self.seeninput("nemo", "msg", chan="#secret")
        assert "never seen" in self.seen("nemo")

    def test_seen_ctcp(self):
        self.seeninput("nemo", "\x01ACTION test lol\x01")
        assert self.seen("nemo").endswith("ago: *nemo test lol*")

    def test_snark_eyes(self):
        assert "eyes checked" in self.seen("skybot", bot_nick="skybot")

    def test_snark_mirror(self):
        assert "mirror" in self.seen("bob", nick="bob")
# user is looking for us, being a smartass