Feature: alpha version of the snekeyes dice rolling bot

3 files changed, 94 insertions, 0 deletions

diff --git a/package.lisp b/package.lisp
new file mode 100644
index 0000000..2936e1a
--- /dev/null
+++ b/package.lisp
@@ -0,0 +1,4 @@
+;;;; package.lisp
+
+(defpackage #:snekeyes
+  (:use #:cl #:granolin))
diff --git a/snekeyes.asd b/snekeyes.asd
new file mode 100644
index 0000000..880532a
--- /dev/null
+++ b/snekeyes.asd
@@ -0,0 +1,11 @@
+;;;; snekeyes.asd
+
+(asdf:defsystem #:snekeyes
+  :description "A Matrix bot to roll dice."
+  :author "Shoshin <shshoshin@protonmail.com>"
+  :license  "AGPL"
+  :version "0.0.1"
+  :serial t
+  :depends-on (#:granolin #:cl-ppcre)
+  :components ((:file "package")
+               (:file "snekeyes")))
diff --git a/snekeyes.lisp b/snekeyes.lisp
new file mode 100644
index 0000000..91fdc52
--- /dev/null
+++ b/snekeyes.lisp
@@ -0,0 +1,79 @@
+;;;; snekeyes.lisp
+
+(in-package #:snekeyes)
+
+;;; A dice rolling bot for Granolin
+
+(defclass snekeyes (client auto-joiner) ())
+
+(defvar *snekeyes* nil
+  "Dynamic variable holding the bot instance. Bound by HANDLE-EVENT.")
+
+(defparameter +die-faces+ '(nil ⚀ ⚁ ⚂ ⚃ ⚄ ⚅)
+  "List of unicode characters standard die faces. The face corresponds to the element number")
+
+(defparameter +dice-regex+ (ppcre:create-scanner "([0-9]+)[dD]([0-9]+)")
+  "Looks for a `dice-string' form like 3d6 or 1d20. The number of N sided dice to roll.")
+
+(defparameter +snekeyes-commands+ '("roll!" "craps!")
+  "List of strings that are valid snekeyes commands.")
+
+;; Looks at every text message in the room and responds if the first word is one
+;; of the +snekeyes-commands+
+(defmethod handle-event :after ((*snekeyes* snekeyes) (event text-message-event))
+  (let* ((words (ppcre:split " " (msg-body event)))
+         (command (car words))
+         (dice-string (cadr words)))
+    (if (dice-command-p command)
+        (send-text-message *snekeyes* *room-id* (handle-dice-command command dice-string)))))
+
+(defun dice-command-p (word)
+  "Returns true if the WORD argument is one of the snekeyes commands."
+  (subsetp (list word) +snekeyes-commands+ :test #'equal))
+
+(defun handle-dice-command (command dice-string)
+  "Returns a string result of rolling dice according to COMMAND and DICE-STRING."
+  (cond ((equal "roll!" command) (format-dice-rolls (roll-dice dice-string)))
+        ((equal "craps!" command) (craps))
+        (t "This should not be!!!!")))
+
+(defun roll-dice (dice-string)
+  (let ((parsed (parse-dice-string dice-string)))
+    (if parsed
+        (loop for i upto (- (car parsed) 1)
+              collect (roll (cdr parsed))))))
+
+(defun format-dice-rolls (rolls)
+  (let ((total (total-dice-rolls rolls)))
+    (if (consp (car rolls))
+        (format nil "You rolled ~a, for a total of ~a." (mapcar 'cdr rolls) total)
+        (format nil "You rolled ~a, for a total of ~a." rolls total))))
+
+(defun total-dice-rolls (rolls)
+  (if (consp (car rolls))
+      (reduce '+ (mapcar 'car rolls))
+      (reduce '+ rolls)))
+
+(defun parse-dice-string (str)
+  "Parses strings like 3d6 and returns a cell with (NUMBER . SIDES)"
+  (multiple-value-bind (res matches) (ppcre:scan-to-strings +dice-regex+ str)
+    (if res (cons (read-from-string (elt matches 0)) (read-from-string (elt matches 1))))))
+
+(defun roll (sides)
+  "Dice rolling function. Returns an integer from 1 to SIDES. 
+   If it is a six sided die, returns the result of calling d6"
+  (if (= sides 6) (d6)
+      (+ 1 (random sides))))
+
+(defun d6 ()
+  "Rolls a six sided die and returns (VALUE . FACE)"
+  (let ((result (+ 1 (random 6))))
+    (cons result (elt +die-faces+ result))))
+
+(defun craps ()
+  "Plays the game of craps. Rolls two six-sided dice and returns the result as a string."
+  (let* ((die-1 (d6))
+         (die-2 (d6))
+         (total (+ (car die-1) (car die-2)))
+         (result (if (= 7 total) "Lucky 7! You win!" "Better luck next time...")))
+    (format nil "You rolled ~a ~a. ~a" (cdr die-1) (cdr die-2) result)))