From 837df0f6eef6f957f6abff0940cd71af0baf8316 Mon Sep 17 00:00:00 2001
From: Michael Manganiello <adamantike@users.noreply.github.com>
Date: Tue, 20 Jun 2017 22:13:57 -0300
Subject: [PATCH] Add basic support for wireless printers

This adds basic support for wireless printers, by responding to the
script's prompts. Currently, it always selects the Auto option, so
it could fail in environments with multiple printers
---
 tasks/main.yml | 71 +++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 64 insertions(+), 7 deletions(-)

diff --git a/tasks/main.yml b/tasks/main.yml
index 08a1ad5..d56594a 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -1,6 +1,10 @@
 ---
-- name: ensure needed dependancy are installed
-  apt: name=cups
+- name: ensure needed dependencies are installed
+  apt:
+    name: "{{ item }}"
+  with_items:
+    - cups
+    - expect
   when: ansible_os_family == "Debian"
   tags: ["packages","brother"]
 
@@ -25,10 +29,63 @@
   tags: ["packages","brother"]
 
 - name: install and configure printers
-  shell: yes |{{brother_bin_dir}}/linux-brprinter-installer {{item}} chdir=/tmp
-  register: install_result
+  shell: |
+    # Exit without changes if the printer is already installed
+    set timeout 2
+    spawn /usr/bin/lpstat -a
+    expect {
+        "{{item}}" { exit 4 }
+        timeout {}
+    }
+
+    # Set larger timeout to wait for updates and cupsd restart
+    set timeout 180
+
+    # Run Brother script for printer
+    spawn {{brother_bin_dir}}/linux-brprinter-installer {{item}}
+
+    expect {
+        # Prompt 1: OK? [y/N]
+        "OK?" {
+            send "y\r"
+            exp_continue
+        }
+        # Prompt 2: Do you agree? [Y/n]
+        # This prompt does not appear if you already agreed
+        # Also, it can appear multiple times, depending on the needed packages
+        # to download
+        "Do you agree" {
+            send "y\r"
+            exp_continue
+        }
+        # Prompt 3: Will you specify the Device URI? [Y/n]
+        "specify the Device URI" {
+            send "y\r"
+            exp_continue
+        }
+        # Prompt 4: select the number of destination Device URI.
+        # "A" is the Auto option
+        "number of destination Device URI" {
+            send "A\r"
+            exp_continue
+        }
+        # Prompt 5: Test Print? [y/N]
+        "Test Print" {
+            send "n\r"
+            exp_continue
+        }
+        "Hit Enter" {
+            send "\r"
+            exp_continue
+        }
+        # Script finished
+        eof {}
+    }
+  args:
+    executable: /usr/bin/expect
+    chdir: /tmp
   with_items: "{{brother_printers_to_install}}"
-  changed_when: "'Invalid IP address' in install_result.stderr"
-  failed_when: "'Invalid IP address' not in install_result.stderr"
-  ignore_errors: True
+  register: response
+  changed_when: response.rc == 0
+  failed_when: response.rc != 0 and response.rc != 4
   tags: ["packages","brother"]