cx

//// The "Context" data type exists to help create and read different types of
//// data structures.
////
//// For example, to store a key and value:
////
//// ```gleam
//// let context = cx.dict()
////    |> cx.add_string("first_name", first_name)
//// ```
////
//// To retrieve that value:
////
//// ```gleam
//// cx.get_string(context, "first_name")
//// ```
////
//// To store a nested key and value:
////
//// ```gleam
////  let context =
////    cx.dict()
////    |> cx.add(
////     "person",
////     cx.dict()
////       |> cx.add_string("first_name", first_name),
////   )
//// ```
////
//// To retrieve that value:
////
//// ```gleam
//// cx.get_string(context, "person.first_name")
//// ```
////

import gleam/dict.{type Dict}
import gleam/int
import gleam/list
import gleam/string

/// A recursive type for creating and reading nested objects (lists and
/// dictionaries) that contain different combinations of types.
pub type Context {
  CBool(value: Bool)
  CString(value: String)
  CInt(value: Int)
  CFloat(value: Float)
  CDict(value: Dict(String, Context))
  CList(value: List(Context))
}

/// Errors that can occur when reading values from the "context" (a data
/// structure).
pub type ContextError {
  KeyNotFound(key: String)
  ValueNotFound(key: String)
  UnexpectedContextType(key: String)
}

/// Create a data structure to store key-value pairs, where the key is a String.
pub fn dict() -> Context {
  CDict(dict.new())
}

/// Add an integer value, stored under a certain key, to the "context" (a data
/// structure).
pub fn add_int(context: Context, key: String, value: Int) -> Context {
  case context {
    CDict(d) -> CDict(dict.insert(d, key, CInt(value)))
    _ -> context
  }
}

/// Add a float value, stored under a certain key, to the "context" (a data
/// structure).
pub fn add_float(context: Context, key: String, value: Float) -> Context {
  case context {
    CDict(d) -> CDict(dict.insert(d, key, CFloat(value)))
    _ -> context
  }
}

/// Add a string value, stored under a certain key, to the "context" (a data
/// structure).
pub fn add_string(context: Context, key: String, value: String) -> Context {
  case context {
    CDict(d) -> CDict(dict.insert(d, key, CString(value)))
    _ -> context
  }
}

/// Add a boolean value, stored under a certain key, to the "context" (a data
/// structure).
pub fn add_bool(context: Context, key: String, value: Bool) -> Context {
  case context {
    CDict(d) -> CDict(dict.insert(d, key, CBool(value)))
    _ -> context
  }
}

/// Get a value, of any type, stored under a certain key, from the "context" (a
/// data structure).
pub fn get(context: Context, key: String) -> Result(Context, ContextError) {
  case context {
    CDict(d) ->
      case dict.get(d, key) {
        Ok(v) -> Ok(v)
        _ -> Error(KeyNotFound(key))
      }
    _ -> Error(UnexpectedContextType(key))
  }
}

/// Returns a list of values, of varying types, using the "final" part of the
/// key under which the string values are stored. For example, if the key is:
/// "settings.themes", then the "final" key would be "themes".
fn get_list_final(
  context: Context,
  key: String,
) -> Result(List(Context), ContextError) {
  case context {
    CDict(d) -> {
      case dict.get(d, key) {
        Ok(value) -> {
          case value {
            CList(l) -> Ok(l)
            _ -> Error(UnexpectedContextType(key))
          }
        }
        _ -> Error(KeyNotFound(key))
      }
    }
    _ -> Error(ValueNotFound(key))
  }
}

/// Returns a list of values, of varying types, stored under a string key in
/// the "context" (a data structure).
pub fn get_list(
  context: Context,
  key: String,
) -> Result(List(Context), ContextError) {
  case string.split(key, on: ".") {
    [key1] -> get_list_final(context, key1)
    [key1, key2, ..] -> {
      case get(context, key1) {
        Ok(dict1) -> get_list(dict1, key2)
        Error(_) -> Error(ValueNotFound(key1))
      }
    }
    [] -> Error(ValueNotFound(key))
  }
}

/// Returns a list of strings using the "final" part of the key under which the
/// string values are stored. For example, if the key is: "settings.themes",
/// then the "final" key would be "themes".
fn get_strings_final(
  context: Context,
  key: String,
) -> Result(List(String), ContextError) {
  case get_list(context, key) {
    Ok(items) -> {
      Ok(
        list.filter_map(items, fn(item) {
          case item {
            CString(s) -> Ok(s)
            _ -> Error(UnexpectedContextType(key))
          }
        }),
      )
    }
    Error(_) -> Error(ValueNotFound(key))
  }
}

/// Get a list of strings stored in the context under a given string key.
pub fn get_strings(
  context: Context,
  key: String,
) -> Result(List(String), ContextError) {
  case string.split(key, on: ".") {
    [key1] -> get_strings_final(context, key1)
    [key1, key2, ..] -> {
      case get(context, key1) {
        Ok(dict1) -> get_strings(dict1, key2)
        Error(_) -> Error(ValueNotFound(key1))
      }
    }
    [] -> Error(ValueNotFound(key))
  }
}

/// Return a boolean using the "final" part of the key under which the string
/// values are stored. For example, if the key is: "settings.is_dark_theme",
/// then the "final" key would be "is_dark_theme".
fn get_bool_final(
  context: Context,
  key: String,
) -> Result(Bool, ContextError) {
  case context {
    CDict(d) -> {
      case dict.get(d, key) {
        Ok(value) -> {
          case value {
            CBool(b) -> Ok(b)
            CInt(i) -> Ok(i != 0)
            CFloat(f) -> Ok(f != 0.0)
            CString(s) -> Ok(s != "")
            _ -> Error(UnexpectedContextType(key))
          }
        }
        _ -> Error(KeyNotFound(key))
      }
    }
    _ -> Error(UnexpectedContextType(key))
  }
}

/// Get a boolean value, stored under a string key, from the "context" (a
/// data structure).
pub fn get_bool(context: Context, key: String) -> Result(Bool, ContextError) {
  case string.split(key, on: ".") {
    [key1] -> get_bool_final(context, key1)
    [key1, key2, ..] -> {
      case get(context, key1) {
        Ok(dict1) -> get_bool(dict1, key2)
        Error(_) -> Error(ValueNotFound(key1))
      }
    }
    [] -> Error(ValueNotFound(key))
  }
}

/// Return a string using the "final" part of the key under which the string
/// values are stored. For example, if the key is: "settings.theme", then the
/// "final" key would be "theme".
fn get_string_final(
  context: Context,
  key: String,
) -> Result(String, ContextError) {
  case context {
    CDict(d) -> {
      case dict.get(d, key) {
        Ok(value) -> {
          case value {
            CInt(i) -> Ok(int.to_string(i))
            CString(s) -> Ok(s)
            _ -> Error(UnexpectedContextType(key))
          }
        }
        _ -> Error(KeyNotFound(key))
      }
    }
    _ -> Error(UnexpectedContextType(key))
  }
}

/// Get a string value, stored under a string key, from the "context" (a
/// data structure).
pub fn get_string(context: Context, key: String) -> Result(String, ContextError) {
  case string.split(key, on: ".") {
    [key1] -> get_string_final(context, key1)
    [key1, key2, ..] -> {
      case get(context, key1) {
        Ok(dict1) -> get_string(dict1, key2)
        Error(_) -> Error(ValueNotFound(key1))
      }
    }
    [] -> Error(ValueNotFound(key))
  }
}

/// Store a list of strings, under a string key, in the "context" (a data
/// structure).
pub fn add_strings(
  context: Context,
  key: String,
  values: List(String),
) -> Context {
  case context {
    CDict(d) ->
      CDict(dict.insert(
        d,
        key,
        CList(list.map(values, fn(value) { CString(value) })),
      ))
    _ -> context
  }
}

/// Add a value, of any type, under a string key, to the "context" (a data
/// structure).
pub fn add(context: Context, key: String, value: Context) -> Context {
  case context {
    CDict(d) -> CDict(dict.insert(d, key, value))
    _ -> context
  }
}

/// Add a list of values, of any type, under a string key, to the "context" (a
/// data structure).
pub fn add_list(context: Context, key: String, value: List(Context)) -> Context {
  case context {
    CDict(d) -> CDict(dict.insert(d, key, CList(value)))
    _ -> context
  }
}