Fixed coke handler, now can dispense a drink ^_^

[tpg/opendispense2.git] / src / server / handler_coke.c

diff --git a/src/server/handler_coke.c b/src/server/handler_coke.c
index 17f73a2..030740c 100644 (file)
--- a/src/server/handler_coke.c
+++ b/src/server/handler_coke.c
@@ -6,6 +6,9 @@
  *
  * This file is licenced under the 3-clause BSD Licence. See the file
  * COPYING for full details.
+ *
+ * NOTES:
+ * - Remember, the coke machine echoes your text back to you!
  */
 #include "common.h"
 #include <stdio.h>
@@ -21,7 +24,7 @@
  int   Coke_InitHandler();
  int   Coke_CanDispense(int User, int Item);
  int   Coke_DoDispense(int User, int Item);
-void   WaitForColon();
+ int   WaitForColon();
  int   ReadLine(int len, char *output);
 
 // === GLOBALS ===
@@ -45,34 +48,44 @@ int Coke_InitHandler()
                fprintf(stderr, "ERROR: Unable to open coke serial port ('%s')\n", gsCoke_SerialPort);
        }
        
-       CompileRegex(&gCoke_StatusRegex, "^slot\\s+(\\d)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED);
+       CompileRegex(&gCoke_StatusRegex, "^slot\\s+([0-9]+)\\s+([^:]+):([a-zA-Z]+)\\s*", REG_EXTENDED);
        return 0;
 }
 
 int Coke_CanDispense(int User, int Item)
 {
-       char    tmp[32], *status;
+       char    tmp[40], *status;
        regmatch_t      matches[4];
         int    ret;
 
        // Sanity please
        if( Item < 0 || Item > 6 )      return -1;      // -EYOURBAD
        
-       write(giCoke_SerialFD, "\r\n", 2);
-       write(giCoke_SerialFD, "\r\n", 2);
-       write(giCoke_SerialFD, "\r\n", 2);
-       
-       WaitForColon();
+       ret = 0;
+       do {
+               write(giCoke_SerialFD, "d7\r\n", 4);
+       } while( WaitForColon() && ret++ < 3 );
+
+       if( ret == 3 ) {
+               fprintf(stderr, "Coke machine timed out\n");
+               return -2;      // -EMYBAD
+       }
+
+       // TODO: Handle "not ok" response to D7
        
        // Ask the coke machine
        sprintf(tmp, "s%i\r\n", Item);
        write(giCoke_SerialFD, tmp, 4);
-       
-       WaitForColon();
 
-       ret = ReadLine(sizeof(tmp)-1, tmp);
+       // Read from the machine (ignoring empty lines)
+       while( (ret = ReadLine(sizeof(tmp)-1, tmp)) == 0 );
        printf("ret = %i, tmp = '%s'\n", ret, tmp);
-       
+       if( tmp[0] == ':' ) {
+               ret = ReadLine(sizeof(tmp)-1, tmp);
+               printf("ret = %i, tmp = '%s'\n", ret, tmp);
+       }
+
+       // Catch an error       
        if( ret <= 0 ) {
                fprintf(stderr, "Coke machine is not being chatty (read = %i)\n", ret);
                if( ret == -1 ) {
@@ -80,11 +93,17 @@ int Coke_CanDispense(int User, int Item)
                }
                return -1;
        }
+
+       // Eat rest of response
+       WaitForColon();
+
+       // Parse status response
        ret = RunRegex(&gCoke_StatusRegex, tmp, sizeof(matches)/sizeof(matches[0]), matches, "Bad Response");
        if( ret ) {
                return -1;
        }
 
+       // Get slot status
        tmp[ matches[3].rm_eo ] = '\0';
        status = &tmp[ matches[3].rm_so ];
 
@@ -101,29 +120,47 @@ int Coke_CanDispense(int User, int Item)
  */
 int Coke_DoDispense(int User, int Item)
 {
-       char    tmp[32], *status;
-       regmatch_t      matches[4];
+       char    tmp[32];
+        int    i, ret;
 
        // Sanity please
        if( Item < 0 || Item > 6 )      return -1;
 
-       WaitForColon();
+       // Wait for prompt
+       i = 0;
+       do {
+               write(Item, "d7\r\n", 4);
+       } while( WaitForColon() && i++ < 3 );
 
        // Dispense
        sprintf(tmp, "d%i\r\n", Item);
        write(giCoke_SerialFD, tmp, 4);
        
-       WaitForColon();
-
+       // Read empty lines
+       while( (ret = ReadLine(sizeof(tmp)-1, tmp)) == -1 );
+       if( ret == -1 ) return -1;
+       // Read d%i
+       while( tmp[0] == ':' ) {
+               ret = ReadLine(sizeof(tmp)-1, tmp);
+               if( ret == -1 ) return -1;
+       }
        // Get status
-       ReadLine(sizeof(tmp)-1, tmp);
-       
-       tmp[ matches[3].rm_eo ] = '\0';
-       status = &tmp[ matches[3].rm_so ];
+       ret = ReadLine(sizeof(tmp)-1, tmp);
+       if( ret == -1 ) return -1;
 
-       printf("Machine responded slot status '%s'\n", status);
+       WaitForColon(); // Eat up rest of response
 
-       return 0;
+       // TODO: Regex
+       if( strcmp(tmp, "ok") == 0 ) {
+               // We think dispense worked
+               // - The machine returns 'ok' if an empty slot is dispensed, even if
+               //   it doesn't actually try to dispense (no sound)
+               return 0;
+       }
+
+       printf("Machine returned unknown value '%s'\n", tmp);   
+
+       return -1;
 }
 
 char ReadChar()
@@ -131,11 +168,16 @@ char ReadChar()
        fd_set  readfs;
        char    ch = 0;
         int    ret;
+       struct timeval  timeout;
+       
+       timeout.tv_sec = 5;     // 5 second timeout
+       timeout.tv_usec = 0;
        
        FD_ZERO(&readfs);
        FD_SET(giCoke_SerialFD, &readfs);
        
-       ret = select(giCoke_SerialFD+1, &readfs, NULL, NULL, NULL);
+       ret = select(giCoke_SerialFD+1, &readfs, NULL, NULL, &timeout);
+       if( ret == 0 )  return 0;       // Timeout
        if( ret != 1 ) {
                printf("readchar return %i\n", ret);
                return 0;
@@ -150,7 +192,7 @@ char ReadChar()
        return ch;
 }
 
-void WaitForColon()
+int WaitForColon()
 {
        fd_set  readfs;
        char    ch = 0;
@@ -158,6 +200,10 @@ void WaitForColon()
        FD_SET(giCoke_SerialFD, &readfs);
        
        while( (ch = ReadChar()) != ':' && ch != 0);
+       
+       if( ch == 0 )   return -1;      // Timeout
+       
+       return 0;
 }
 
 int ReadLine(int len, char *output)
@@ -168,20 +214,24 @@ int ReadLine(int len, char *output)
        for(;;)
        {
                ch = ReadChar();
-               
-               
+                       
                if( i < len )
                        output[i++] = ch;
                
                if( ch == '\0' ) {
-                       return -1;
+                       break;
                }
                if( ch == '\n' || ch == '\r' ) {
                        if( i < len )
                                output[--i] = '\0';
-                       return i;
+                       break;
                }
        }
+
+       //printf("ReadLine: output=%s\n", output);
+
+       if( !ch )       return -1;
+       return i;
 }