awl/args.ts

import { Args } from "./deps.ts";
import { isRecordType, ServerOptions } from "./lib/utils.ts";
/**
 * A handler for parsing the arguments passed in
 * @param {ServerOptions} server - The DNS server to query
 * @param {Deno.RecordType} type - The type of DNS request, see Deno.RecordType for more info
 * @param {string} name - Server to look up
 */
export type arguments = {
  server?: ServerOptions;
  type?: Deno.RecordType;
  name?: string;
};

/**
 * @param {Args} args - The arguments, directly passed in
 * @returns {arguments} The arguments, parsed
 */
export function parseArgs(args: Args): arguments {
  const parsed: arguments = {} as arguments;
  args._.forEach((arg) => {
    arg = arg.toString();

    // if it starts with an @, it's a server
    if (arg.includes("@")) {
      parsed.server = {
        server: arg.split("@").pop() as string,
        port: args.port,
      };
      return;
    }
    // if there is a dot, it's a name
    if (arg.includes(".")) {
      parsed.name = arg;
      return;
    }

    if (isRecordType(arg)) {
      parsed.type = arg;
      return;
    }

    // if all else fails, assume it's a name
    parsed.name = arg;
  });

  // Add a . to the end of the name if it's not there
  if (parsed.name?.charAt(parsed.name.length - 1) !== ".") {
    parsed.name = parsed.name?.concat(".");
  }

  return parsed;
}