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-2008, Dave Pare, Jeff Bailey, Thomas Ruschak,
   4  *                           Ken Stevens, Steve McClure
   5  *
   6  *  This program 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 2 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, write to the Free Software
  18  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  *  ---
  21  *
  22  *  See files README, COPYING and CREDITS in the root of the source
  23  *  tree for related information and legal notices.  It is expected
  24  *  that future projects/authors will amend these files as needed.
  25  *
  26  *  ---
  27  *
  28  *  natarg.c: Return countr # given country name or country #
  29  *
  30  *  Known contributors to this file:
  31  *     Markus Armbruster, 2006
  32  */
  33
  34 #include <config.h>
  35
  36 #include <ctype.h>
  37 #include <stdlib.h>
  38 #include "file.h"
  39 #include "match.h"
  40 #include "misc.h"
  41 #include "nat.h"
  42 #include "optlist.h"
  43 #include "player.h"
  44 #include "prototypes.h"
  45
  46 /*
  47  * Get nation argument.
  48  * If ARG is not empty, use it, 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 == 0 || *arg == 0)
  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, 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 && !getcontact(getnatp(player->cnum), np->nat_cnum)) {
  99             if (np->nat_stat != STAT_GOD) {
 100                 pr("Country '%s' has not been contacted.\n", arg);
 101                 return -1;
 102             }
 103         }
 104     }
 105     return np->nat_cnum;
 106 }