git.pond.sub.org Git - empserver/blob - src/lib/subs/natarg.c

   1 /*
   2  *  Empire - A multi-player, client/server Internet based war game.
   3  *  Copyright (C) 1986-2020, Dave Pare, Jeff Bailey, Thomas Ruschak,
   4  *                Ken Stevens, Steve McClure, Markus Armbruster
   5  *
   6  *  Empire is free software: you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation, either version 3 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  *
  19  *  ---
  20  *
  21  *  See files README, COPYING and CREDITS in the root of the source
  22  *  tree for related information and legal notices.  It is expected
  23  *  that future projects/authors will amend these files as needed.
  24  *
  25  *  ---
  26  *
  27  *  natarg.c: Return countr # given country name or country #
  28  *
  29  *  Known contributors to this file:
  30  *     Markus Armbruster, 2006-2016
  31  */
  32
  33 #include <config.h>
  34
  35 #include <ctype.h>
  36 #include <stdlib.h>
  37 #include "match.h"
  38 #include "misc.h"
  39 #include "nat.h"
  40 #include "optlist.h"
  41 #include "player.h"
  42 #include "prototypes.h"
  43
  44 /*
  45  * Get nation argument.
  46  * If @arg is not empty, use it
  47  * Else prompt for input using @prompt.
  48  * If no input is provided, return NULL.
  49  * If the argument identifies a country, return its getnatp() value.
  50  * Else complain and return NULL.
  51  * Caution: this function doesn't care for lack of contact.
  52  */
  53 struct natstr *
  54 natargp(char *arg, char *prompt)
  55 {
  56     char buf[1024];
  57     int n;
  58     struct natstr *np;
  59
  60     arg = getstarg(arg, prompt, buf);
  61     if (!arg || !*arg)
  62         return NULL;
  63     if (isdigit(*arg))
  64         n = atoi(arg);
  65     else {
  66         n = cnumb(arg);
  67         if (n == M_NOTUNIQUE) {
  68             pr("Country '%s' is ambiguous\n", arg);
  69             return NULL;
  70         }
  71     }
  72     np = getnatp(n);
  73     if (!np || np->nat_stat == STAT_UNUSED) {
  74         pr("Country '%s' doesn't exist.\n", arg);
  75         return NULL;
  76     }
  77     return np;
  78 }
  79
  80 /*
  81  * Get nation argument.
  82  * If @arg is not empty, use it.
  83  * Else prompt for input using @prompt.
  84  * If no input is provided, return -1.
  85  * If the argument identifies a country, return its number.  getnatp()
  86  * can be assumed to succeed for this number.
  87  * Else complain and return -1.
  88  * If HIDDEN is enabled, countries not contacted are not eligible
  89  * unless the player is a deity.
  90  */
  91 int
  92 natarg(char *arg, char *prompt)
  93 {
  94     struct natstr *np = natargp(arg, prompt);
  95     if (!np)
  96         return -1;
  97     if (opt_HIDDEN) {
  98         if (!player->god
  99             && !in_contact(player->cnum, np->nat_cnum)) {
 100             if (np->nat_stat != STAT_GOD) {
 101                 pr("Country '%s' has not been contacted.\n", arg);
 102                 return -1;
 103             }
 104         }
 105     }
 106     return np->nat_cnum;
 107 }