rpl.rb/lib/core/string.rb

# frozen_string_literal: true

module Rpl
  module Lang
    module Core
      module_function

      # convert an object into a string
      def to_string( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [:any] )

        stack << { type: :string,
                   value: Rpl::Lang.stringify( args[0] ) }

        [stack, dictionary]
      end

      # convert a string into an object
      def from_string( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[string]] )

        stack += Rpl::Interpreter.new.parse( args[0][:value] )

        [stack, dictionary]
      end

      # convert ASCII character code in stack level 1 into a string
      def chr( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[numeric]] )

        stack << { type: :string,
                   value: args[0][:value].to_i.chr }

        [stack, dictionary]
      end

      # return ASCII code of the first character of the string in stack level 1 as a real number
      def num( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[string]] )

        stack << { type: :numeric,
                   base: 10,
                   value: args[0][:value].ord }

        [stack, dictionary]
      end

      # return the length of the string
      def size( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[string]] )

        stack << { type: :numeric,
                   base: 10,
                   value: args[0][:value].length }

        [stack, dictionary]
      end

      # search for the string in level 1 within the string in level 2
      def pos( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[string], %i[string]] )

        stack << { type: :numeric,
                   base: 10,
                   value: args[1][:value].index( args[0][:value] ) }

        [stack, dictionary]
      end

      # return a substring of the string in level 3
      def sub( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[numeric], %i[numeric], %i[string]] )

        stack << { type: :string,
                   value: args[2][:value][ (args[1][:value] - 1)..(args[0][:value] - 1) ] }

        [stack, dictionary]
      end

      # reverse string or list
      def rev( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[string list]] )

        result = args[0]

        case args[0][:type]
        when :string
          result = { type: :string,
                     value: args[0][:value].reverse }
        when :list
          result[:value].reverse!
        end

        stack << result

        [stack, dictionary]
      end

      # split string
      def split( stack, dictionary )
        stack, args = Rpl::Lang.stack_extract( stack, [%i[string], %i[string]] )

        args[1][:value].split( args[0][:value] ).each do |elt|
          stack << { type: :string,
                     value: elt }
        end

        [stack, dictionary]
      end
    end
  end
end