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-2015, 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-2009
  31  */
  32
  33 #include <config.h>
  34
  35 #include <ctype.h>
  36 #include <stdlib.h>
  37 #include "file.h"
  38 #include "match.h"
  39 #include "misc.h"
  40 #include "nat.h"
  41 #include "optlist.h"
  42 #include "player.h"
  43 #include "prototypes.h"
  44
  45 /*
  46  * Get nation argument.
  47  * If @arg is not empty, use it
  48  * Else prompt for input using @prompt.
  49  * If no input is provided, return NULL.
  50  * If the argument identifies a country, return its getnatp() value.
  51  * Else complain and return NULL.
  52  * Caution: this function doesn't care for lack of contact.
  53  */
  54 struct natstr *
  55 natargp(char *arg, char *prompt)
  56 {
  57     char buf[1024];
  58     int n;
  59     struct natstr *np;
  60
  61     arg = getstarg(arg, prompt, buf);
  62     if (!arg || !*arg)
  63         return NULL;
  64     if (isdigit(*arg))
  65         n = atoi(arg);
  66     else {
  67         n = cnumb(arg);
  68         if (n == M_NOTUNIQUE) {
  69             pr("Country '%s' is ambiguous\n", arg);
  70             return NULL;
  71         }
  72     }
  73     np = getnatp(n);
  74     if (!np || np->nat_stat == STAT_UNUSED) {
  75         pr("Country '%s' doesn't exist.\n", arg);
  76         return NULL;
  77     }
  78     return np;
  79 }
  80
  81 /*
  82  * Get nation argument.
  83  * If @arg is not empty, use it.
  84  * Else prompt for input using @prompt.
  85  * If no input is provided, return -1.
  86  * If the argument identifies a country, return its number.  getnatp()
  87  * can be assumed to succeed for this number.
  88  * Else complain and return -1.
  89  * If HIDDEN is enabled, countries not contacted are not eligible
  90  * unless the player is a deity.
  91  */
  92 int
  93 natarg(char *arg, char *prompt)
  94 {
  95     struct natstr *np = natargp(arg, prompt);
  96     if (!np)
  97         return -1;
  98     if (opt_HIDDEN) {
  99         if (!player->god && !getcontact(getnatp(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 }