+++ /dev/null
-#!/usr/bin/guile \
--e main -s
-!#
-
-;;; 8sync --- Asynchronous programming for Guile
-;;; Copyright (C) 2015 Christopher Allan Webber <cwebber@dustycloud.org>
-;;;
-;;; This file is part of 8sync.
-;;;
-;;; 8sync is free software: you can redistribute it and/or modify it
-;;; under the terms of the GNU Lesser General Public License as
-;;; published by the Free Software Foundation, either version 3 of the
-;;; License, or (at your option) any later version.
-;;;
-;;; 8sync is distributed in the hope that it will be useful,
-;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
-;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
-;;; GNU Lesser General Public License for more details.
-;;;
-;;; You should have received a copy of the GNU Lesser General Public
-;;; License along with 8sync. If not, see <http://www.gnu.org/licenses/>.
-
-(define-module (eightsync systems irc)
- #:use-module (eightsync repl)
- #:use-module (eightsync agenda)
- #:use-module (srfi srfi-9)
- #:use-module (ice-9 getopt-long)
- #:use-module (ice-9 format)
- #:use-module (ice-9 receive)
- #:use-module (ice-9 q)
- #:use-module (ice-9 match)
- #:export (;; The only things you definitely need if writing a bot
- make-irc-bot-cli
- irc-format irc-display irc-send-message irc-send-formatted
-
- ;; Useful things if you're making something more complicated
- irc-line
- irc-eol
-
- default-irc-port
-
- startswith-colon?
-
- <irc-line>
- make-irc-line irc-line?
- irc-line-prefix irc-line-command irc-line-params
-
- parse-line
- irc-line-username
-
- condense-privmsg-line
- echo-back-message
-
- make-handle-line make-basic-irc-handler
- queue-and-start-irc-agenda!))
-
-\f
-;;; Network stuff
-;;; =============
-
-(define default-irc-port 6665)
-
-(define* (irc-socket-setup hostname #:optional (inet-port default-irc-port))
- (let ((s (socket PF_INET SOCK_STREAM 0))
- (ip-address (inet-ntoa (car (hostent:addr-list (gethost hostname))))))
- (connect s AF_INET
- (inet-pton AF_INET ip-address)
- inet-port)
- s))
-
-(define (install-socket socket handler)
- (display "Installing socket...\n") ; debugging :)
- (make-port-request socket #:read handler))
-
-(define irc-eol "\r\n")
-
-(define (irc-line line)
- (string-concatenate (list line irc-eol)))
-
-(define-syntax-rule (irc-format dest format-string rest ...)
- (let ((line (string-concatenate
- (list (format #f format-string rest ...)
- irc-eol))))
- (match dest
- (#f line)
- (#t (display line))
- (else
- (display line dest)))))
-
-(define* (irc-display line #:optional dest)
- (if dest
- (display (irc-line line) dest)
- (display (irc-line dest))))
-
-(define (irc-send-message socket channel message)
- (irc-format socket "PRIVMSG ~a :~a" channel message))
-
-(define-syntax-rule (irc-send-formatted socket channel format-string
- args ...)
- (irc-format socket "PRIVMSG ~a :~a" channel
- (format #f format-string args ...)))
-
-(define* (handle-login socket username
- #:key
- (hostname "*")
- (servername "*")
- (realname username)
- (channels '()))
- (irc-format socket "USER ~a ~a ~a :~a"
- username hostname servername realname)
- (irc-format socket "NICK ~a" username)
- (for-each
- (lambda (channel)
- (irc-format socket "JOIN ~a" channel))
- channels))
-
-(define (startswith-colon? str)
- (and (> (string-length str) 0)
- (eq? (string-ref str 0)
- #\:)))
-
-(define-record-type <irc-line>
- (make-irc-line prefix command params)
- irc-line?
- (prefix irc-line-prefix)
- (command irc-line-command)
- (params irc-line-params))
-
-
-(define (parse-line line)
- (define (parse-params pre-params)
- ;; This is stupid and imperative but I can't wrap my brain around
- ;; the right way to do it in a functional way :\
- (let ((param-list '())
- (currently-building '()))
- (for-each
- (lambda (param-item)
- (cond
- ((startswith-colon? param-item)
- (if (not (eq? currently-building '()))
- (set! param-list
- (cons
- (reverse currently-building)
- param-list)))
- (set! currently-building (list param-item)))
- (else
- (set! currently-building (cons param-item currently-building)))))
- pre-params)
- ;; We're still building something, so tack that on there
- (if (not (eq? currently-building '()))
- (set! param-list
- (cons (reverse currently-building) param-list)))
- ;; return the reverse of the param list
- (reverse param-list)))
-
- (match (string-split line #\space)
- (((? startswith-colon? prefix)
- command
- pre-params ...)
- (make-irc-line prefix command
- (parse-params pre-params)))
- ((command pre-params ...)
- (make-irc-line #f command
- (parse-params pre-params)))))
-
-(define (strip-colon-if-necessary string)
- (if (and (> (string-length string) 0)
- (string-ref string 0))
- (substring/copy string 1)
- string))
-
-;; @@: Not sure if this works in all cases, like what about in a non-privmsg one?
-(define (irc-line-username irc-line)
- (let* ((prefix-name (strip-colon-if-necessary (irc-line-prefix irc-line)))
- (exclaim-index (string-index prefix-name #\!)))
- (if exclaim-index
- (substring/copy prefix-name 0 exclaim-index)
- prefix-name)))
-
-(define (condense-privmsg-line line)
- "Condense message line and do multiple value return of
- (channel message is-action)"
- (define (strip-last-char string)
- (substring/copy string 0 (- (string-length string) 1)))
- (let* ((channel-name (caar line))
- (rest-params (apply append (cdr line))))
- (match rest-params
- (((or "\x01ACTION" ":\x01ACTION") middle-words ... (= strip-last-char last-word))
- (values channel-name
- (string-join
- (append middle-words (list last-word))
- " ")
- #t))
- (((= strip-colon-if-necessary first-word) rest-message ...)
- (values channel-name
- (string-join (cons first-word rest-message) " ")
- #f)))))
-
-(define (echo-back-message socket my-name speaker
- channel-name message is-action)
- (if is-action
- (format #t "~a emoted ~s in channel ~a\n"
- speaker message channel-name)
- (format #t "~a said ~s in channel ~a\n"
- speaker message channel-name)))
-
-(define default-handle-privmsg echo-back-message)
-
-(define* (make-handle-line #:key
- (handle-privmsg default-handle-privmsg))
- (define (handle-line socket line my-username)
- (let ((parsed-line (parse-line line)))
- (match (irc-line-command parsed-line)
- ("PING"
- (irc-display "PONG" socket))
- ("PRIVMSG"
- (receive (channel-name message is-action)
- (condense-privmsg-line (irc-line-params parsed-line))
- (let ((username (irc-line-username parsed-line)))
- (handle-privmsg socket my-username username
- channel-name message is-action))))
- (_
- (display line)
- (newline)))))
- handle-line)
-
-(define (make-basic-irc-handler handle-line username)
- (let ((buffer '()))
- (define (reset-buffer)
- (set! buffer '()))
- (define (should-read-char socket)
- (and (char-ready? socket) (not (eof-object? (peek-char socket)))))
- (define (irc-handler socket)
- (while (should-read-char socket)
- (set! buffer (cons (read-char socket) buffer))
- (match buffer
- ((#\newline #\return (? char? line-chars) ...)
- (let ((ready-line (list->string (reverse line-chars))))
- ;; reset buffer
- (set! buffer '())
- ;; run it
- (%8sync (%run (handle-line
- socket
- ready-line
- username)))))
- (_ #f))))
- irc-handler))
-
-(define default-line-handler (make-handle-line))
-
-(define* (queue-and-start-irc-agenda! agenda socket #:key
- (username "syncbot")
- (inet-port default-irc-port)
- (line-handler default-line-handler)
- (channels '()))
- (dynamic-wind
- (lambda () #f)
- (lambda ()
- (enq! (agenda-queue agenda)
- (wrap (install-socket
- socket
- (make-basic-irc-handler
- line-handler
- username))))
- (enq! (agenda-queue agenda) (wrap (handle-login socket username
- #:channels channels)))
- (start-agenda agenda))
- (lambda ()
- (display "Cleaning up...\n")
- (close socket))))
-
-
-\f
-;;; CLI
-;;; ===
-
-(define option-spec
- `((server (single-char #\s) (required? #t) (value #t))
- (port (single-char #\p)
- (value #t)
- (predicate
- ,(lambda (s)
- (if (string->number s) #t #f))))
- (username (single-char #\u) (required? #t) (value #t))
- (channels (value #t))
- (listen)))
-
-(define* (make-irc-bot-cli #:optional
- (line-handler default-line-handler)
- (print-and-continue-on-error #t))
- (define (main args)
- (let* ((options (getopt-long args option-spec))
- (hostname (option-ref options 'server #f))
- (port (or (option-ref options 'port #f)
- default-irc-port))
- (username (option-ref options 'username #f))
- (listen (option-ref options 'listen #f))
- (channels (option-ref options 'channels ""))
- (agenda (if print-and-continue-on-error
- (make-agenda #:pre-unwind-handler print-error-and-continue)
- (make-agenda))))
- (display `((server ,hostname) (port ,port) (username ,username)
- (listen ,listen) (channels-split ,(string-split channels #\space))))
- (newline)
- (if listen
- (spawn-and-queue-repl-server! agenda))
- (queue-and-start-irc-agenda!
- agenda
- (irc-socket-setup hostname port)
- #:inet-port port
- #:username username
- #:channels (string-split channels #\space)
- #:line-handler line-handler)))
- main)
-
-(define main (make-irc-bot-cli))
projects / 8sync.git / blobdiff