-- Counting_Coffee - An example of tabular programming with the goal of unstructured data summation.
-- Copyright (C) 2023 Prince Trippy <programmer@verisimilitudes.net>.

-- This program is free software: you can redistribute it and/or modify it under the terms of the
-- GNU Affero General Public License version 3 as published by the Free Software Foundation.

-- This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
-- even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
-- See the GNU Affero General Public License for more details.

-- You should have received a copy of the GNU Affero General Public License along with this program.
-- If not, see <http://www.gnu.org/licenses/>.

-- This program requires Ada 2005 support due entirely to the function form of Ada.Text_IO.Get_Line.

with Trivial_Trie, Ada.Text_IO, Ada.Command_Line;

procedure Counting_Coffee is
   package Trie is new Trivial_Trie(Index_Type => Positive, Domain_Type => Character,
                                    Input_Type => String,   Stored_Type => Natural);
   Tray : Trie.Trie;
   File : Ada.Text_IO.File_Type;
   Count : Natural := 0;
begin
   Ada.Command_Line.Set_Exit_Status(Ada.Command_Line.Failure);
   Ada.Text_IO.Set_Output(Ada.Text_IO.Standard_Error);
   if Ada.Command_Line.Argument_Count /= 1 then
      Ada.Text_IO.Put_Line("Supply an input file.");
      return;
   end if;
   Ada.Text_IO.Open(File, Mode => Ada.Text_IO.In_File, Name => Ada.Command_Line.Argument(1));
   while not Ada.Text_IO.End_Of_File(File) loop
      declare
         Line : String := Ada.Text_IO.Get_Line(File);
         Here : Boolean;
         Size : Natural;
      begin
         Trie.Find(Path => Line, Tree => Tray, Seen => Here, Unto => Size);
         if not Here then
            loop
               begin
                  Ada.Text_IO.Put_Line(Line);
                  Size := Natural'Value(Ada.Text_IO.Get_Line);
                  exit;
               exception
                  when Constraint_Error => Ada.Text_IO.Put_Line("Supply a natural number.");
               end;
            end loop;
            Trie.Give(Path => Line, Tree => Tray, What => Size);
         end if;
         Count := Count + Size;
      end;
   end loop;
   Ada.Text_IO.Put_Line(Ada.Text_IO.Standard_Output, Natural'Image(Count));
   Ada.Text_IO.Flush(Ada.Text_IO.Standard_Error);
   Ada.Text_IO.Flush(Ada.Text_IO.Standard_Output);
   Ada.Text_IO.Close(File);
   Ada.Command_Line.Set_Exit_Status(Ada.Command_Line.Success);
exception
   when Storage_Error => Ada.Text_IO.Put_Line("Working storage has been exhausted.");
   when others => Ada.Text_IO.Put_Line("Either integer overflow or a file error has occurred.");
end Counting_Coffee;
.