termsize cross platform

1 files changed, 21 insertions, 13 deletions

diff --git a/termsize.c b/termsize.c
index 1c0f61b..890ceef 100644
--- a/termsize.c
+++ b/termsize.c
@@ -1,17 +1,6 @@
-#include <sys/ioctl.h>
-#include <stdio.h>
-#include <unistd.h>
-
-int main (int argc, char **argv)
-{
-    struct winsize w;
-    ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
-
-    printf ("lines %d\n", w.ws_row);
-    printf ("columns %d\n", w.ws_col);
-    return 0;  // make sure your main returns int
-}
+#ifdef _WIN32
 
+#include <stdio.h>
 #include <windows.h>
 
 int main(int argc, char *argv[]) 
@@ -27,3 +16,22 @@ int main(int argc, char *argv[])
     printf("rows: %d\n", rows);
     return 0;
 }
+
+#else
+
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+
+int main (int argc, char **argv)
+{
+    struct winsize w;
+    ioctl(STDOUT_FILENO, TIOCGWINSZ, &w);
+
+    printf ("lines %d\n", w.ws_row);
+    printf ("columns %d\n", w.ws_col);
+    return 0;  // make sure your main returns int
+}
+
+#endif
+