FlatFile.pm 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151
  1. package Trog::Data::FlatFile;
  2. use v5.36;
  3. use re '/aa';
  4. use Carp qw{confess};
  5. use JSON::MaybeXS;
  6. use File::Slurper;
  7. use File::Copy;
  8. use Path::Tiny();
  9. use Capture::Tiny qw{capture_merged};
  10. use lib 'lib';
  11. use Trog::SQLite::TagIndex;
  12. use parent qw{Trog::DataModule};
  13. our $datastore = 'data/files';
  14. sub lang { 'Perl Regex in Quotemeta' }
  15. sub help { 'https://perldoc.perl.org/functions/quotemeta.html' }
  16. =head1 Trog::Data::FlatFile
  17. This data model has multiple drawbacks, but is "good enough" for most low-content and few editor applications.
  18. You can only post once per second due to it storing each post as a file named after the timestamp.
  19. =cut
  20. our $parser = JSON::MaybeXS->new( utf8 => 1 );
  21. # Initialize the list of posts by tag for all known tags.
  22. # This is because the list won't ever change between HUPs
  23. our @tags = Trog::SQLite::TagIndex::tags();
  24. our %posts_by_tag;
  25. sub read ( $self, $query = {} ) {
  26. $query->{limit} //= 25;
  27. #Optimize direct ID
  28. my @index;
  29. if ( $query->{id} ) {
  30. @index = ("$datastore/$query->{id}");
  31. }
  32. else {
  33. # Remove tags which we don't care about and sort to keep memoized memory usage down
  34. @{ $query->{tags} } = sort grep {
  35. my $t = $_;
  36. grep { $t eq $_ } @tags
  37. } @{ $query->{tags} };
  38. my $tagkey = join( '&', @{ $query->{tags} } );
  39. # Check against memoizer
  40. $posts_by_tag{$tagkey} //= [];
  41. @index = @{ $posts_by_tag{$tagkey} } if @{ $posts_by_tag{$tagkey} };
  42. if ( !@index && -f 'data/posts.db' ) {
  43. @index = map { "$datastore/$_" } Trog::SQLite::TagIndex::posts_for_tags( @{ $query->{tags} } );
  44. $posts_by_tag{$tagkey} = \@index;
  45. }
  46. @index = $self->_index() unless @index;
  47. }
  48. my @items;
  49. foreach my $item (@index) {
  50. next unless -f $item;
  51. my $slurped = eval { File::Slurper::read_text($item) };
  52. if ( !$slurped ) {
  53. print "Failed to Read $item:\n$@\n";
  54. next;
  55. }
  56. my $parsed;
  57. capture_merged {
  58. $parsed = eval { $parser->decode($slurped) }
  59. };
  60. if ( !$parsed ) {
  61. # Try and read it in binary in case it was encoded incorrectly the first time
  62. $slurped = eval { File::Slurper::read_binary($item) };
  63. $parsed = eval { $parser->decode($slurped) };
  64. if ( !$parsed ) {
  65. print "JSON Decode error on $item:\n$@\n";
  66. next;
  67. }
  68. }
  69. #XXX this imposes an inefficiency in itself, get() will filter uselessly again here
  70. my @filtered = $query->{raw} ? @$parsed : $self->filter( $query, @$parsed );
  71. push( @items, @filtered ) if @filtered;
  72. next if $query->{limit} == 0; # 0 = unlimited
  73. last if scalar(@items) == $query->{limit};
  74. }
  75. return \@items;
  76. }
  77. sub _index ($self) {
  78. confess "Can't find datastore in $datastore !" unless -d $datastore;
  79. opendir( my $dh, $datastore ) or confess;
  80. my @index = grep { -f } map { "$datastore/$_" } readdir $dh;
  81. closedir $dh;
  82. return sort { $b cmp $a } @index;
  83. }
  84. sub routes ($self) {
  85. return Trog::SQLite::TagIndex::routes();
  86. }
  87. sub aliases ($self) {
  88. return Trog::SQLite::TagIndex::aliases();
  89. }
  90. sub write ( $self, $data ) {
  91. foreach my $post (@$data) {
  92. my $file = "$datastore/$post->{id}";
  93. my $update = [$post];
  94. if ( -f $file ) {
  95. my $slurped = File::Slurper::read_binary($file);
  96. my $parsed = $parser->decode($slurped);
  97. $update = [ ( @$parsed, $post ) ];
  98. }
  99. mkdir $datastore;
  100. open( my $fh, '>', $file ) or confess "Could not open $file";
  101. print $fh $parser->encode($update);
  102. close $fh;
  103. Trog::SQLite::TagIndex::add_post( $post, $self );
  104. }
  105. }
  106. sub count ($self) {
  107. my @index = $self->_index();
  108. return scalar(@index);
  109. }
  110. sub delete ( $self, @posts ) {
  111. foreach my $update (@posts) {
  112. unlink "$datastore/$update->{id}" or confess;
  113. Trog::SQLite::TagIndex::remove_post($update);
  114. }
  115. return 0;
  116. }
  117. sub tags ($self) {
  118. return Trog::SQLite::TagIndex::tags();
  119. }
  120. 1;