acoes.rb

require 'date'
require 'csv'
require 'i18n'
require 'nokogiri'
require 'restclient'
require 'json'
require 'colorize'
require 'sqlite3'
require_relative 'lib/cpf'

class Acoes
  # declaração de constantes para usar no menu
  DADOS_USUARIO = 0
  CADASTRAR_ACAO = 1
  APAGAR_CARTEIRA = 2
  VER_ACOES = 3
  BUSCAR_ACAO = 4
  GRAVAR_CSV = 5
  SAIR = 6

  private_constant :DADOS_USUARIO, :CADASTRAR_ACAO, :VER_ACOES, :APAGAR_CARTEIRA, :VER_ACOES, :BUSCAR_ACAO, :GRAVAR_CSV, :SAIR


  def call
    comprador = {}
    acoes  =  [ ]
    # aux = '1'
    opcao = 0

    #mensagem de boas vindas
    bem_vindo()
    while ( opcao != SAIR )do
      opcao = menu() 
      if ( opcao == DADOS_USUARIO )
        comprador.merge(inserir_dados_usuario)
      elsif ( opcao == CADASTRAR_ACAO )
        acoes << inserir_acao()
      elsif ( opcao == APAGAR_CARTEIRA )
        apagar_carteira(acoes)
      elsif( opcao == VER_ACOES)
        lista_acao(acoes)
      elsif ( opcao == BUSCAR_ACAO)
        puts "Entre a sigla da ação a ser buscada"
        sigla = gets.chomp()
        buscar_acao(acoes,sigla)
      elsif ( opcao == GRAVAR_CSV)
        gravar_csv(acoes)
      end
    end
    apaga_tela()
    puts "Programa encerrado"
  end

  private

  def db
    @db ||= SQLite3::Database.open 'dados/database.db'
  end

  def calcData(stringData)
  end

  def inserir_dados_usuario()
    puts "Nome Completo => "
    nome = gets.chomp()
    puts "C.P.F => "
    cpf = gets.chomp()
    return {nome: nome, cpf: cpf, cpf_valido: Cpf.new(cpf).verifica_cpf}
  end

  def  bem_vindo()
    puts "Lista de Ações com Cálculo do DARF 15%"
    puts "Licença Apache 2.0"
    #puts "Ajude o desenvolvedor"
    # puts "Mande ethereum para o seguinte endereço, com mensagem de solicitaçao"
    # puts "0x2b29C94371E0155f8B0785C98eD1718746366a2b"
    # puts "Obrigado"
  end

  def inserir_acao()
    puts "Digite a sigla da ação :"
    nome = gets.chomp()
    puts "Digite a data de compra no formato: dia(xx)/mes(xx)/ano(xxxx)"
    dataCompra =gets.chomp()
    dataCompra = Date.strptime(dataCompra,'%d/%m/%Y')
    # calculo de dias holdando
    dataAtual = Date.today
    dataHolder = dataAtual - dataCompra
    dataHolder = dataHolder.to_i
    # final do cálculo
    puts "Digite o valor de compra da ação:"
    preco  = gets.to_f()
    puts "Digite a quantidade de ações comprada:"
    quantidade = gets.to_i()    
    puts "Digite o custo da compra - Emolumentos + corretora"
    custoCompra = gets.to_f()
    totalComprado = preco * quantidade
    precoAtual = preco_fechamento(nome)
    crescimento = ((precoAtual * quantidade - custoCompra - totalComprado) / totalComprado) * 100
    apaga_tela()
    puts "Dados digitados foram "
    puts " "
    puts "Data de Compra #{dataCompra} Sigla #{nome} Quantidade #{quantidade} Preço #{preco} Custo #{custoCompra}"
    puts " "
    puts "Verifique as informações. Se estiverem corretas aperte S para sim e N para não =>  "
    sim = gets.chomp()
    if sim == 's' || sim == 'S'
      puts "Salvando os dados"
    else  
      puts "Insira os dados corretos"
      inserir_acao
    end
    return { dia: dataCompra, sigla: nome, quantidade: quantidade, valor: preco, fechamento: precoAtual, total: totalComprado, custos:custoCompra, crescimento: crescimento, dataHolder: dataHolder }
  end

  def apagar_carteira(acoes)
    puts "Escreva o número a ação ou T para apagar toda a carteira" 
    puts " "
    contador = 1
    acoes.each do |acao|   

      puts "\t#{"DATA".ljust(10) } \t#{"SIGLA".ljust(10) } \t#{"QUANTIDADE".ljust(10) } \t#{"VALOR COMPRA".ljust(10) } \t#{"PREÇO ATUAL".ljust(10).green} \t#{"TOTAL".ljust(10) } \t#{"CUSTOS".ljust(10) }"
      puts " "
      print contador.to_s.yellow
      puts "* \t#{ acao[:dia].to_s.ljust(10) } \t #{ acao[:sigla].to_s.ljust(10) } \t#{ acao[:quantidade].to_s.ljust(10) } \tR$ #{ acao[:valor].to_s.ljust(10) } \tR$ #{ acao[:fechamento].to_s.ljust(10).green } \tR$ #{ acao[:total] } \tR$ #{ acao[:custos] }"            
      contador +=1
    
      puts " "
    end
    print "Digite sua escolha => " 
    escolha = gets.chomp()
    if escolha == 't' || escolha == 'T'
        coluna = acoes.first.keys
        arquivo = CSV.generate do |csv|
            csv << coluna
            acoes.each do |acao|
                csv << acao.values
            end
        end
        dia_apagado = Time.now
        #File.write('acoes_antigas_apagadas.txt', dia_apagado, mode: "a")
        open("acoes_apagadas.txt","a"){ |f|
            f << dia_apagado.to_s+"\n"
            f << "====================\n"
            f << arquivo
            f << "====================\n"
            
        }
        File.write('acoes_antigas_apagadas.txt', arquivo)
        puts "Todas as ações cadastradas foram apagadas com sucesso #{dia_apagado}"
        return acoes.clear
  
    elsif escolha.to_i.is_a? Numeric
        escolha = escolha.to_i
        puts "Apagando => #{acoes[escolha-1]} "        
        escolha = escolha - 1
        dia_apagado = Time.now
        open("acoes_apagadas.txt","a"){ |f|
            f << "\nData e Hora em que foi apagada a ação \n"
            f << "\n\t"+dia_apagado.to_s+"\n\n"
            f << "** === Ação Única Apagada ===\n\n"
            f << "\t"+acoes[escolha].to_s
            f << "\n\n xxxx-----------xxxx\n\n"
            
        }

        puts "Ação apagada com sucesso e registrada no arquivo de log"
        acoes.delete_at(escolha)
        return acoes
      end

  end

  def lista_acao(acoes)
    puts "Lista de ações cadastradas" 
    puts " "
    contador = 1
    acoes.each do |acao|
      if acao[:fechamento] > acao[:valor]
          puts "\t#{"DATA".ljust(10) } \t#{"SIGLA".ljust(10) } \t#{"QUANTIDADE".ljust(10) } \t#{"VALOR COMPRA".ljust(10) } \t#{"PREÇO ATUAL".ljust(10).green} \t#{"TOTAL".ljust(10) } \t#{"CUSTOS".ljust(10) } \t#{"DIAS HOLDADOS".ljust(10) }"
          puts " "
          print contador.to_s.yellow
          puts "* \t#{ acao[:dia].to_s.ljust(10) } \t #{ acao[:sigla].to_s.ljust(10) } \t#{ acao[:quantidade].to_s.ljust(10) } \tR$ #{ acao[:valor].to_s.ljust(10) } \tR$ #{ acao[:fechamento].to_s.ljust(10).green } \tR$ #{ acao[:total] } \tR$ #{ acao[:custos] } \t\t Dias #{acao[:dataHolder] }"
          puts " "
          puts "\t#{"CRESCIMENTO".ljust(10).green} => #{ acao[:crescimento].to_s.ljust(10).green }#{"%".green}"
      elsif acao[:fechamento] < acao[:valor]
          puts "\t#{"DATA".ljust(10) } \t#{"SIGLA".ljust(10) } \t#{"QUANTIDADE".ljust(10) } \t#{"VALOR COMPRA".ljust(10) } \t#{"PREÇO ATUAL".ljust(10).red} \t#{"TOTAL".ljust(10) } \t#{"CUSTOS".ljust(10) } \t#{"DIAS HOLDADOS".ljust(10) }"
          puts " "
          print contador.to_s.yellow
          puts "* \t#{ acao[:dia].to_s.ljust(10) } \t #{ acao[:sigla].to_s.ljust(10) } \t#{ acao[:quantidade].to_s.ljust(10) } \tR$ #{ acao[:valor].to_s.ljust(10) } \tR$ #{ acao[:fechamento].to_s.ljust(10).red } \tR$ #{ acao[:total] } \tR$ #{ acao[:custos] } \t\t Dias #{acao[:dataHolder] }"
          puts " "
          puts "\t#{"CRESCIMENTO".ljust(10).red} => #{ acao[:crescimento].to_s.ljust(10).red }#{"%".red}"
          
      end
      contador +=1
      
      puts " "
    end
  end

  def gravar_csv(acoes)
    if acoes.empty?
      apaga_tela()
      return puts "Erro - Necessário Cadastrar as Ações Primeiro."
    end
    coluna = acoes.first.keys
    arquivo = CSV.generate do |csv|
      csv << coluna
      acoes.each do |acao|
        csv << acao.values
      end
    end
    File.write('dados/csv-gerado.csv',arquivo,mode: "a")
    puts "Arquivo gerado com sucesso "
    puts "Nome do arquivo é csv-gerado.csv e esta no diretório do programa"
  end

  def buscar_acao(acoes,nome_acao)
    if acoes.empty?
        apaga_tela()
        return puts "Erro - Necessário Cadastrar as Ações Primeiro."
    end
    preco_medio = 0
    quantidade_acoes = 0
    dtCompra = "Data Compra"
    simbolo = "Sigla"
    qtx = "Quantidade"
    preco = "Valor"
    tt = "Total"
    cost = "Custos"

    puts "\t#{dtCompra.ljust(10)} \t #{simbolo.ljust(10)} \t #{qtx.ljust(10)} \t #{preco.ljust(10)} \t #{tt.ljust(10)} \t #{cost.ljust(10)}"
    puts ""
    acoes.each do |acao|  
      if acao[:sigla] == nome_acao            
        puts "* \t#{ acao[:dia].to_s.ljust(10) } \t #{ acao[:sigla].to_s.ljust(10) } \t#{ acao[:quantidade].to_s.ljust(10) } \tR$ #{ acao[:valor].to_s.ljust(10) } \tR$ #{ acao[:quantidade].to_f * acao[:valor].to_f } \tR$ #{ acao[:custos] }"
        quantidade_acoes = acao[:quantidade] + quantidade_acoes
        preco_medio = preco_medio + acao[:total]
      end
    end
    puts " "
    puts "* \t Preço Médio para #{nome_acao} => R$ #{preco_medio / quantidade_acoes}"
  end

  def preco_fechamento(sigla)
  # pega a sigla e devolve o valor atual pelo google finance
    url = "https://www.google.com/finance/quote/#{sigla}:BVMF"
    response = Nokogiri::HTML(RestClient.get(url))
    fechamento = response.css('.kf1m0').css('.kf1m0').first.text
    fechamento.slice!("R$")
    valor = fechamento.to_f  
    puts valor
    return valor
  end

  def apaga_tela()
    system("clear") || system("cls")
  end

  def menu()
    puts "================================================="
    puts "[ #{DADOS_USUARIO} ] Dados do Usuário"
    puts "[ #{CADASTRAR_ACAO} ] Cadastrar nova Ação ou Carteira"
    puts "[ #{APAGAR_CARTEIRA} ] Apagar Carteira ou Ação"
    puts "[ #{VER_ACOES} ] Ver Todas as Ações cadastradas"
    puts "[ #{BUSCAR_ACAO} ] Buscar Ação"
    puts "[ #{GRAVAR_CSV} ] Gravar o arquivo CSV"
    puts "[ #{SAIR} ] Sair"
    puts "================================================="
    print 'Digite a Opção  =>  '
    return gets.to_i
  end
end

Acoes.new.call