local helpers = require('personal.luasnip-helper-funcs')
local get_visual = helpers.get_visual

local line_begin = require("luasnip.extras.expand_conditions").line_begin

return
  {
    -- PRINT STATEMENT
    s({trig="pp", snippetType="autosnippet"},
      fmta(
        [[print(<>)]],
        {
          d(1, get_visual),
        }
      ),
      {condition = line_begin}
    ),
    -- MAIN FUNCTION
    s({trig="main", snippetType="autosnippet"},
      fmta(
        [[
      if __name__ == "__main__":
          <>
      ]],
        {
          d(1, get_visual),
        }
      ),
      {condition = line_begin}
    ),
    -- CLASS
    s({trig="class"},
      fmta(
        [[
        class <>(<>):
            def __init__(self<>):
                <>
        ]],
        {
          i(1),
          i(2),
          i(3),
          i(4),
        }
      ),
      {condition = line_begin}
    ),
    -- EXIT MAIN FUNCTION with sys.exit()
    s({trig="XX", snippetType="autosnippet"},
      { t("sys.exit()") },
      {condition = line_begin}
    ),
    -- FUNCTION DEFINITION WITH CHOICE NODE DOCSTRING
    -- The idea is to let you choose if you want to use the docstring or not
    s({trig="ff", snippetType="autosnippet"},
      fmta(
        [[
      def <>(<>):
          <><>
      ]],
        {
          i(1),
          i(2),
          c(3, {sn(nil, {t({"\"\"\"", ""}), t("    "), i(1, ""), t({"", "    \"\"\"", "    "})}), t("")}),
          -- t("    "),
          d(4, get_visual),
        }
      ),
      {condition = line_begin}
    ),
    -- TIME, i.e. snippet for timing code execution
    s({trig="time"},
      fmta(
        [[
        start = time.time()
        <>
        end = time.time()
      ]],
        {
          d(1, get_visual),
        }
      )
    ),
    -- for _ in range()
    s({trig="frr", snippetType = "autosnippet"},
      fmta(
        [[
        for <> in range(<>):
            <>
      ]],
        {
          i(1),
          i(2),
          i(3)
        }
      ),
      {condition = line_begin}
    ),
    -- IF STATEMENT
    s({trig="iff", snippetType = "autosnippet"},
      fmta(
        [[
        if <>:
            <>
      ]],
        {
          i(1),
          d(2, get_visual),
        }
      ),
      {condition = line_begin}
    ),
    -- with open(filename) as file
    s({trig="wof", snippetType = "autosnippet"},
      fmta(
        [[
        with open(<>) as <>:
            <>
      ]],
        {
          i(1),
          i(2, 'file'),
          i(0),
        }
      ),
      {condition = line_begin}
    ),
    -- RETURN STATEMENT
    s({trig = ";r", snippetType = "autosnippet"},
      { t("return") },
      { condition = line_begin }
    ),
    -- KWARGS STATEMENT
    s({trig = ";k", snippetType = "autosnippet", wordTrig=false},
      { t("kwargs") }
    ),
    -- KWARGS.GET
    s({trig="kgg", snippetType = "autosnippet"},
      fmta(
        [[
        kwargs.get('<>', '<>')
        ]],
        {
          d(1, get_visual),
          i(2)
        }
      )
    ),
    -- SELF (for use in classes)
    s({trig = ";s", snippetType = "autosnippet"},
      { t("self.") }
    ),
    -- SELF (for use in classes) without dot
    s({trig = ";S", snippetType = "autosnippet"},
      { t("self") }
    ),
    -- TRACEBACK
    s({trig = "tbb", snippetType = "autosnippet"},
      { t("print(traceback.format_exc())") },
      { condition = line_begin }
    ),

  }