Return to head.cpp CVS log

Up to [Local Repository] / solidite / utils

Annotation of solidite/utils/head.cpp, revision 1.1.1.1

1.1       user        1: /* Copyright (c) 2026 The Solidite Developers
                      2:  *
                      3:  * This program is free software: you can redistribute it and/or modify it under
                      4:  * the terms of the GNU Affero General Public License as published by the Free
                      5:  * Software Foundation, either version 3 of the License, or (at your option)
                      6:  * any later version.
                      7:  *
                      8:  * This program is distributed in the hope that it will be useful, but WITHOUT
                      9:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
                     10:  * FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
                     11:  * details.
                     12:  *
                     13:  * You should have received a copy of the GNU Affero General Public License
                     14:  * along with this program. If not, see <https://www.gnu.org/licenses/>. */
                     15: 
                     16: #include <bitset>
                     17: #include <fstream>
                     18: #include <iostream>
                     19: #include <locale>
                     20: #include <string>
                     21: #include <vector>
                     22: 
                     23: #include <cstring>
                     24: #include <cstdint>
                     25: 
                     26: int main(int argc, char* argv[]) {
                     27:   // locale work
                     28:   std::locale::global(std::locale(""));
                     29:   std::cerr.imbue(std::locale()); 
                     30: 
                     31:   size_t number = 10;
                     32:   std::vector<std::string> files;
                     33:   std::bitset<3> options = 0b000;
                     34:   /* options[0]=stop process options (-- stop processing options if set to 1)
                     35:    * options[1]=skip next argv - in case of -c 10, dont process 10 as a file
                     36:    * options[2]=type (either -c or -n, default=0 -n 10) */
                     37: 
                     38:   for (int i = 1; i < argc; ++i) {
                     39:     if (options.test(1) == 1) {
                     40:       options.reset(1);
                     41:       continue;
                     42:     } else if (options.test(0) == 0) {
                     43:       if (std::strcmp(argv[i], "--") == 0) {
                     44:         options.set(0);
                     45:         continue;
                     46:       } else if (std::strcmp(argv[i], "-c") == 0) {
                     47:         if (i + 1 > argc - 1) {
                     48:           std::cerr << "byte required after -c\n";
                     49:           return 1;
                     50:         } else {
                     51:           number = std::atoi(argv[i + 1]);
                     52:           options.set(2);
                     53:           options.set(1);
                     54:           continue;
                     55:         }
                     56:       } else if (std::strcmp(argv[i], "-n") == 0) {
                     57:         if (i + 1 > argc - 1) {
                     58:           std::cerr << "line number required after -n\n";
                     59:           return 1;
                     60:         } else {
                     61:           number = std::atoi(argv[i + 1]);
                     62:           options.reset(2);
                     63:           options.set(1);
                     64:           continue;
                     65:         }
                     66:       }
                     67:     }
                     68:     files.push_back(argv[i]);
                     69:   }
                     70: 
                     71:   if (files.empty())
                     72:     files.push_back("-");
                     73:   // induce an stdin into the files if empty
                     74:   for (std::string file : files) {
                     75:     if (file == "-") {
                     76:       std::string line;
                     77:       if (options.test(2) == 0) {
                     78:         for (size_t i = 0; i < number; ++i) {
                     79:           std::getline(std::cin, line);
                     80:           std::cout << line + '\n';
                     81:         }
                     82:       } else if (options.test(2) == 1) {
                     83:         size_t chars = 0;
                     84:         while (chars < number) {
                     85:           std::getline(std::cin, line);
                     86:           if (chars + line.length() < number) {
                     87:             line += '\n';
                     88:             chars += line.length();
                     89:           } else {
                     90:             line = line.substr(0, number - chars);
                     91:             std::cout << line;
                     92:             break;
                     93:           }
                     94:           std::cout << line;
                     95:         }
                     96:       }
                     97:       continue;
                     98:     }
                     99: 
                    100:     std::ifstream read(file);
                    101:     if (!read.is_open()) {
                    102:       std::cerr << "failed to read: " + file + '\n';
                    103:       continue;
                    104:     }
                    105:     
                    106:     std::string line;
                    107:     if (options.test(2) == 0) {
                    108:       for (size_t i = 0; i < number; ++i) {
                    109:         std::getline(read, line);
                    110:         std::cout << line + '\n';
                    111:       }
                    112:     } else if (options.test(2) == 1) {
                    113:       size_t chars = 0;
                    114:       while (chars < number) {
                    115:         std::getline(read, line);
                    116:         if (chars + line.length() < number) {
                    117:           line += '\n';
                    118:           chars += line.length();
                    119:         } else {
                    120:           line = line.substr(0, number - chars);
                    121:           std::cout << line;
                    122:           break;
                    123:         }
                    124:         std::cout << line;
                    125:       }
                    126:     }
                    127:     read.close();
                    128:   }
                    129:   return 0;
                    130: }