Permite pagamento para despesa com dependente.

[cascardo/declara.git] / lib / pagamento.c

diff --git a/lib/pagamento.c b/lib/pagamento.c
index 1e91115..821a3f1 100644 (file)
--- a/lib/pagamento.c
+++ b/lib/pagamento.c
@@ -69,7 +69,7 @@ static int pagamento_cmp(void *p1, void *p2)
        return 0;
 }
 
-static struct pagamento * pagamento_new(char **args)
+static struct pagamento * pagamento_new(char **args, int argc)
 {
        struct pagamento *pagamento;
        int r = 0;
@@ -89,6 +89,15 @@ static struct pagamento * pagamento_new(char **args)
                pagamento_free(pagamento);
                return NULL;
        }
+       if (argc == 7) {
+               r = set_int(&args[5], 2, &pagamento->dependente);
+       } else {
+               pagamento->dependente = 0;
+       }
+       if (r < 0 || pagamento->dependente < 0) {
+               pagamento_free(pagamento);
+               return NULL;
+       }
        return pagamento;
 }
 
@@ -96,11 +105,15 @@ static int run_pagamento(struct declaracao *dec, char **args, int argc)
 {
        struct pagamento *pagamento;
        int r;
-       if (argc != 6)
+       if (argc != 6 && argc != 7)
                return -EINVAL;
-       pagamento = pagamento_new(args);
+       pagamento = pagamento_new(args, argc);
        if (!pagamento)
                return -ENOMEM;
+       if (pagamento->dependente > list_size(dec->dependentes)) {
+               pagamento_free(pagamento);
+               return -EINVAL;
+       }
        r = list_insert_ordered(&dec->pagamentos, pagamento, pagamento_cmp);
        if (r < 0) {
                pagamento_free(pagamento);
@@ -119,8 +132,8 @@ void pagamento_salva(struct declaracao *dec, FILE *f)
        int i;
        struct pagamento *j;
        for (i = 0; j = list_get(dec->pagamentos, i); i++)
-               fprintf(f, "pagamento %d \"%s\" \"%s\" %lld %lld\n",
-                       j->codigo, j->cnpj, j->nome, j->pagamento, j->reembolso);
+               fprintf(f, "pagamento %d \"%s\" \"%s\" %lld %lld %d\n",
+                       j->codigo, j->cnpj, j->nome, j->pagamento, j->reembolso, j->dependente);
 }
 
 static struct cmd cmd_pagamento = {