#include "src/common.h"
#include "src/initpass.h"
#include "src/showpass.h"
#include "src/yankpass.h"
#include "src/listpass.h"
#include "src/addpass.h"
#include "src/delpass.h"
#include "src/genpass.h"
#include "src/otppass.h"

const char *sofname = "sp";
const char *version = "1.3.0";

void helpme() {
  printf("０７６ %s %s - シンプルなパスワードマネージャー\n", sofname, version);
  printf("https://076.moe/ | https://gitler.moe/suwako/%s\n\n", sofname);
  puts  ("使い方：");
  printf(
    "%s -i <gpg-id>               ：GPGと使ってパスワードストレージを初期設定\n",
    sofname
  );
  printf("%s -s <パスワード名>         ：パスワードを表示\n", sofname);
  printf(
    "%s %s%s\n",
    sofname,
    "-y <パスワード名>         ：",
    "パスワードを表示せずクリップボードにコピーする"
  );
  printf("%s -l                        ：パスワード一覧を表示\n", sofname);
  printf("%s -a <パスワード名>         ：パスワードを追加\n", sofname);
  printf("%s -d <パスワード名>         ：パスワードを削除\n", sofname);
  printf("%s -e <パスワード名>         ：パスワードを変更\n", sofname);
  printf(
    "%s %s%s%s\n",
    sofname,
    "-g <文字数> [risk|secure] ：",
    "希望文字数でパスワードをランダムに作成する。",
    "risk＝英数字のみ（不安）、secure＝英数字＋特別文字（デフォルト）を使用"
  );
  printf(
    "%s %s%s\n",
    sofname,
    "-o <パスワード名>         ：",
    "ワンタイムパスワード（TOTP）を表示。存在しなければ、創作する"
  );
  printf("%s -h                        ：ヘルプを表示\n", sofname);
  printf("%s -v                        ：バージョンを表示\n", sofname);
}

void helpme_en() {
  printf("076 %s %s - Simple Password Manager\n", sofname, version);
  printf("https://076.moe/ | https://gitler.moe/suwako/%s\n", sofname);
  puts  ("When reporting issues, please report in Japanese.\n");
  puts  ("Usage:");
  printf(
    "%s %s%s\n",
    sofname,
    "-i <gpg-id>                          : ",
    "First setting for using GPG and password storage"
  );
  printf(
    "%s -s <Password name>                   : Show password\n", sofname
  );
  printf(
    "%s %s%s\n",
    sofname,
    "-y <Password name>                   : ",
    "Copy password to clipboard without show"
  );
  printf(
    "%s %s\n",
    sofname,
    "-l                                   : Show me list of password"
  );
  printf("%s -a <Password name>                   : Add password\n", sofname);
  printf("%s -d <Password name>                   : Delete password\n", sofname);
  printf("%s -e <Password name>                   : Edit password\n", sofname);
  printf("%s %s%s%s%s\n",
    sofname,
    "-g <Characters amount> [risk|secure] : ",
    "Randomly make password with hoped amount.",
    "Using risk = only english letter and number (abnoxious),",
    "secure = english letter and digit + special character (default)"
  );
  printf(
    "%s %s%s\n",
    sofname,
    "-o <Password name>                   : ",
    "Show one time password. If not exist, construct"
  );
  printf("%s -h                                   : Show help\n", sofname);
  printf("%s -v                                   : Show version\n", sofname);
}

char *getfullpath(char *arg) {
  char *lang = getlang();

  char *homedir = getenv("HOME");
  if (homedir == NULL) {
    if (strncmp(lang, "en", 2) == 0)
      perror("Failed to getting home directory");
    else perror("ホームディレクトリを受取に失敗");
    return NULL;
  }

  char *basedir = "/.local/share/sp/";
  size_t fullPathLen;
  char *fullPath;
  if (arg != NULL) {
    fullPathLen = strlen(homedir) + strlen(basedir) + strlen(arg) + 5;
  } else {
    fullPathLen = strlen(homedir) + strlen(basedir);
  }

  fullPath = (char *)malloc(fullPathLen);
  if (fullPath == NULL) {
    if (strncmp(lang, "en", 2) == 0)
      perror("Failed to allocating memory");
    else perror("メモリの役割に失敗");
    if (fullPath) free(fullPath);
    if (homedir) free(homedir);
    return NULL;
  }

  if (arg != NULL) {
    snprintf(fullPath, fullPathLen, "%s%s%s.gpg", homedir, basedir, arg);
  } else {
    snprintf(fullPath, fullPathLen, "%s%s", homedir, basedir);
  }

  return fullPath;
}

int main(int argc, char *argv[]) {
  char *lang = getlang();

  if (argc < 2) {
    if (lang != NULL && strncmp(lang, "en", 2) == 0) helpme_en();
    else helpme();
    return 0;
  }

  if (strcmp(argv[1], "-g") == 0) {
    if (argc != 3 && argc != 4) {
      if (strncmp(lang, "en", 2) == 0) helpme_en();
      else helpme();
      return 1;
    }

    if      (argc == 3) genpass(atoi(argv[2]), true);
    else if (argc == 4 && strcmp(argv[3], "risk") == 0)
      genpass(atoi(argv[2]), false);
    else if (argc == 4 && strcmp(argv[3], "secure") == 0)
      genpass(atoi(argv[2]), true);

    return 0;
  }

  if (argc == 3) {
    if      (strcmp(argv[1], "-i") == 0) initpass(argv[2]);
    else if (strcmp(argv[1], "-s") == 0) printf("%s\n", showpass(argv[2]));
    else if (strcmp(argv[1], "-y") == 0) yankpass(argv[2]);
    else if (strcmp(argv[1], "-a") == 0) addpass(argv[2]);
    else if (strcmp(argv[1], "-d") == 0) delpass(argv[2], 0);
    else if (strcmp(argv[1], "-e") == 0) {
      delpass(argv[2], 1);
      addpass(argv[2]);
    }
    else if (strcmp(argv[1], "-o") == 0) {
      char *fullPath = getfullpath(argv[2]);
      if (fullPath == NULL) return -1;
      otppass(fullPath);
      if (fullPath) free(fullPath);
    } else {
      if (lang != NULL && strncmp(lang, "en", 2) == 0) helpme_en();
      else helpme();
      return 1;
    }
  } else if (argc == 2) {
    char *basePath = getfullpath(NULL);
    if (basePath == NULL) return -1;

    if      (strcmp(argv[1], "-l") == 0) listpass(basePath, 0);
    else if (strcmp(argv[1], "-v") == 0) printf("%s-%s\n", sofname, version);
    else {
      if (strncmp(lang, "en", 2) == 0) helpme_en();
      else helpme();
      if (basePath) free(basePath);
      return 1;
    }
    if (basePath) free(basePath);
  } else {
     if (strncmp(lang, "en", 2) == 0) helpme_en();
     else helpme();
     return 1;
  }

  return 0;
}