do some sanity checking on url/quality from command line

[presencevnc] / src / main.cpp

diff --git a/src/main.cpp b/src/main.cpp
index f26d239..bb24fae 100644 (file)
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -16,31 +16,70 @@
     with this program; if not, write to the Free Software Foundation, Inc.,
     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
+#include "mainwindow.h"
+
 #include <QApplication>
-#include <QInputDialog>
 #include <QString>
 
-#include "mainwindow.h"
+#include <iostream>
+
+
+const QString APPNAME("Presence VNC");
+
+void printHelp() {
+       std::cout << "Usage: " << qPrintable(QCoreApplication::arguments().at(0)) << " [options] [URL [quality]]\n"
+
+               << "\nOptions:\n"
+               << " --help\t\t Print this text and exit.\n"
+               << " --viewonly\t Don't send mouse/keyboard input to remote desktop. This is only useful if you also supply a URL.\n"
+
+               << "\nURLs:\n"
+               << " vnc://:password@server:display\n\n"
+               << " Password and display can be omitted, e.g. vnc://server is a valid URL.\n"
+               << " Optionally, you can define the quality as a second argument (1-3, where 1 is the best). Default is 2.\n";
+}
 
 int main(int argc, char *argv[])
 {
-       QCoreApplication::setOrganizationName("Presence VNC");
-       QCoreApplication::setApplicationName("Presence VNC");
+       QCoreApplication::setOrganizationName(APPNAME);
+       QCoreApplication::setApplicationName(APPNAME);
 
        QApplication app(argc, argv);
 
        QString url;
        int quality = 2;
+       bool view_only = false;
+
        QStringList arguments = QCoreApplication::arguments();
-       if (arguments.count() > 1) {
-               //Example: vnc://:password@server:1\n"
-               //"Optionally, you can define the quality as second argument (1-3, where 1 is the best). Default is 2.");
+       for(int i = 1; i < arguments.count(); i++) {
+               if(arguments.at(i) == "--help") {
+                       printHelp();
+
+                       return 0;
+               } else if(arguments.at(i) == "--viewonly") {
+                       view_only = true;
+               } else { //not a valid command line option, should be the url
+                       url = arguments.at(i);
 
-               url = arguments.at(1);
-               if(arguments.count() > 2)
-                       quality = arguments.at(2).toInt();
+                       //check url
+                       if(!url.startsWith("vnc://")) {
+                               std::cerr << "\"" << qPrintable(url) << "\" is not a valid command line option!\n\n";
+                               printHelp();
+
+                               return 1;
+                       }
+
+                       if(arguments.count() > i+1) { //having a --quality option would make more sense.
+                               int arg = arguments.at(i+1).toInt();
+                               if(1 <= arg and arg <= 3) { //check if arg is valid, might also be another option
+                                       quality = arg;
+                                       i++;
+                               }
+                       }
+               }
        }
-       MainWindow main(url, quality);
+
+       MainWindow main(url, quality, view_only);
        main.show();
        return app.exec();
 }