Update copyright notice

[empserver] / src / lib / subs / natarg.c

diff --git a/src/lib/subs/natarg.c b/src/lib/subs/natarg.c
index 13e27034e612b3da5094f8891660827af01cb6aa..544923b34507f66b6f6003966025aebd86a9e354 100644 (file)
--- a/src/lib/subs/natarg.c
+++ b/src/lib/subs/natarg.c
@@ -1,11 +1,11 @@
 /*
  *  Empire - A multi-player, client/server Internet based war game.
- *  Copyright (C) 1986-2000, Dave Pare, Jeff Bailey, Thomas Ruschak,
- *                           Ken Stevens, Steve McClure
+ *  Copyright (C) 1986-2020, Dave Pare, Jeff Bailey, Thomas Ruschak,
+ *                Ken Stevens, Steve McClure, Markus Armbruster
  *
- *  This program is free software; you can redistribute it and/or modify
+ *  Empire is free software: you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
- *  the Free Software Foundation; either version 2 of the License, or
+ *  the Free Software Foundation, either version 3 of the License, or
  *  (at your option) any later version.
  *
  *  This program is distributed in the hope that it will be useful,
@@ -14,68 +14,94 @@
  *  GNU General Public License for more details.
  *
  *  You should have received a copy of the GNU General Public License
- *  along with this program; if not, write to the Free Software
- *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  *
  *  ---
  *
- *  See the "LEGAL", "LICENSE", "CREDITS" and "README" files for all the
- *  related information and legal notices. It is expected that any future
- *  projects/authors will amend these files as needed.
+ *  See files README, COPYING and CREDITS in the root of the source
+ *  tree for related information and legal notices.  It is expected
+ *  that future projects/authors will amend these files as needed.
  *
  *  ---
  *
  *  natarg.c: Return countr # given country name or country #
- * 
+ *
  *  Known contributors to this file:
- *     (List known contributors to this file)
+ *     Markus Armbruster, 2006-2016
  */
 
+#include <config.h>
+
 #include <ctype.h>
+#include <stdlib.h>
+#include "match.h"
 #include "misc.h"
 #include "nat.h"
+#include "optlist.h"
 #include "player.h"
-#include "file.h"
 #include "prototypes.h"
-#include "optlist.h"
 
-int
-natarg(s_char *arg, s_char *prompt)
+/*
+ * Get nation argument.
+ * If @arg is not empty, use it
+ * Else prompt for input using @prompt.
+ * If no input is provided, return NULL.
+ * If the argument identifies a country, return its getnatp() value.
+ * Else complain and return NULL.
+ * Caution: this function doesn't care for lack of contact.
+ */
+struct natstr *
+natargp(char *arg, char *prompt)
 {
-       s_char  buf[1024];
-       int     n;
-        struct  natstr *np;
-        int     byname = 0;
+    char buf[1024];
+    int n;
+    struct natstr *np;
 
-       arg = getstarg(arg, prompt, buf);
-       if (arg == 0 || *arg == 0)
-               return -1;
-       if (isdigit(*arg))
-               n = atoi(arg);
-       else {
-               n = cnumb(arg);
-               if (opt_HIDDEN) {
-                   byname = 1;
-               }
+    arg = getstarg(arg, prompt, buf);
+    if (!arg || !*arg)
+       return NULL;
+    if (isdigit(*arg))
+       n = atoi(arg);
+    else {
+       n = cnumb(arg);
+       if (n == M_NOTUNIQUE) {
+           pr("Country '%s' is ambiguous\n", arg);
+           return NULL;
        }
-       if (n < 0 || n >= MAXNOC) {
-               pr("No such country exists.\n");
-               n = -1;
-       }
-       if (opt_HIDDEN) {
-           if (!player->god && !getcontact(getnatp(player->cnum), n)) {
-                if ((np = getnatp(n)) == 0)
-                        return -1;
-                if ((np->nat_stat & STAT_GOD) == 0) {
-                        if (byname) {
-                                pr("No such country exists.\n");
-                                n = -1;
-                        } else {
-                                pr("Country has not been contacted.\n");
-                                n = -2;
-                        }
-                }
+    }
+    np = getnatp(n);
+    if (!np || np->nat_stat == STAT_UNUSED) {
+       pr("Country '%s' doesn't exist.\n", arg);
+       return NULL;
+    }
+    return np;
+}
+
+/*
+ * Get nation argument.
+ * If @arg is not empty, use it.
+ * Else prompt for input using @prompt.
+ * If no input is provided, return -1.
+ * If the argument identifies a country, return its number.  getnatp()
+ * can be assumed to succeed for this number.
+ * Else complain and return -1.
+ * If HIDDEN is enabled, countries not contacted are not eligible
+ * unless the player is a deity.
+ */
+int
+natarg(char *arg, char *prompt)
+{
+    struct natstr *np = natargp(arg, prompt);
+    if (!np)
+       return -1;
+    if (opt_HIDDEN) {
+       if (!player->god
+           && !in_contact(player->cnum, np->nat_cnum)) {
+           if (np->nat_stat != STAT_GOD) {
+               pr("Country '%s' has not been contacted.\n", arg);
+               return -1;
            }
        }
-       return n;
+    }
+    return np->nat_cnum;
 }